【Hackathon 6th Fundable Projects 3 No.353】shuffle_channel (#64654)

* Fix * Fix
PaddlePaddle · May 29, 2024 · 2fab7de · 2fab7de
1 parent 362c877
commit 2fab7de
Show file tree

Hide file tree

Showing 8 changed files with 19 additions and 197 deletions.
diff --git a/paddle/fluid/operators/ops_signature/shuffle_channel_sig.cc b/paddle/fluid/operators/ops_signature/shuffle_channel_sig.cc
diff --git a/paddle/fluid/operators/shuffle_channel_op.cc b/paddle/fluid/operators/shuffle_channel_op.cc
diff --git a/paddle/phi/ops/yaml/backward.yaml b/paddle/phi/ops/yaml/backward.yaml
@@ -2564,6 +2564,16 @@
     func: set_value_grad
     param: [out_grad, starts, ends, steps, axes, decrease_axes, none_axes]
 
+- backward_op : shuffle_channel_grad
+  forward : shuffle_channel (Tensor x, int group = 1) -> Tensor(out)
+  args : (Tensor out_grad, int group = 1)
+  output : Tensor(x_grad)
+  infer_meta :
+    func : UnchangedInferMeta
+    param : [out_grad]
+  kernel :
+    func : shuffle_channel_grad
+
 - backward_op : sigmoid_cross_entropy_with_logits_grad
   forward : sigmoid_cross_entropy_with_logits (Tensor x, Tensor label, Tensor pos_weight, bool normalize=false, int ignore_index=-100) -> Tensor(out)
   args : (Tensor x, Tensor label, Tensor pos_weight, Tensor out_grad, bool normalize, int ignore_index)

diff --git a/paddle/phi/ops/yaml/inconsistent/dygraph_backward.yaml b/paddle/phi/ops/yaml/inconsistent/dygraph_backward.yaml
@@ -380,16 +380,6 @@
     func: set_value_with_scalar_grad
     param: [out_grad, starts, ends, steps, axes, decrease_axes, none_axes]
 
-- backward_op : shuffle_channel_grad
-  forward : shuffle_channel (Tensor x, int group=1) -> Tensor(out)
-  args : (Tensor out_grad, int group)
-  output : Tensor(x_grad)
-  infer_meta :
-    func : UnchangedInferMeta
-    param : [out_grad]
-  kernel :
-    func : shuffle_channel_grad
-
 - backward_op : slice_double_grad
   forward : slice_grad (Tensor input, Tensor grad_out, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis) -> Tensor(grad_input)
   args : (Tensor grad_input_grad, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis)

diff --git a/paddle/phi/ops/yaml/inconsistent/dygraph_ops.yaml b/paddle/phi/ops/yaml/inconsistent/dygraph_ops.yaml
@@ -505,15 +505,6 @@
     func : set_value
   backward: set_value_grad
 
-- op : shuffle_channel
-  args : (Tensor x, int group=1)
-  output : Tensor(out)
-  infer_meta :
-    func : ShuffleChannelInferMeta
-  kernel :
-    func : shuffle_channel
-  backward : shuffle_channel_grad
-
 - op : slice
   args : (Tensor input, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis)
   output : Tensor

diff --git a/paddle/phi/ops/yaml/inconsistent/static_backward.yaml b/paddle/phi/ops/yaml/inconsistent/static_backward.yaml
@@ -490,16 +490,6 @@
     func: assign
     param: [out_grad]
 
-- backward_op : shuffle_channel_grad
-  forward : shuffle_channel (Tensor x, int group = 1) -> Tensor(out)
-  args : (Tensor out_grad, int group = 1)
-  output : Tensor(x_grad)
-  infer_meta :
-    func : UnchangedInferMeta
-    param : [out_grad]
-  kernel :
-    func : shuffle_channel_grad
-
 - backward_op : slice_double_grad
   forward : slice_grad (Tensor input, Tensor grad_out, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis) -> Tensor(grad_input)
   args : (Tensor grad_input_grad, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis)

diff --git a/paddle/phi/ops/yaml/inconsistent/static_ops.yaml b/paddle/phi/ops/yaml/inconsistent/static_ops.yaml
@@ -1038,15 +1038,6 @@
   backward : shuffle_batch_grad
   traits : pir::SideEffectTrait
 
-- op : shuffle_channel
-  args : (Tensor x, int group = 1)
-  output : Tensor(out)
-  infer_meta :
-    func : ShuffleChannelInferMeta
-  kernel :
-    func : shuffle_channel
-  backward : shuffle_channel_grad
-
 - op : slice
   args : (Tensor input, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis)
   output : Tensor

diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -3511,6 +3511,15 @@
   kernel :
     func : shard_index
 
+- op : shuffle_channel
+  args : (Tensor x, int group = 1)
+  output : Tensor(out)
+  infer_meta :
+    func : ShuffleChannelInferMeta
+  kernel :
+    func : shuffle_channel
+  backward : shuffle_channel_grad
+
 - op : sigmoid
   args : (Tensor x)
   output : Tensor