For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

validate_config

def validate_config[BM: Int, BN: Int, BK: Int, WM: Int, WN: Int, WK: Int, m_warps: Int, n_warps: Int, producer_a: Int, producer_b: Int, consumer: Int]()

Validates the configuration parameters for the matrix multiplication kernel.

Parameters:

• ​BM (Int): Block tile size along the M dimension, must be divisible by WM.
• ​BN (Int): Block tile size along the N dimension, must be divisible by WN.
• ​BK (Int): Block tile size along the K dimension.
• ​WM (Int): Warp tile size along the M dimension.
• ​WN (Int): Warp tile size along the N dimension.
• ​WK (Int): Warp tile size along the K dimension.
• ​m_warps (Int): Number of warps covering the M dimension of the block, must be divisible by producer_a.
• ​n_warps (Int): Number of warps covering the N dimension of the block, must be divisible by producer_b.
• ​producer_a (Int): Number of warps assigned to loading matrix A tiles.
• ​producer_b (Int): Number of warps assigned to loading matrix B tiles.
• ​consumer (Int): Number of warps assigned to computing the matmul, must be a power of two and at least producer_a and producer_b.