Mojo module

config

SM100 matmul configuration types and utilities.

This module provides configuration structs for SM100 (Blackwell) GPU matmul operations, including standard matmul and block-scaled matmul variants.

Two config types share 17 common fields (tile shapes, pipeline stages, swizzle modes, etc.) and the same __init__ helpers (_compute_block_tile_shape, _compute_output_tile_shape, _compute_swizzle_modes, _maximize_pipeline_stages). BlockScaledMatmulConfig extends this with 3 scaling-specific fields (scaling_kind, vec_sf_size, num_sf_k_tiles). Each has its own heuristic (choose_config / choose_block_scaled_config) because the valid MMA shape ranges and alignment requirements differ between standard and scaled kernels.

Structs

• ​BlockScaledMatmulConfig: Static configuration of GPU matmul.
• ​MatmulConfig: Static configuration of GPU matmul.
• ​OutputPipelineConfig: Configuration for the MMA-to-Epilogue output pipeline.

Functions

• ​build_block_scaled_configs:
• ​build_configs:
• ​choose_block_scaled_config:
• ​choose_config:
• ​default_matmul_config_bf16_fp8:

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