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
CPU entry points for block-scaled SM100 matmul.
Creates TMA descriptors for A, B, C and scaling factors (SFA, SFB), then launches the warp-specialized kernel.
Functions
-
blackwell_block_scaled_matmul_tma_umma_warp_specialized: Launch block-scaled FP8 matmul kernel on SM100.