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

fix prelu memcpy bug

This commit is contained in:
jianghui58 2021-02-03 16:25:01 +08:00
parent 097431af6d
commit 3a1647a808
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/lite/src/runtime/kernel/arm/fp32/prelu_fp32.cc
View File

@ -70,7 +70,7 @@ int PReluCPUKernel::ProcessInput() {
MS_LOG(ERROR) << "malloc input_data_ failed.";
return RET_ERROR;
}
memcpy(input_data_, ori_input_, tile_block * TILE_NUM * channel_num * sizeof(float));
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
return RET_OK;
}
@ -85,7 +85,7 @@ int PReluCPUKernel::ProcessShareChannelInput() {
MS_LOG(ERROR) << "malloc input_data_ failed.";
return RET_ERROR;
}
memcpy(input_data_, ori_input_, prelu_param_->tile_block_ * 64 * sizeof(float));
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
#elif ENABLE_ARM32
prelu_param_->tile_block_ = UP_DIV(prelu_param_->input_num_, 32);
input_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(prelu_param_->tile_block_ * 32 * sizeof(float)));
@ -93,7 +93,7 @@ int PReluCPUKernel::ProcessShareChannelInput() {
MS_LOG(ERROR) << "malloc input_data_ failed.";
return RET_ERROR;
}
memcpy(input_data_, ori_input_, prelu_param_->tile_block_ * 32 * sizeof(float));
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
#else
prelu_param_->tile_block_ = UP_DIV(prelu_param_->input_num_, 32);
input_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(prelu_param_->tile_block_ * 32 * sizeof(float)));
@ -101,7 +101,7 @@ int PReluCPUKernel::ProcessShareChannelInput() {
MS_LOG(ERROR) << "malloc input_data_ failed.";
return RET_ERROR;
}
memcpy(input_data_, ori_input_, prelu_param_->tile_block_ * 32 * sizeof(float));
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
#endif
return RET_OK;
}