WaitGetTensorC
功能说明
当使用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 操作 }
父主题: Matmul