(tvm) branch main updated: [BugFix] Fix NCCL build with GlobalDef registration (#18158)

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 5aa4dfd644 [BugFix] Fix NCCL build with GlobalDef registration (#18158)
5aa4dfd644 is described below

commit 5aa4dfd64417c0837970d2cd1bc6445fd6b1e947
Author: Ruihang Lai <ruiha...@cs.cmu.edu>
AuthorDate: Mon Jul 21 07:56:30 2025 -0400

    [BugFix] Fix NCCL build with GlobalDef registration (#18158)
    
    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 9e41bbd0de..32a1940726 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) {