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.