Browse Source

fix concat bug

tags/v1.1.0
lzk 5 years ago
parent
commit
8beaebbe4c
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      mindspore/lite/nnacl/fp32/concat_fp32.c

+ 5
- 1
mindspore/lite/nnacl/fp32/concat_fp32.c View File

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