For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/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.