文档
注册
评分
提单
论坛
小AI

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 操作 
}
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词