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!