From e0cbe113745a38be7b3afa0dff63a819e4490005 Mon Sep 17 00:00:00 2001 From: lzk Date: Fri, 21 May 2021 01:11:07 -0700 Subject: [PATCH] div 0 bug fix --- mindspore/lite/src/runtime/kernel/arm/base/split_base.cc | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/mindspore/lite/src/runtime/kernel/arm/base/split_base.cc b/mindspore/lite/src/runtime/kernel/arm/base/split_base.cc index 2448a407dd..b96e5aacb8 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/split_base.cc +++ b/mindspore/lite/src/runtime/kernel/arm/base/split_base.cc @@ -50,6 +50,15 @@ int SplitBaseCPUKernel::ReSize() { param->strides_[i] = param->strides_[i + 1] * input_shape.at(i + 1); } + if (input_shape.at(param->split_dim_) == 0) { + MS_LOG(ERROR) << "input_shape[" << param->split_dim_ << "] must not be zero!"; + return RET_ERROR; + } + if (param->strides_[param->split_dim_] == 0) { + MS_LOG(ERROR) << "param->strides_[" << param->split_dim_ << "] must not be zero!"; + return RET_ERROR; + } + MS_ASSERT(static_cast(param->split_dim_) < input_shape.size()); param->split_count_ = param->strides_[0] * input_shape.at(0) / (input_shape.at(param->split_dim_) * param->strides_[param->split_dim_]); -- 2.17.1