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_ARM

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

Returns the micro kernel shape tuned for ARM targets.

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