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_dispatch_table

def get_dispatch_table[a_type: DType, b_type: DType, c_type: DType, transpose_b: Bool]() -> Dict[String, MatmulConfig[a_type, b_type, c_type, transpose_b], Fnv1a]

Returns the dispatch table of named Ampere matmul tile configurations for the given dtypes and transpose setting.

Parameters:

a_type (DType): Element type of the left-hand input matrix.
b_type (DType): Element type of the right-hand input matrix.
c_type (DType): Element type of the output matrix.
transpose_b (Bool): Whether the right-hand input matrix is transposed.

Returns:

Dict[String, MatmulConfig[a_type, b_type, c_type, transpose_b], Fnv1a]