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

matmul

Provides SM90 warp-specialized GEMM dispatchers with TMA multicasting, split-K reduction, and optional epilogue fusion for Hopper matmuls.

comptime values

logger

comptime logger = Logger(stdout, prefix=String(""), source_location=False)

Functions

• ​warp_specialize_gemm_with_multicasting: Dispatches an SM90 warp-specialized matmul to split-K or regular kernel paths.
• ​warp_specialize_gemm_with_multicasting_splitk: Dispatches an SM90 warp-specialized split-K matmul that partitions the K dimension across thread blocks and reduces partial sums via a workspace buffer and locks.