For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo module
block_scaled_matmul
Structsβ
Functionsβ
- β
blackwell_block_scaled_matmul_tma_umma_warp_specialized: Launch block-scaled FP8 matmul kernel on SM100. - β
blackwell_block_scaled_tma_umma_warp_specialized_kernel: - β
consumer_main_loop: TileTensor-based consumer_main_loop for block-scaled MMA. - β
load_AB_SFA_SFB: