Skip to main content
Version: 1.0

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.