Mojo function

get_sfb_num_cols

get_sfb_num_cols[config: BlockScaledMatmulConfig[a_type, b_type, c_type, sfa_dtype, sfb_dtype, transpose_b]]() -> Int

Get the number of TMEM columns needed for B scaling factors.

Returns:

Int

View source

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!