Mojo module
config
RDNA Wave32 attention config.
Single fixed config for both prefill and decode: MMA shape 16x16x16 (the only RDNA WMMA shape), no shared K/V SMEM, BK-strip K SMEM, single-buffered, depth-padded V SMEM.
comptime values
RDNA_MMA_K
comptime RDNA_MMA_K = 16
RDNA_MMA_M
comptime RDNA_MMA_M = 16
RDNA_MMA_N
comptime RDNA_MMA_N = 16
Structs
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!