This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch tirx/standalone-package-cleanup in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 04f31ef66962d3c4587f618f5b246931eba906a7 Author: Tianqi Chen <[email protected]> AuthorDate: Thu Oct 15 07:54:21 2026 +0000 [FIX][Testing] Probe the loaded NCCL runtime capability --- tirx/python/tvm_tirx/testing/env.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tirx/python/tvm_tirx/testing/env.py b/tirx/python/tvm_tirx/testing/env.py index 0fbf896493..7e58d5c269 100644 --- a/tirx/python/tvm_tirx/testing/env.py +++ b/tirx/python/tvm_tirx/testing/env.py @@ -284,7 +284,12 @@ def has_cublas() -> bool: def has_nccl() -> bool: """True if NCCL was built in and a CUDA device is present.""" - return has_cuda() and build_flag_enabled("USE_NCCL") + try: + return has_cuda() and ( + tvm_tirx.get_global_func("runtime.disco.nccl.init_ccl", allow_missing=True) is not None + ) + except Exception: # pylint: disable=broad-except + return False def has_hipblas() -> bool:
