Skip to main content

Mojo function

build_block_scaled_configs

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

Returns:

Set

Was this page helpful?