1be168c0dSopenharmony_ciFrom 2d816538b2663a21240a3bff38a876e75e2d7e2d Mon Sep 17 00:00:00 2001 2be168c0dSopenharmony_ciFrom: rabbit-fgh <2955722401@qq.com> 3be168c0dSopenharmony_ciDate: Thu, 29 Aug 2024 19:47:59 +0800 4be168c0dSopenharmony_ciSubject: [PATCH] fix ArgMaxFusion 5be168c0dSopenharmony_ci 6be168c0dSopenharmony_ci--- 7be168c0dSopenharmony_ci mindspore/lite/src/litert/kernel/cpu/int8/argminmax_int8.cc | 1 + 8be168c0dSopenharmony_ci 1 file changed, 1 insertion(+) 9be168c0dSopenharmony_ci 10be168c0dSopenharmony_cidiff --git a/mindspore/lite/src/litert/kernel/cpu/int8/argminmax_int8.cc b/mindspore/lite/src/litert/kernel/cpu/int8/argminmax_int8.cc 11be168c0dSopenharmony_ciindex 7cb872d9..ecf7920a 100644 12be168c0dSopenharmony_ci--- a/mindspore/lite/src/litert/kernel/cpu/int8/argminmax_int8.cc 13be168c0dSopenharmony_ci+++ b/mindspore/lite/src/litert/kernel/cpu/int8/argminmax_int8.cc 14be168c0dSopenharmony_ci@@ -69,6 +69,7 @@ int ArgMinMaxInt8CPUKernel::Prepare() { 15be168c0dSopenharmony_ci compute_param_->topk_ = param->topk_; 16be168c0dSopenharmony_ci compute_param_->out_value_ = param->out_value_; 17be168c0dSopenharmony_ci compute_param_->keep_dims_ = param->keep_dims_; 18be168c0dSopenharmony_ci+ compute_param_->get_max_ = (param->op_parameter_.type_ == PrimType_ArgMinFusion ? false : true); 19be168c0dSopenharmony_ci 20be168c0dSopenharmony_ci out_quant_arg_ = reinterpret_cast<QuantArg *>(malloc(sizeof(QuantArg))); 21be168c0dSopenharmony_ci if (out_quant_arg_ == nullptr) { 22be168c0dSopenharmony_ci-- 23be168c0dSopenharmony_ci2.45.1.windows.1 24be168c0dSopenharmony_ci 25