For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo module
tile_scheduler_splitk
Provides the split-K tile scheduler for GPU matmul kernels on SM90.
Partitions the K dimension across thread blocks and reduces partial accumulations with semaphore-ordered or atomic strategies for deterministic or nondeterministic numeric behavior.
Structs
-
ReductionMode: Reduction strategy for split-K partial accumulations. -
SplitKTileScheduler: Tile scheduler for split-K GPU matmul reductions on SM90.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!