Update oneflow/user/kernels/collective_communication/include/recv.h

Co-authored-by: binbinHan <[email protected]>
Oneflow-Inc · Jan 24, 2025 · 3e67ade · 3e67ade
1 parent b369a2c
commit 3e67ade
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/oneflow/user/kernels/collective_communication/include/recv.h b/oneflow/user/kernels/collective_communication/include/recv.h
@@ -33,7 +33,7 @@ class Recv : public CollectiveCommunication {
   virtual void Launch(ep::Stream* stream, void* out, size_t elem_cnt, int64_t src) const = 0;
 
   virtual void Launch(ep::Stream* stream, void* out, size_t elem_cnt, int64_t src,
-                      CclComm ccl_comm) const = 0;
+                      const CclComm& ccl_comm) const = 0;
 };
 
 inline bool IsRecvRegistered(DeviceType device_type) {