From 8eafbc8fb9f86e6e1975bc611ac4bc4367cc1312 Mon Sep 17 00:00:00 2001 From: zhaodezan Date: Wed, 12 Aug 2020 05:52:18 -0400 Subject: [PATCH] using qsort function correctly --- .../lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc b/mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc index b10ba90a8c..ff34c184a3 100644 --- a/mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc +++ b/mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/arg_min_max.cc @@ -23,8 +23,11 @@ int ArgCompareAscFp32(const void *a, const void *b) { } int ArgCompareDescFp32(const void *a, const void *b) { - return reinterpret_cast(b)->data_.f_data_ - - reinterpret_cast(a)->data_.f_data_; + // cmp funtion of qsort must return int type + auto b_value = reinterpret_cast(b)->data_.f_data_; + auto a_value = reinterpret_cast(a)->data_.f_data_; + int res = b_value > a_value ? 1 : -1; + return res; } void ArgMaxDim0OutValue(const float *input, float *output, const int *in_shape, ArgMinMaxParameter *param) {