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

!3875 fix reduce for loop size bug 

Merge pull request !3875 from zhaozhenlong/lite/issue/reduce_for_loop_size_unsign_compare
This commit is contained in:
mindspore-ci-bot 2020-08-03 19:06:49 +08:00 committed by Gitee
parent a182a87a6d aa6be43437
commit 952f96d860
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/lite/src/runtime/kernel/arm/fp32/reduce.cc
View File

@ -71,13 +71,13 @@ int ReduceCPUKernel::CheckParameters() {
return RET_ERROR;
}
for (auto i = 0; i < num_axes_; i++) {
if (static_cast<size_t>(axes_[i]) < -input_rank || static_cast<size_t>(axes_[i]) >= input_rank) {
MS_LOG(ERROR) << "Reduce got invalid axis " << axes_[i] << ", axis should be in [" << -input_rank << ", "
<< input_rank - 1 << "].";
if (axes_[i] < -static_cast<int>(input_rank) || static_cast<size_t>(axes_[i]) >= input_rank) {
MS_LOG(ERROR) << "Reduce got invalid axis " << axes_[i] << ", axis should be in ["
<< -static_cast<int>(input_rank) << ", " << input_rank - 1 << "].";
return RET_ERROR;
}
if (axes_[i] < 0) {
axes_[i] += input_rank;
axes_[i] += static_cast<int>(input_rank);
}
}
@ -155,7 +155,7 @@ int ReduceImpl(int task_id, LiteParallelGroupEnv *penv, void *cdata) {
int ReduceCPUKernel::Run() {
tmp_shape_ = inputs_.at(0)->shape();
src_data_ = static_cast<float *>(inputs_.at(0)->Data());
for (int i = 0; i < tmp_shape_.size(); ++i) {
for (int i = 0; i < data_buffers_.size(); ++i) {
dst_data_ = data_buffers_[i];
int axis = axes_[i];
outer_size_ = 1;