Fix

PaddlePaddle · Aug 29, 2024 · 7363b18 · 7363b18
1 parent d4ef0f4
commit 7363b18
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 24 deletions.
diff --git a/paddle/phi/ops/yaml/inconsistent/static_ops.yaml b/paddle/phi/ops/yaml/inconsistent/static_ops.yaml
@@ -910,18 +910,6 @@
   backward : subtract_grad
   interfaces : paddle::dialect::InferSymbolicShapeInterface
 
-- op : sync_comm_stream
-  args : (Tensor x, int ring_id=0)
-  output : Tensor(out)
-  infer_meta :
-    func : UnchangedInferMeta
-    param : [x]
-  kernel :
-    func : sync_comm_stream
-    param: [x, ring_id]
-    data_type : DataType::FLOAT32
-  inplace : (x -> out)
-
 - op : tile
   args : (Tensor x, IntArray repeat_times = {})
   output : Tensor(out)

diff --git a/paddle/phi/ops/yaml/legacy/static_ops.yaml b/paddle/phi/ops/yaml/legacy/static_ops.yaml
@@ -883,18 +883,6 @@
     func : swish
   backward : swish_grad
 
-- op : sync_comm_stream
-  args : (Tensor x, int ring_id=0)
-  output : Tensor(out)
-  infer_meta :
-    func : UnchangedInferMeta
-    param : [x]
-  kernel :
-    func : sync_comm_stream
-    param: [x, ring_id]
-    data_type : DataType::FLOAT32
-  inplace : (x -> out)
-
 - op : topk_v1
   args : (Tensor x, Scalar(int) k = 1)
   output : Tensor(out), Tensor(indices)

diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -4628,6 +4628,18 @@
     func : sync_calc_stream
   inplace : (x -> out)
 
+- op : sync_comm_stream
+  args : (Tensor x, int ring_id=0)
+  output : Tensor(out)
+  infer_meta :
+    func : UnchangedInferMeta
+    param : [x]
+  kernel :
+    func : sync_comm_stream
+    param: [x, ring_id]
+    data_type : DataType::FLOAT32
+  inplace : (x -> out)
+
 - op : take_along_axis
   args : (Tensor arr, Tensor indices, int axis)
   output : Tensor