[Feature] Support nvfp4 tensor core on B200
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 492
- Forks
- 30
- Avg merge
- 4d 9h
- Merged PRs (30d)
- 2
Description
Currently, we only support float16 and float8 tensor core on B200 (the tcgen05 instruction set). We need to support nvfp4 tensor core for the self.tcgen05.mma instruction, and add a kernel example.
Contributor guide
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
Locate the existing B200 support for float16 and float8 in the self.tcgen05.mma implementation, then review how its kernel examples are structured. Add nvfp4 support for that instruction and a corresponding kernel example; done means the new example demonstrates the supported path.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- hpc
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 48/100