[FEA] Block-Scaled Data Format Support for SM120 in CuTe DSL
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 10.5k
- Forks
- 2.1k
- Avg merge
- 3d 11h
- Merged PRs (30d)
- 7
Description
Which component requires the feature?
CuTe DSL
Feature Request
Is your feature request related to a problem? Please describe.
I wish I could use CuTe DSL (Python) to implement block-scaled GEMM operations on SM120 (Blackwell) architecture. Currently, all CuTe DSL Python examples for block-scaled formats (MXFP4, MXFP6, MXFP8) only support SM100.
Describe the solution you'd like
Add CuTe DSL Python support for SM120 block-scaled operations:
- Python helper functions for SM120 (similar to existing SM100 helpers in blackwell_helpers.py)
- SM120 MMA operation wrappers for block-scaled formats (MmaMXF4Op, MmaMXF8Op, etc.)
- Example scripts for dense and grouped block-scaled GEMM on SM120
- Support for MXFP4 (Float4E2M1FN), MXFP6, and MXFP8 formats with corresponding scale factor types
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reading the existing SM100 helpers in blackwell_helpers.py and the current CuTe DSL Python examples for block-scaled formats. Trace how their MMA wrappers and dense or grouped GEMM examples are organized before mapping the requested SM120 variants. Done means the listed SM120 formats, scale types, wrappers, and example scripts are supported.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp, python
- Domain
- ai-infra-agents, hpc
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Active
- Clarity
- Mostly clear
- Newbie friendliness
- 38/100