Script.wgmma.wait_group

Script.wgmma.wait_group

Script.wgmma.wait_group(n)[source]

Wait for warp group MMA commit groups to complete.

Waits until at most n commit groups are pending (i.e., all but the most recent n groups have completed).

Parameters:

n (Expr | int) – The number of commit groups allowed to remain pending. Use 0 to wait for all committed groups to complete.

Return type:

None

Notes

  • Thread group: Must be executed by a warp group (4 warps).

  • Hardware: Requires compute capability 9.0a+ (sm_90a).

  • PTX: wgmma.wait_group.sync.aligned

Contents