m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source 
From: @yangruoqi713
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
pull/13751/MERGE
mindspore-ci-bot 5 years ago 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

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

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

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

@ -65,8 +65,8 @@ int GatherCPUKernel::DoGather(int task_id) {
int8_t *int8_out = reinterpret_cast<int8_t *>(out_tensor->data_c()); int8_t *int8_out = reinterpret_cast<int8_t *>(out_tensor->data_c());
int data_size = lite::DataTypeSize(input_tensor->data_type()); int data_size = lite::DataTypeSize(input_tensor->data_type());
int8_in += thread_stride * limit * data_size; int8_in += thread_stride * limit * inner_size * data_size;
int8_out += thread_stride * indices_element_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); 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
Unescape View File

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

Write Preview
Loading…
Cancel
Save