当使用GetTensorC异步接口将结果矩阵从GM拷贝到UB,且UB后续需要进行Vector计算时,需要调用WaitGetTensorC进行同步。
1 | __aicore__ inline void WaitGetTensorC() |
无
无
Atlas A2训练系列产品/Atlas 800I A2推理产品
无
// 异步模式样例
mm.template Iterate<false>();
…… ……
for (int i = 0; i < singleM/baseM*singleN/baseN; ++i) {
mm.GetTensorC<false>(ubCmatrix);
mm.WaitGetTensorC();
... Vector 操作
}