Mojo function
get_rdna_warp_layout
get_rdna_warp_layout() -> Layout
Get the warp thread layout for RDNA WMMA operations.
For RDNA Wave32 with 16x16 tiles, using col_major(16, 2):
- Lane i: row = i % 16, col = i // 16
- Groups lanes (0, 16), (1, 17), ..., (15, 31) for cross-group reduction
- Matches C/D mapping where lanes 0-15 hold even rows, 16-31 hold odd rows.
Returns:
Layout: Layout for col_major(16, 2) thread distribution.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!