Fix

PaddlePaddle · Oct 25, 2024 · 0bdc1d3 · 0bdc1d3
1 parent 0c10106
commit 0bdc1d3
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 18 deletions.
diff --git a/paddle/phi/ops/yaml/backward.yaml b/paddle/phi/ops/yaml/backward.yaml
@@ -2088,6 +2088,12 @@
   kernel :
     func : mode_grad
 
+- backward_op : mp_allreduce_sum_grad
+  forward : mp_allreduce_sum(Tensor x, int ring_id = 0) -> Tensor(out)
+  args : (Tensor out_grad, int ring_id = 0)
+  output : Tensor(x_grad)
+  invoke : c_identity(out_grad, ring_id, false, false)
+
 - backward_op : multi_dot_grad
   forward : multi_dot (Tensor[] x) -> Tensor(out)
   args : (Tensor[] x, Tensor out_grad)

diff --git a/paddle/phi/ops/yaml/inconsistent/static_backward.yaml b/paddle/phi/ops/yaml/inconsistent/static_backward.yaml
@@ -409,12 +409,6 @@
     func : minimum_grad
   composite : minimum_grad(x, y, out_grad, x_grad, y_grad)
 
-- backward_op : mp_allreduce_sum_grad
-  forward : mp_allreduce_sum(Tensor x, int ring_id = 0) -> Tensor(out)
-  args : (Tensor out_grad, int ring_id = 0)
-  output : Tensor(x_grad)
-  invoke : c_identity(out_grad, ring_id, false, false)
-
 - backward_op : multiply_double_grad
   forward : multiply_grad (Tensor x, Tensor y, Tensor grad_out, int axis = -1) -> Tensor(grad_x), Tensor(grad_y)
   args : (Tensor x, Tensor y, Tensor grad_out, Tensor grad_x_grad, Tensor grad_y_grad, int axis = -1)

diff --git a/paddle/phi/ops/yaml/inconsistent/static_ops.yaml b/paddle/phi/ops/yaml/inconsistent/static_ops.yaml
@@ -643,18 +643,6 @@
   backward : minimum_grad
   interfaces : paddle::dialect::InferSymbolicShapeInterface
 
-- op : mp_allreduce_sum
-  args : (Tensor x, int ring_id = 0)
-  output : Tensor(out)
-  infer_meta :
-    func : AllReduceInferMeta
-    param: [x]
-  kernel :
-    func : mp_allreduce_sum
-    param: [x]
-  backward: mp_allreduce_sum_grad
-  inplace: (x -> out)
-
 - op : multiply
   args : (Tensor x, Tensor y)
   output : Tensor

diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -3444,6 +3444,18 @@
   inplace : (param -> param_out), (velocity -> velocity_out), (master_param -> master_param_out)
   traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait
 
+- op : mp_allreduce_sum
+  args : (Tensor x, int ring_id = 0)
+  output : Tensor(out)
+  infer_meta :
+    func : AllReduceInferMeta
+    param: [x]
+  kernel :
+    func : mp_allreduce_sum
+    param: [x]
+  backward: mp_allreduce_sum_grad
+  inplace: (x -> out)
+
 - op : multi_dot
   args : (Tensor[] x)
   output : Tensor