MCPcopy Index your code
hub / github.com/apache/tvm / has_cuda

Function has_cuda

python/tvm/testing/env.py:175–177  ·  view source on GitHub ↗

True if a CUDA device is present and enabled in TVM_TEST_TARGETS.

()

Source from the content-addressed store, hash-verified

173
174
175def has_cuda() -> bool:
176 """True if a CUDA device is present and enabled in TVM_TEST_TARGETS."""
177 return _device_usable("cuda")
178
179
180def has_rocm() -> bool:

Callers 8

has_cudnnFunction · 0.85
has_cublasFunction · 0.85
has_ncclFunction · 0.85
has_nvshmemFunction · 0.85
has_cuda_computeFunction · 0.85
has_nvcc_versionFunction · 0.85
has_tensorcoreFunction · 0.85
has_cudagraphFunction · 0.85

Calls 1

_device_usableFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…