From 0b1e7aebcd18962351e3d9640f6778589b0cd03b Mon Sep 17 00:00:00 2001
From: fuzhiye <fuzhiye@huawei.com>
Date: Tue, 23 Mar 2021 10:05:40 +0800
Subject: [PATCH] fix bug of reduce

---
 mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc
index d0f61a8098..c221536bfa 100644
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc
@@ -125,7 +125,7 @@ int ReduceCPUKernel::Run() {
     }
     src_data_ = dst_data_;
   }
-  if (reduce_param_->reduce_to_end_ && abs(reduce_param_->coeff - 1.0f) > 1e-5) {
+  if (reduce_param_->reduce_to_end_ && abs(reduce_param_->coeff) > 1e-5) {
     ret = CalculateCoeffOutput();
     if (ret != RET_OK) {
       FreeTmpBuffer();