MCPcopy Create free account
hub / github.com/dmlc/xgboost / _checkcu

Function _checkcu

demo/guide-python/distributed_extmem_basic.py:41–45  ·  view source on GitHub ↗
(status: "cudaError_t")

Source from the content-addressed store, hash-verified

39
40
41def _checkcu(status: "cudaError_t") -> None:
42 import cuda.bindings.runtime as cudart
43
44 if status != cudart.cudaError_t.cudaSuccess:
45 raise RuntimeError(cudart.cudaGetErrorString(status))
46
47
48def device_mem_total() -> int:

Callers 2

device_mem_totalFunction · 0.70
setup_async_poolFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected