Mojo function

tile_layout_k_major

tile_layout_k_major[dtype: DType, BM: Int, BK: Int, swizzle_mode: TensorMapSwizzle = TensorMapSwizzle.SWIZZLE_NONE]() -> Layout

Creates a K-major layout for tensor core operations.

Constructs a layout optimized for K-major access patterns in tensor core operations, with optional swizzling for improved memory access patterns.

Parameters:

• ​dtype (DType): Element data type of the tensor.
• ​BM (Int): Size of the M dimension in the tile.
• ​BK (Int): Size of the K dimension in the tile.
• ​swizzle_mode (TensorMapSwizzle): Memory access pattern swizzling mode (default: SWIZZLE_NONE).

Returns:

Layout: Layout - A K-major layout configured for the specified dimensions and swizzle mode.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit