For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

st_matrix_n_atom

def st_matrix_n_atom[num_stmatrix: Int]() -> Layout

Creates a layout for N-major st_matrix atom in the context of WGMMA C matrix.

The domain of this layout is the warp group local thread index. Thus, the layout takes [0, 128) as input and returns an offset for a logical array with an element size of 128-bit.

Parameters:

• ​num_stmatrix (Int): Number of N-dimension tiles in the C matrix.

Returns:

Layout: Layout - A layout that maps warp group local thread index to an offset for a logical array with an element size of 128-bit.