Update oneflow/user/kernels/eager_nccl_s2s_kernel.cu

Co-authored-by: binbinHan <[email protected]>
Oneflow-Inc · Jan 24, 2025 · 8110bd2 · 8110bd2
1 parent 5636f1b
commit 8110bd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/oneflow/user/kernels/eager_nccl_s2s_kernel.cu b/oneflow/user/kernels/eager_nccl_s2s_kernel.cu
@@ -35,7 +35,7 @@ class EagerNcclOpKernelCache final : public user_op::OpKernelCache {
   ~EagerNcclOpKernelCache() override = default;
 
   Symbol<ParallelDesc> parallel_desc() const { return parallel_desc_; }
-  ccl::CclComm ccl_comm() const { return ccl_comm_; }
+  const ccl::CclComm& ccl_comm() const { return ccl_comm_; }
 
  private:
   void Init(user_op::KernelCacheContext* ctx) {