For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo module
buffers
Q, P, and Output register buffers for gfx950 attention kernels.
TileTensor-only: no LayoutTensor imports, aliases, or return types.
Structs
-
OutputRegisterBuffer: Holds the attention output accumulator tile in register memory. -
PRegisterBuffer: Holds the P attention-score tile across register and shared memory. -
QRegisterBuffer: Holds the Q query tile in register memory for gfx950 attention MMAs.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!