Add error message for slice op(PaddlePaddle#30851)

Layssy · Feb 8, 2021 · 97f7a70 · 97f7a70
1 parent 87197f8
commit 97f7a70
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/paddle/fluid/operators/slice_op.cc b/paddle/fluid/operators/slice_op.cc
@@ -121,6 +121,13 @@ class SliceOp : public framework::OperatorWithKernel {
           start = std::max(start, 0);
           end = std::max(end, 0);
           end = std::min(end, dim_value);
+
+          PADDLE_ENFORCE_LE(start, dim_value,
+                            platform::errors::InvalidArgument(
+                                "start should be less than or equal to the "
+                                "dimension value, but received "
+                                "start = %d, shape[%d] = %d.",
+                                starts[i], axes[i], out_dims[axes[i]]));
           PADDLE_ENFORCE_GT(end, start,
                             platform::errors::InvalidArgument(
                                 "end should greater than start, but received "