Skip to content

feat(recipe): backport NT3 Ultra performance updates to r0.7.0 (#6228) - #6437

Merged
malay-nagda merged 1 commit into
r0.7.0from
codex/backport-6228-r0.7.0
Oct 13, 2026
Merged

malay-nagda merged 1 commit into
r0.7.0from
codex/backport-6228-r0.7.0

Conversation

@malay-nagda

Copy link
Copy Markdown
Contributor

Backport of #6228 to r0.7.0, cherry-picked from 2eca46823b0a2cb24f895bd567af14bef8350d37 without conflicts. The backport patch ID matches the merged source commit.

Adds GTP sharding, partial CUDA graphs, cuDNN layer normalization, and NCCL EP updates to the Nemotron 3 Ultra MXFP8 performance recipes for GB200, GB300, and VRNVL72. Includes the original training integration, CUDA graph validation changes, and updated unit tests.

The required VRNVL72 rename is already on r0.7.0 through #6436.

Validation:

  • All pre-commit hooks passed using uv run --no-project --with pre-commit pre-commit run --all-files.
  • git diff --cached --check passed; patch ID matches the source commit.
  • Attempted the five changed recipe and training unit-test files with isolated pytest; collection is blocked by missing PyTorch/Megatron-Core dependencies in the local environment. GPU runtime validation remains for CI.

Add GTP, partial CUDA graphs, cuDNN layer normalization, and NCCL EP
updates for Nemotron 3 Ultra MXFP8 performance recipes.

(cherry picked from commit 2eca468)

Signed-off-by: Ravi Ghadia <rghadia@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
@malay-nagda malay-nagda added feature New capabilities, enhancements, or enablement work area:recipe Training recipes and launch configs needs-review PR is ready for code review and waiting on a reviewer labels Oct 13, 2026
@github-actions

Copy link
Copy Markdown
Contributor

Automatic Claude reviews have been retired. To request a pull-request review, post a comment containing:

/review

Add model=claude to use a Claude reviewer (the default is model=codex). mode=light|strict selects the review depth; for example, /review model=claude mode=strict. Comment /review help for all options.

@malay-nagda
malay-nagda merged commit 3a88793 into r0.7.0 Oct 13, 2026
86 checks passed
@malay-nagda
malay-nagda deleted the codex/backport-6228-r0.7.0 branch October 13, 2026 14:14

This branch was successfully deployed

1 active deployment
test — 2a05d0ed Deployed Oct 13, 2026 by copy-pr-bot[bot] via cicd-wait-in-queue #22817
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:recipe Training recipes and launch configs feature New capabilities, enhancements, or enablement work needs-review PR is ready for code review and waiting on a reviewer

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants