Fix

PaddlePaddle · Sep 23, 2024 · e069aa8 · e069aa8
1 parent b299a46
commit e069aa8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/paddle/phi/ops/yaml/inconsistent/dygraph_ops.yaml b/paddle/phi/ops/yaml/inconsistent/dygraph_ops.yaml
@@ -326,6 +326,16 @@
   inplace : (x -> out)
   backward : subtract_grad
 
+- op : sync_comm_stream
+  args : (Tensor[] x, int ring_id = 0)
+  output : Tensor[](out){x.size()}
+  infer_meta :
+    func : UnchangedVectorInferMeta
+    param : [x]
+  kernel :
+    func : sync_comm_stream
+    data_type : DataType::FLOAT32
+
 - op : tile
   args : (Tensor x, IntArray repeat_times = {})
   output : Tensor(out)

diff --git a/python/paddle/distributed/communication/group.py b/python/paddle/distributed/communication/group.py
@@ -255,9 +255,7 @@ def _sync_calc_stream(tensor):
 
 def _sync_comm_stream(tensor, ring_id=0):
     if framework.in_dynamic_mode():
-        return paddle._legacy_C_ops.sync_comm_stream(
-            [tensor], [tensor], 'ring_id', ring_id
-        )
+        return paddle._C_ops.sync_comm_stream([tensor], ring_id)
     else:
         op_type = 'c_sync_comm_stream'
         helper = framework.LayerHelper(op_type, **locals())