Merged
topk sync bugfix #6352
DianaBlessYouAdd on Sep 23
topk sync bugfix #6352
Merged
Merged fromlijunjie147/asc-devkit_9351topk-exampleintocann/asc-devkitmaster
Total: 1 files changed+3-0
| @@ -127,6 +127,9 @@ __aicore__ inline void TopKCompute( | |||
| 127 | GatherDstValAndDstIdx(dstValueLocal, dstIndexLocal, tmpLocal, tiling, dstOffsetFourBytes, j); | 127 | GatherDstValAndDstIdx(dstValueLocal, dstIndexLocal, tmpLocal, tiling, dstOffsetFourBytes, j); |
| 128 | 128 | ||
| 129 | if constexpr (isHasfinish) { | 129 | if constexpr (isHasfinish) { |
| 130 | + auto eventIDMte2ToS = GetTPipePtr()->FetchEventID(HardEvent::MTE2_S); | ||
| 131 | + SetFlag<HardEvent::MTE2_S>(eventIDMte2ToS); | ||
| 132 | + WaitFlag<HardEvent::MTE2_S>(eventIDMte2ToS); | ||
| 130 | bool finishValue = finishLocal.GetValue(j); | 133 | bool finishValue = finishLocal.GetValue(j); |
| 131 | auto eventID = GetTPipePtr()->FetchEventID(HardEvent::S_V); | 134 | auto eventID = GetTPipePtr()->FetchEventID(HardEvent::S_V); |
| 132 | SetFlag<HardEvent::S_V>(eventID); | 135 | SetFlag<HardEvent::S_V>(eventID); |