From 5cba2b061b588016335fb20f2c6468637bacfb2a Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Sun, 20 Jul 2025 14:37:32 -0400 Subject: [PATCH] [BugFix] Fix NCCL build with GlobalDef registration This PR fixes a build failure in nccl.cc due to the recent switch of global function registration. --- src/runtime/disco/nccl/nccl.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/runtime/disco/nccl/nccl.cc b/src/runtime/disco/nccl/nccl.cc index 9e41bbd0deb5..32a194072653 100644 --- a/src/runtime/disco/nccl/nccl.cc +++ b/src/runtime/disco/nccl/nccl.cc @@ -329,9 +329,9 @@ void SyncWorker() { TVM_FFI_STATIC_INIT_BLOCK({ namespace refl = tvm::ffi::reflection; - refl::GlobalDef().def("runtime.disco.compiled_ccl", - []() -> String { return TVM_DISCO_CCL_NAME; }); - .def("runtime.disco." TVM_DISCO_CCL_NAME ".init_ccl", InitCCL) + refl::GlobalDef() + .def("runtime.disco.compiled_ccl", []() -> String { return TVM_DISCO_CCL_NAME; }) + .def("runtime.disco." TVM_DISCO_CCL_NAME ".init_ccl", InitCCL) .def("runtime.disco." TVM_DISCO_CCL_NAME ".init_ccl_per_worker", InitCCLPerWorker) .def("runtime.disco." TVM_DISCO_CCL_NAME ".allreduce", [](NDArray send, int kind, bool in_group, NDArray recv) {