产品 |
是否支持 |
|---|---|
Ascend 950PR/Ascend 950DT |
√ |
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
等待IterateAll异步接口返回,支持连续输出到Global Memory。
1 | __aicore__ inline void WaitIterateAll() |
无
无
详细的使用样例请参考IterateAll异步场景样例。
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[offsetB]); 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完成 |