Skip to main content

Mojo function

choose_block_scaled_config

choose_block_scaled_config[a_type: DType, b_type: DType, c_type: DType, sfa_dtype: DType, sfb_dtype: DType, transpose_b: Bool = True](M: Int, N: Int, K: Int) -> BlockScaledMatmulConfig[a_type, b_type, c_type, sfa_dtype, sfb_dtype, transpose_b]

Returns:

BlockScaledMatmulConfig

Was this page helpful?