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

config

SM100 configuration for fused GEMM+SwiGLU.

Provides FusedSwiGLUMatmulConfig, a self-contained compile-time config for the SM100 fused GEMM+SwiGLU kernel.

Utilities:

• swiglu_extra_fixed_smem: computes the fixed SMEM bytes consumed by the double-buffered half-output tiles on the SMEM→TMA path.
• swiglu_matmul_config: convenience factory for FusedSwiGLUMatmulConfig with automatic pipeline-stage calculation.
• build_sm100_matmul_configs: returns FusedSwiGLUMatmulConfigs for tuned (N, K) shapes from the SwiGLU tuning table. Returns an empty set for untuned shapes; dispatch falls back to its safety-net config in that case.

Structs

• ​FusedSwiGLUMatmulConfig: Compile-time configuration for the SM100 fused GEMM+SwiGLU kernel.

Functions

• ​build_sm100_matmul_configs: Build FusedSwiGLUMatmulConfigs for SM100 fused GEMM+SwiGLU.
• ​build_sm100_swiglu_heuristic_configs: Build all FusedSwiGLUMatmulConfigs reachable by the heuristic.
• ​choose_swiglu_config: Select a FusedSwiGLUMatmulConfig using a wave-minimization heuristic.
• ​swiglu_extra_fixed_smem: Compute extra fixed SMEM bytes for the SwiGLU SMEM-path half tiles.
• ​swiglu_matmul_config: Create a FusedSwiGLUMatmulConfig with auto-computed pipeline stages.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit