From ab31a9a98623a4262e27618631f31d6190ce8df2 Mon Sep 17 00:00:00 2001
From: yangruoqi713 <yangruoqi@huawei.com>
Date: Mon, 2 Nov 2020 16:57:13 +0800
Subject: [PATCH] [MSLITE][Develop] fix bug of arm32 fp32 cpu op: topk

---
 mindspore/lite/nnacl/fp32/topk.c | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/mindspore/lite/nnacl/fp32/topk.c b/mindspore/lite/nnacl/fp32/topk.c
index 96b3b05e46..a9b48d7295 100644
--- 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;
   }
 }