Mojo module
block_scaled_tile_loader
TMA loaders for 5D MXFP8 scaling factor tensors.
- ScalingFactorLoader: Async TMA loads for SFA/SFB tiles
- copy_sf_tmem: SMEM → TMEM transfer before MMA operations
Structs
-
ScalingFactorLoader: TMA loader for 5D MXFP8 scaling factor tensors (SFA or SFB).
Functions
-
copy_sf_tmem: Copy scaling factors from shared memory to tensor memory.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!