!10561 [ms][lite][cpu] fix concat bug

From: @lzkcode
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
pull/10561/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 8927f2d644

@ -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;

Loading…
Cancel
Save