diff --git a/mindspore/lite/nnacl/fp32/concat_fp32.c b/mindspore/lite/nnacl/fp32/concat_fp32.c index 99e547bd66..4f2568f341 100644 --- a/mindspore/lite/nnacl/fp32/concat_fp32.c +++ b/mindspore/lite/nnacl/fp32/concat_fp32.c @@ -35,7 +35,11 @@ void Concat(const void **input, int input_num, int axis, const int **inputs_outp const uint8_t *src_base = (input[i]); size_t input_stride = after_axis_size * inputs_output_shape[i][axis]; int offset = UP_DIV(input_stride, thread_num); - int count = MSMIN(offset, input_stride - offset * task_id); + int count = input_stride - offset * task_id; + if (count <= 0) { + continue; + } + count = MSMIN(offset, count); for (int j = 0; j < before_axis_size; j++) { const uint8_t *src = src_base + j * input_stride + task_id * offset; uint8_t *dst = dst_base + j * output_stride + axis_offset * after_axis_size + task_id * offset;