已合并
修复topkv2算子的oom问题 #2857
cy_hw创建于 5月20日
修复topkv2算子的oom问题 #2857
已合并
共 1 个文件变更+1-1
| @@ -658,7 +658,7 @@ __aicore__ inline void RadixSortTopK<T, UNSIGNED_TYPE, NUM_PASS, IS_LARGEST, IS_ | |||
| 658 | } | 658 | } |
| 659 | PipeBarrier<PIPE_ALL>(); | 659 | PipeBarrier<PIPE_ALL>(); |
| 660 | SyncAll(); | 660 | SyncAll(); |
| 661 | - if (inBlockDimRange_) { | 661 | + if (inBlockDimRange_ && unsortedDimIndex < unsortedDimNum_) { |
| 662 | if (IS_SORT && (startTileId == 0) && !needSortWithIndex_) { | 662 | if (IS_SORT && (startTileId == 0) && !needSortWithIndex_) { |
| 663 | if (NUM_PASS != topkV2::B64_BITE_SIZE) { | 663 | if (NUM_PASS != topkV2::B64_BITE_SIZE) { |
| 664 | SortTopKRes(xLocal); | 664 | SortTopKRes(xLocal); |