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

Merge pull request !8123 from yangruoqi713/lite
pull/8123/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit fa859cf1a5

@ -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;
}
}

Loading…
Cancel
Save