bug fix for stride_slice when strides < 0 on XPU (#62923)

PaddlePaddle · Mar 27, 2024 · 064a998 · 064a998
1 parent 84a7446
commit 064a998
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/paddle/phi/kernels/xpu/stride_slice_grad_kernel.cc b/paddle/phi/kernels/xpu/stride_slice_grad_kernel.cc
@@ -66,7 +66,12 @@ void StridedSliceRawGradKernel(const Context& dev_ctx,
       end = xshape[cur_axe];
     }
     if (end < 0) {
-      end += xshape[cur_axe];
+      if (!(end == -1 && strides_[i] < 0)) {
+        end = end + xshape[cur_axe];
+        if (end < 0) {
+          end = 0;
+        }
+      }
     }
 
     ends_in[cur_axe] = end;

diff --git a/paddle/phi/kernels/xpu/stride_slice_kernel.cc b/paddle/phi/kernels/xpu/stride_slice_kernel.cc
@@ -81,7 +81,12 @@ void StridedSliceRawKernel(const Context& dev_ctx,
       end = xshape[cur_axe];
     }
     if (end < 0) {
-      end += xshape[cur_axe];
+      if (!(end == -1 && strides_[i] < 0)) {
+        end = end + xshape[cur_axe];
+        if (end < 0) {
+          end = 0;
+        }
+      }
     }
 
     ends_in[cur_axe] = end;