For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
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
-
MHAAttentionConfigRDNA: Holds the fixed RDNA Wave32 multi-head attention configuration and index math.