From 14a30e2547f3939c8aa15593e51a8c0e00a6e42c Mon Sep 17 00:00:00 2001 From: liuf9 Date: Thu, 25 Sep 2025 16:44:35 +0800 Subject: [PATCH] bugfix for nnacl ops --- mindspore-lite/providers/nnie/src/nnie_cfg_parser.cc | 4 ++-- .../litert/kernel/cpu/nnacl_c/fp32/non_max_suppression_fp32.c | 2 +- .../cpu/nnacl_c/infer/control/tensorlist_reserve_infer.c | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore-lite/providers/nnie/src/nnie_cfg_parser.cc b/mindspore-lite/providers/nnie/src/nnie_cfg_parser.cc index d51744ff..c88a1312 100644 --- a/mindspore-lite/providers/nnie/src/nnie_cfg_parser.cc +++ b/mindspore-lite/providers/nnie/src/nnie_cfg_parser.cc @@ -65,9 +65,9 @@ int Flags::ParserBool(const std::map &nnie_arg, const auto iter = nnie_arg.find(key); if (iter != nnie_arg.end()) { auto str = iter->second; - if (str.find("on") != std::string::npos) { + if (str == "on") { *val = true; - } else if (str.find("off") != std::string::npos) { + } else if (str == "off") { *val = false; } else { PrintInvalidChar(key, str); diff --git a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/non_max_suppression_fp32.c b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/non_max_suppression_fp32.c index 63ed8ab3..0d854cd9 100644 --- a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/non_max_suppression_fp32.c +++ b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/non_max_suppression_fp32.c @@ -148,7 +148,7 @@ int NonMaxSuppressionSelecte(NonMaxSuppressionStruct *nm_suppression, bool simpl SortCandidates(env, sorted_candidates, above_score_candidates, above_score_candidates_size); int selected_box_per_class_size = 0; - while (sorted_candidates_size <= 0 && selected_index_size < nm_suppression->max_output_per_class_) { + while (sorted_candidates_size >= 0 && selected_index_size < nm_suppression->max_output_per_class_) { NMSBox *cand = sorted_candidates[sorted_candidates_size - 1]; bool selected = true; for (int k = 0; k < selected_box_per_class_size; k++) { diff --git a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/infer/control/tensorlist_reserve_infer.c b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/infer/control/tensorlist_reserve_infer.c index 544041a7..34ec410c 100644 --- a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/infer/control/tensorlist_reserve_infer.c +++ b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/infer/control/tensorlist_reserve_infer.c @@ -46,7 +46,7 @@ int TensorListReserveInferShape(const TensorC *const *inputs, size_t inputs_size const TensorC *input1 = inputs[1]; int num_ele_type = input1->data_type_; - if (num_ele_type != kNumberTypeInt && ele_shape_type != kNumberTypeInt32) { + if (num_ele_type != kNumberTypeInt && num_ele_type != kNumberTypeInt32) { return NNACL_ERR; } if (input1->data_ == NULL) { -- Gitee