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 function

a_frag_coord

def a_frag_coord(lane: Int, i: Int) -> IndexList[Int(2)]

(row, col) in the 16x16 A tile for this lane's A element i (0..7).

Args:

lane (Int): Simdgroup lane index in [0, 32); determines the lane's base (row, col) in the 16x16 A tile.
i (Int): A-fragment element index in [0, 8); selects which of this lane's 8 held A elements to map to (row, col).

Returns:

IndexList[Int(2)]