polish

PaddlePaddle · Nov 13, 2023 · 27ef949 · 27ef949
1 parent b25e06e
commit 27ef949
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 13 deletions.
diff --git a/paddle/phi/api/yaml/legacy_backward.yaml b/paddle/phi/api/yaml/legacy_backward.yaml
@@ -662,7 +662,7 @@
   infer_meta :
     func : GeneralUnaryGradInferMeta
     param : [x]
-    spmd_rule : StridedGradInferSpmdDynamic
+    spmd_rule : StridedSliceGradInferSpmdDynamic
   kernel :
     func : strided_slice_grad
   no_need_buffer : x

diff --git a/paddle/phi/infermeta/spmd_rules/slice.cc b/paddle/phi/infermeta/spmd_rules/slice.cc
@@ -288,12 +288,12 @@ SpmdInfo StridedSliceInferSpmdDynamic(const DistMetaTensor& input,
   return SliceInferSpmdBase(input, axes_bridge);
 }
 
-SpmdInfo StridedGradInferSpmdDynamic(const DistMetaTensor& input,
-                                     const DistMetaTensor& out_grad,
-                                     const std::vector<int>& axes,
-                                     const IntArray& starts,
-                                     const IntArray& ends,
-                                     const IntArray& strides) {
+SpmdInfo StridedSliceGradInferSpmdDynamic(const DistMetaTensor& input,
+                                          const DistMetaTensor& out_grad,
+                                          const std::vector<int>& axes,
+                                          const IntArray& starts,
+                                          const IntArray& ends,
+                                          const IntArray& strides) {
   std::vector<int64_t> axes_bridge(axes.begin(), axes.end());
   return SliceGradInferBase(input, out_grad, axes_bridge);
 }

diff --git a/paddle/phi/infermeta/spmd_rules/slice.h b/paddle/phi/infermeta/spmd_rules/slice.h
@@ -62,12 +62,12 @@ SpmdInfo StridedSliceInferSpmdDynamic(const DistMetaTensor& input,
                                       const IntArray& ends,
                                       const IntArray& strides);
 
-SpmdInfo StridedGradInferSpmdDynamic(const DistMetaTensor& input,
-                                     const DistMetaTensor& out_grad,
-                                     const std::vector<int>& axes,
-                                     const IntArray& starts,
-                                     const IntArray& ends,
-                                     const IntArray& strides);
+SpmdInfo StridedSliceGradInferSpmdDynamic(const DistMetaTensor& input,
+                                          const DistMetaTensor& out_grad,
+                                          const std::vector<int>& axes,
+                                          const IntArray& starts,
+                                          const IntArray& ends,
+                                          const IntArray& strides);
 
 }  // namespace distributed
 }  // namespace phi