Mojo function
tile_sf_layout_k_major
tile_sf_layout_k_major[BM: Int, BK: Int, SF_SCALE_SIZE: Int]() -> Layout
Creates a K-major layout for tensor core scale factors.
Constructs a layout for K-major access patterns for scale factors.
Parameters:
- BM (
Int): Size of the M dimension in the tile. - BK (
Int): Size of the K dimension in the tile. - SF_SCALE_SIZE (
Int): Number of elements in a scale factor vector.
Returns:
Layout: Layout - A K-major layout configured for the specified dimensions and scale factor size.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!