IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /docs/manual/basics.md). For the complete Mojo documentation index, see llms.txt.
Skip to main content
Version: Nightly
For the complete Mojo documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /docs/manual/basics.md).

st_matrix_n_atom

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.