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
create_matmul_configs_ampere
def create_matmul_configs_ampere[key: String, a_type: DType, b_type: DType, c_type: DType, transpose_b: Bool]() -> MatmulConfig[a_type, b_type, c_type, transpose_b]
Returns the Ampere matmul config registered under the given key, falling back to a default config when the key is absent.
Parameters:
- key (
String): Tile configuration key to look up in the dispatch table. - 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:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!