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