已合并
修复GatherElementsV2非尾轴场景同步错误 #2828
huang-wei-chen创建于 3月17日
修复GatherElementsV2非尾轴场景同步错误 #2828
已合并
共 2 个文件变更+8-0
| @@ -128,6 +128,13 @@ protected: | |||
| 128 | WaitFlag<HardEvent::V_MTE2>(eventIDVToMTE2); | 128 | WaitFlag<HardEvent::V_MTE2>(eventIDVToMTE2); |
| 129 | } | 129 | } |
| 130 | 130 | ||
| 131 | + __aicore__ inline void MTE3ToVSync() | ||
| 132 | + { | ||
| 133 | + event_t eventIDMTE3ToV = static_cast<event_t>(GetTPipePtr()->FetchEventID(HardEvent::MTE3_V)); | ||
| 134 | + SetFlag<HardEvent::MTE3_V>(eventIDMTE3ToV); | ||
| 135 | + WaitFlag<HardEvent::MTE3_V>(eventIDMTE3ToV); | ||
| 136 | + } | ||
| 137 | + | ||
| 131 | template <typename T> | 138 | template <typename T> |
| 132 | __aicore__ inline T Min(T a, T b) | 139 | __aicore__ inline T Min(T a, T b) |
| 133 | { | 140 | { |
| @@ -256,6 +256,7 @@ private: | |||
| 256 | GatherInUb(this->xGatherDim_, gatherParam_.idxGatherDimSlice); | 256 | GatherInUb(this->xGatherDim_, gatherParam_.idxGatherDimSlice); |
| 257 | this->VToMTE2Sync(); | 257 | this->VToMTE2Sync(); |
| 258 | CopyOutY2Workspace(idxWorkspaceOffset, gatherParam_.idxGatherDimSlice); | 258 | CopyOutY2Workspace(idxWorkspaceOffset, gatherParam_.idxGatherDimSlice); |
| 259 | + this->MTE3ToVSync(); | ||
| 259 | } | 260 | } |
| 260 | // tail | 261 | // tail |
| 261 | if (gatherParam_.idxGatherDimTail) { | 262 | if (gatherParam_.idxGatherDimTail) { |