[MSLITE] fix arithmetic 'scalar_opt_' status not reset bug

2023-01-03 14:52:06 +08:00 · 2023-01-03 14:52:06 +08:00 · dbe4a6c89e
parent c4b52ab8af
commit dbe4a6c89e
1 changed files with 1 additions and 0 deletions
--- a/mindspore/lite/src/runtime/kernel/cpu/base/arithmetic_base.cc
+++ b/mindspore/lite/src/runtime/kernel/cpu/base/arithmetic_base.cc
@ -295,6 +295,7 @@ void ArithmeticBaseCPUKernel::ComputeOfflineInfo() {
      c_matric_.batch_post_sum[i] = c_matric_.shape[i] * c_matric_.batch_post_sum[i + 1];
    }
  }
+  scalar_opt_ = false;
  if (a_matric_.inner_size == 1) {
    param_->in_elements_num0_ = 1;
    scalar_opt_ = true;