fix bug of strided_slice (#43388)

* fix stride_slice bug * fix bug
PaddlePaddle · Jun 13, 2022 · abc5d0c · abc5d0c
1 parent ee60567
commit abc5d0c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/paddle/phi/kernels/funcs/strided_slice.h b/paddle/phi/kernels/funcs/strided_slice.h
@@ -74,10 +74,14 @@ static void StridedSliceOutDims(const std::vector<int64_t>& starts,
 
     if (start_index < 0) {
       start_index = start_index + axis_size;
+      start_index = std::max<int64_t>(start_index, 0);
     }
     if (end_index < 0) {
       if (!(end_index == -1 && stride_index < 0)) {  // skip None stop condition
         end_index = end_index + axis_size;
+        if (end_index < 0) {
+          end_index = 0;
+        }
       }
     }