From dec92bf2d703db698d0f208cd566a7c5444671cb Mon Sep 17 00:00:00 2001 From: fuzhiye Date: Fri, 5 Mar 2021 10:13:33 +0800 Subject: [PATCH] fix arithmetic_compare_fp16 op bug --- .../src/runtime/kernel/arm/fp16/arithmetic_compare_fp16.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_compare_fp16.cc b/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_compare_fp16.cc index 278f1b0fe6..2e1fdc100a 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_compare_fp16.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_compare_fp16.cc @@ -21,6 +21,7 @@ #include "schema/model_generated.h" #include "src/kernel_registry.h" #include "src/runtime/runtime_api.h" +#include "src/ops/arithmetic.h" #include "include/errorcode.h" using mindspore::kernel::KERNEL_ARCH::kCPU; @@ -74,6 +75,8 @@ int ArithmeticCompareFP16CPUKernel::Init() { } int ArithmeticCompareFP16CPUKernel::ReSize() { + auto arithmetic_lite_primitive = (lite::Arithmetic *)primitive_; + param_->broadcasting_ = arithmetic_lite_primitive->Broadcasting(); param_->in_elements_num0_ = in_tensors_.at(0)->ElementsNum(); param_->in_elements_num1_ = in_tensors_.at(1)->ElementsNum(); param_->out_elements_num_ = out_tensors_.at(0)->ElementsNum();