Skip to content

Fix a bug of strided_slice op, about the axes parameter access memory out of bounds #35062

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 7 commits into from
Aug 23, 2021
Merged

Fix a bug of strided_slice op, about the axes parameter access memory out of bounds #35062

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
9e38dbf
Pass compat of conv_transpose_bias_mkldnn_fuse_pass
TeslaZhao Jun 21, 2021
c07b8d7
fix confilct
TeslaZhao Jul 13, 2021
a50d27b
Merge branch 'PaddlePaddle-develop' into develop-new
TeslaZhao Jul 13, 2021
9eb942a
Merge branch 'PaddlePaddle:develop' into develop-new
TeslaZhao Aug 20, 2021
36b66c1
Merge branch 'PaddlePaddle:develop' into develop-new
TeslaZhao Aug 22, 2021
d695a5d
Fix a bug of strided_slice op, about the axes parameter access memory…
TeslaZhao Aug 22, 2021
0fc6499
Fix a bug of strided_slice op, about the axes parameter access memory…
TeslaZhao Aug 23, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions paddle/fluid/operators/strided_slice_op.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,20 @@ class StridedSliceOp : public framework::OperatorWithKernel {
auto ends_size = ends.size();
auto strides_size = strides.size();

for (size_t i = 0; i < axes.size(); ++i) {
PADDLE_ENFORCE_GE(axes[i], 0,
platform::errors::InvalidArgument(
"The axis should be greater than or equal to 0."
"But received %d of axes[%d]",
axes[i], i));
PADDLE_ENFORCE_LT(
axes[i], in_dims.size(),
platform::errors::InvalidArgument(
"The axes should be less than or equal to input tensor's rank."
"But received %d of axes[%d], input tensor shape [%d]",
axes[i], i, in_dims.size()));
}

if (ctx->HasInputs("StartsTensorList")) {
auto StartsTensorList = ctx->Inputs("StartsTensorList");
PADDLE_ENFORCE_GT(
Expand Down