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

get_matmul_kernel_shape

def get_matmul_kernel_shape[a_type: DType, b_type: DType, c_type: DType, kernel_type: Bool]() -> MicroKernelShape

Returns the micro kernel shape for the current target and dtypes.

Dispatches to the ARM or x86 variant based on the compilation target.

Parameters:

a_type (DType): The dtype of the A tensor.
b_type (DType): The dtype of the B tensor.
c_type (DType): The dtype of the C tensor.
kernel_type (Bool): Selects between the two tuned shapes for the target.

Returns:

MicroKernelShape