p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!13751 [MSLITE][Develop] fix bug of arm cpu op gather while running in multi-threaded mode 

From: @yangruoqi713
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-03-23 10:26:41 +08:00 committed by Gitee
parent ab5d77f943 36187c427e
commit 12a29ce040
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/src/runtime/kernel/arm/fp16/gather_fp16.cc
View File

@ -119,8 +119,8 @@ int GatherFp16CPUKernel::DoGather(int task_id) {
}
int8_t *int8_out = reinterpret_cast<int8_t *>(out_tensor->data_c());
int data_size = lite::DataTypeSize(kNumberTypeFloat16);
int8_in += thread_stride * limit * data_size;
int8_out += thread_stride * indices_element_size * data_size;
int8_in += thread_stride * limit * inner_size * data_size;
int8_out += thread_stride * indices_element_size * inner_size * data_size;
int error_code = Gather(int8_in, count, inner_size, limit, indices_data_, indices_element_size, int8_out, data_size);
return error_code;
}

4
mindspore/lite/src/runtime/kernel/arm/fp32/gather_fp32.cc
View File

@ -65,8 +65,8 @@ int GatherCPUKernel::DoGather(int task_id) {
int8_t *int8_out = reinterpret_cast<int8_t *>(out_tensor->data_c());
int data_size = lite::DataTypeSize(input_tensor->data_type());
int8_in += thread_stride * limit * data_size;
int8_out += thread_stride * indices_element_size * data_size;
int8_in += thread_stride * limit * inner_size * data_size;
int8_out += thread_stride * indices_element_size * inner_size * data_size;
int error_code = Gather(int8_in, count, inner_size, limit, indices_data_, indices_element_size, int8_out, data_size);

4
mindspore/lite/src/runtime/kernel/arm/int8/gather_int8.cc
View File

@ -82,8 +82,8 @@ int GatherInt8CPUKernel::DoGather(int task_id) {
int count = MSMIN(stride, outer_size - stride * task_id);
auto thread_stride = stride * task_id;
input_ptr += thread_stride * limit;
output_ptr += thread_stride * indices_element_size;
input_ptr += thread_stride * inner_size * limit;
output_ptr += thread_stride * inner_size * indices_element_size;
return GatherInt8(input_ptr, output_ptr, count, inner_size, limit, indices_ptr, indices_element_size, param_);
}