Mojo package
amd_rdna
AMD RDNA GPU attention kernels for Wave32 WMMA operations.
Includes MHA prefill/decode, MMA primitives, and buffer management for RDNA consumer GPUs (Radeon RX 7000/8000 series).
Modules
-
attention_rdna: RDNA-specific Attention struct for Wave32 WMMA operations. -
buffers_rdna: RDNA-specific buffer implementations for Wave32 WMMA attention. -
mha_rdna: RDNA-specific MHA kernel configurations and entry points. -
mma_rdna: RDNA-specific MMA wrapper for Wave32 WMMA operations.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!