From e0cbe113745a38be7b3afa0dff63a819e4490005 Mon Sep 17 00:00:00 2001
From: lzk <liuzhongkai2@huawei.com>
Date: Fri, 21 May 2021 01:11:07 -0700
Subject: [PATCH] div 0 bug fix

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

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 044fd170e1..f1e45d47e0 100644
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc
@@ -117,6 +117,10 @@ int ReduceCPUKernel::Run() {
     outer_size_ = outer_sizes_.at(i);
     inner_size_ = inner_sizes_.at(i);
     axis_size_ = axis_sizes_.at(i);
+    if (axis_size_ == 0) {
+      MS_LOG(ERROR) << "axis_size_ is must not be zero!";
+      return RET_ERROR;
+    }
     auto error_code = ParallelLaunch(static_cast<const lite::InnerContext *>(this->context_)->thread_pool_, ReduceImpl,
                                      this, context_->thread_num_);
     if (error_code != RET_OK) {
-- 
2.17.1

