Browse Source

[MSLITE][Develop] fix bug of arm fp16 cpu op: reduce, arithmetic

tags/v1.1.0
yangruoqi713 5 years ago
parent
commit
26d1485819
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      mindspore/lite/nnacl/fp16/reduce_fp16.c
  2. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp32/arithmetic.cc

+ 2
- 2
mindspore/lite/nnacl/fp16/reduce_fp16.c View File

@@ -30,11 +30,11 @@ int ReduceMeanFp16(const int outer_size, const int inner_size, const int axis_si
for (k = 0; k < inner_size; k++) {
const float16_t *inner_src = outer_src + k;
float16_t *inner_dst = outer_dst + k;
float16_t tmp = 0.0;
float tmp = 0.0;
for (i = 0; i < axis_size; i++) {
tmp += inner_src[i * inner_size];
}
*inner_dst = tmp / (float16_t)axis_size;
*inner_dst = (float16_t)(tmp / axis_size);
}
}
return NNACL_OK;


+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp32/arithmetic.cc View File

@@ -36,7 +36,7 @@ int ArithmeticCPUKernel::Init() {
if (!InferShapeDone()) {
return RET_OK;
}
if (in_tensors_[0]->data_type() == kNumberTypeFloat32) {
if (in_tensors_[0]->data_type() == kNumberTypeFloat32 || in_tensors_[0]->data_type() == kNumberTypeFloat16) {
data_type_ = kDataTypeFloat;
} else {
data_type_ = kDataTypeInt;


Loading…
Cancel
Save