Mojo function
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 - A layout that maps warp group local thread index to an offset
for a logical array with an element size of 128-bit.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!