Mojo module
config
Structsβ
- β
MatmulConfig: Static configuration of SM90 GPU matmul.
Functionsβ
- β
build_configs: - β
build_configs_generic: - β
swapAB_largeM_clustered: Config for m in [129, 240] range with cluster=(2,1,1). - β
swapAB_midM_linear: Config for m in [65, 128] range with linear BN pattern. - β
swapAB_smallM: - β
swapAB_smallM_ceildiv: Config for m < 41 range with BN = ceildiv(m, 8) * 8 pattern.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!