WaitIterateAll

Applicability

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	x
Atlas inference product's AI Core	x
Atlas inference product's Vector Core	x
Atlas training products	x

Function

Waits for the return of the asynchronous IterateAll API and continuously outputs data to the Global Memory.

Prototype

__aicore__ inline void WaitIterateAll()

Parameters

None

Returns

None

Restrictions

This API is used together with the asynchronous IterateAll API.
Data can only be continuously output to the Global Memory.

Example

For more examples of using this API, see matrix multiplication in IterateAll asynchronous scenarios.

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(); // Wait for IterateAll to complete.

Parent topic: Matmul Kernel APIs