Mojo package
amd
AMD GPU attention operations.
Modules
-
attention: -
attention_rdna: RDNA-specific Attention struct for Wave32 WMMA operations. -
buffers: -
buffers_rdna: RDNA-specific buffer implementations for Wave32 WMMA attention. -
mha_gfx942: -
mha_gfx950: -
mha_rdna: RDNA-specific MHA kernel configurations and entry points. -
mla: -
mma: -
mma_rdna: RDNA-specific MMA wrapper for Wave32 WMMA operations. -
softmax: -
utils:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!