Update oneflow/user/kernels/nccl_logical_2d_sbp_kernels.cpp

Co-authored-by: binbinHan <[email protected]>
Oneflow-Inc · Jan 24, 2025 · 9bb1fb8 · 9bb1fb8
1 parent 8110bd2
commit 9bb1fb8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/oneflow/user/kernels/nccl_logical_2d_sbp_kernels.cpp b/oneflow/user/kernels/nccl_logical_2d_sbp_kernels.cpp
@@ -42,7 +42,7 @@ class NcclLogical2DSameDim0KernelCommState : public user_op::OpKernelState {
   }
   ~NcclLogical2DSameDim0KernelCommState() override = default;
 
-  ccl::CclComm ccl_comm() {
+  const ccl::CclComm& ccl_comm() const {
     if (!is_init_) { Init(); }
     return ccl_comm_;
   }