Skip to content

Commit 8649fdd

Browse files
committed
validate reduction axes before nchw remap in static reduce
1 parent 76de138 commit 8649fdd

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

src/subgraph/static-reduce.c

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,14 @@ static enum xnn_status reshape_reduce_operator(
134134
reduction_axes[i] = 0 <= opdata->reduction_axes[i]
135135
? opdata->reduction_axes[i]
136136
: input_num_dims + opdata->reduction_axes[i];
137+
if (reduction_axes[i] < 0 || reduction_axes[i] >= (int64_t) input_num_dims) {
138+
xnn_log_error(
139+
"failed to reshape Reduce operator with input ID #%" PRIu32
140+
": reduction axis #%d (%" PRId64
141+
") is out of bounds for a %zu-dimensional input",
142+
input_id, i, reduction_axes[i], input_num_dims);
143+
return xnn_status_invalid_parameter;
144+
}
137145
}
138146
size_t input_dims[XNN_MAX_TENSOR_DIMS];
139147
memcpy(input_dims, input_value->shape.dim, input_num_dims * sizeof(size_t));

0 commit comments

Comments
 (0)