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 struct

MiniIterSpec

struct MiniIterSpec

One mini-iter of the 4-wave body: (DRAM prefetch, frag-load, MMA).

Fields

• ​load_tag (Int): LOAD_A or LOAD_B — channel-A vs channel-B prefetch.
• ​load_channel (Int): 0 (A) or 1 (B). Redundant with load_tag but needed by the framework's edge derivation.
• ​load_subtile (Int): Which sub-tile of the source SMEM half this prefetch writes (0 or 1).
• ​frag_tag (Int): MMA_LOAD_A or MMA_LOAD_B — register frag-load.
• ​frag_channel (Int): 0 (A) or 1 (B) for the frag-load.
• ​frag_subtile (Int): Which sub-tile of the SMEM stage to read into the fragment register.
• ​frag_cross_stage (Bool): True iff the frag-load reads from the cross K-partition's stage (cross-stage rotation pre-loads the next partition's leading quadrants while this partition's MMAs are still issuing). False = read from the same stage as the prefetches in this partition.
• ​mma_m_quad (Int): Which m_quad of the warp's 2x2 quadrant grid this MMA computes.
• ​mma_n_quad (Int): Which n_quad of the warp's 2x2 quadrant grid this MMA computes.

Implemented traits

AnyType, Copyable, ImplicitlyDeletable, Movable

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit