NVIDIA / NVIDIA/TransformerEngine
[Docs] Add datatype × hardware support matrix
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 3.5k
- Forks
- 831
- Avg merge
- 3d 11h
- Merged PRs (30d)
- 65
Description
Is your feature request related to a problem? Please describe.
It's difficult to determine which datatypes and features are supported on which GPU architectures. Information is currently scattered across:
• High-level mentions in README (e.g., "FP8 on Hopper, Ada, and Blackwell")
• Inline code conditionals (e.g., get_device_compute_capability() >= (9, 0))
• Test skip conditions (e.g., pytest.skip("THD is only supported on Hopper+"))
• Error messages at runtime
Describe the solution you'd like
Add a clear Hardware Support Matrix page to the documentation with tables for "Datatype Support by Architecture" and "Feature Support by Architecture" (e.g. THD format).
Describe alternatives you've considered
Automatic capability detection at import - Could print a summary of available features, but doesn't address pre-deployment planning.
Additional context
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
Use the README, get_device_compute_capability() conditionals, pytest skip conditions, and runtime error messages named in the issue as source material; first locate these references and the documentation structure. Done means a Hardware Support Matrix page has separate datatype-by-architecture and feature-by-architecture tables, including examples such as THD.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- documentation
- Issue type
- Documentation
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 45/100