Mojo module

block_scaled_smem

Shared memory layout for block-scaled SM100 matmul.

Extends standard SMEM with scaling factor tile storage (SFA, SFB) following MXFP8 layout conventions. Also includes all pipeline barriers and TMEM state.

Structs

• ​BlockScaledSmem: SMEM struct containing A/B tiles, scaling factors, C output, and barriers.

Functions

• ​sf_bk: Compute SF_BK from config.
• ​sf_k_group_size: Compute SF_K_GROUP_SIZE from config.
• ​sfa_dim0: Compute SFA first dimension from config.
• ​sfa_dim1: Compute SFA second dimension from config.
• ​sfb_dim0: Compute SFB first dimension from config.
• ​sfb_dim1: Compute SFB second dimension from config.

Was this page helpful?

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

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit