[MSLITE][Develop] fix bug of arm32 fp32 cpu op: topk

4 years ago · ab31a9a986
parent 2124ab1a8c
commit ab31a9a986
1 changed files with 8 additions and 2 deletions
--- a/mindspore/lite/nnacl/fp32/topk.c
+++ b/mindspore/lite/nnacl/fp32/topk.c
@ -22,8 +22,11 @@ int DescendCmp(const void *a, const void *b) {
    return 1;
  } else if (sub < 0) {
    return -1;
+  }
+  if (((const TopkNode *)a)->index > ((const TopkNode *)b)->index) {
+    return 1;
  } else {
-    return 0;
+    return -1;
  }
 }

@ -33,8 +36,11 @@ int AscendCmp(const void *a, const void *b) {
    return 1;
  } else if (sub < 0) {
    return -1;
+  }
+  if (((const TopkNode *)a)->index > ((const TopkNode *)b)->index) {
+    return -1;
  } else {
-    return 0;
+    return 1;
  }
 }