WaitIterateAll
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
等待IterateAll异步接口返回,支持连续输出到Global Memory。
函数原型
1 | __aicore__ inline void WaitIterateAll() |
参数说明
无
返回值说明
无
约束说明
- 配套IterateAll异步接口使用。
- 仅支持连续输出至Global Memory。
调用示例
1 2 3 4 5 6 7 8 9 | AscendC::Matmul<aType, bType, cType, biasType> mm; mm.SetTensorA(gm_a[offsetA]); mm.SetTensorB(gm_b[offsetA]); if (tiling.isBias) { mm.SetBias(gm_bias[offsetBias]); } mm.template IterateAll<false>(gm_c[offsetC], 0, false, true); // do some others compute mm.WaitIterateAll(); // 等待IterateAll完成 |
父主题: Matmul