MCPcopy Create free account
hub / github.com/apache/tvm / has_nccl

Function has_nccl

python/tvm/testing/env.py:250–252  ·  view source on GitHub ↗

True if NCCL was built in and a CUDA device is present.

()

Source from the content-addressed store, hash-verified

248
249
250def has_nccl() -> bool:
251 """True if NCCL was built in and a CUDA device is present."""
252 return has_cuda() and _build_flag_enabled("USE_NCCL")
253
254
255def has_hipblas() -> bool:

Callers

nothing calls this directly

Calls 2

has_cudaFunction · 0.85
_build_flag_enabledFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…