Skip to content

[Docs] Add compute capability 11.0 to the support matrix - #3707

Draft
0z5a wants to merge 1 commit into
NVIDIA:mainfrom
0z5a:docs-support-matrix-cc110
Draft

0z5a wants to merge 1 commit into
NVIDIA:mainfrom
0z5a:docs-support-matrix-cc110

Conversation

@0z5a

@0z5a 0z5a commented Oct 17, 2026

Copy link
Copy Markdown
Contributor

Description

docs/support_matrix.rst does not list compute capability 11.0 (Jetson AGX Thor). Footnote 3 also says MXFP8 is limited to 10.x, but _compute_mxfp8_support() in transformer_engine/pytorch/quantization.py enables it from 10.0 up to, but not including, 12.0, which includes 11.0.

This follows up on #2556, where I posted a full L0 run on Jetson AGX Thor: #2556 (comment)

Depends on #3706. The 11.0 NVFP4 cell says "Yes" because stochastic rounding in the RHT cast-fusion kernels works on 11.0 only with that fix. Without it, the default NVFP4 recipe returns all-zero wgrad on 11.0. If #3706 does not land, that cell should be "Partial" with a footnote like the one for 12.x.

Type of change

  • Documentation change (change only to the documentation, either a fix or a new content)
  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • Infra/Build change
  • Code refactoring

Changes

  • Compute capability table: add 11.0, Blackwell (Jetson), Jetson AGX Thor.
  • Format table: add 11.0 with BF16, FP8 (per tensor), FP8 block scaling, MXFP8 and NVFP4 all "Yes".
  • Footnote 3: MXFP8 is supported on 10.0 through 11.x.
  • Footnote 4: the full NVFP4 recipe is supported on 10.x and 11.0.
  • Default recipe table: 11.0 uses MXFP8BlockScaling.

Basis

Jetson AGX Thor, JetPack 7.2, CUDA 13.2, cuDNN 9.20, TE built from source with NVTE_CUDA_ARCHS=110.

I did not build the Sphinx docs. The change adds table rows in the existing format and edits two footnotes.

Checklist:

  • I have read and followed the contributing guidelines
  • The functionality is complete
  • I have commented my code, particularly in hard-to-understand areas (n/a)
  • I have made corresponding changes to the documentation
  • My changes generate no new warnings
  • I have added tests that prove my fix is effective or that my feature works (n/a, docs only)
  • New and existing unit tests pass locally with my changes (n/a, docs only)

🤖 Generated with Claude Code

List CC 11.0 (Jetson AGX Thor) in the compute capability and format
tables. The capability checks in transformer_engine/pytorch/quantization.py
already enable FP8, FP8 block scaling, MXFP8 and NVFP4 there, and
get_default_fp8_recipe() returns MXFP8BlockScaling. Footnote 3 said
MXFP8 was limited to 10.x, but the check covers 10.0 through 11.x.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Signed-off-by: 0z5a <dezhen.lu@student.uni-tuebingen.de>
@github-actions github-actions Bot added the community-contribution PRs from external contributor outside the core maintainers, representing community-driven work. label Oct 17, 2026

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant