Repository navigation
build: update Transformer Engine to release_v2.20.2-nemo tip - #6384
Closed
balasaajay wants to merge 1 commit into
Closed
balasaajay wants to merge 1 commit into
balasaajay wants to merge 1 commit into
Conversation
Signed-off-by: Ajay Balasa <abalasa@nvidia.com>
|
Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually. Contributors can view more details about this message here. |
Contributor
Author
|
/ok to test 3aed45c |
1 task
This branch was successfully deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
What does this PR do?
Pin Transformer Engine to
44f31bbeaf99bd30a67b928000d4b980026debd2, the tip ofrelease_v2.20.2-nemoresolved on 2026-10-09. The effective TE source advances from6ea2a74ato44f31bbe, picking up the fused grouped MLP deterministic-dprob fix in NVIDIA/TransformerEngine#3635.Changelog
441a987409b8fdd2b90b463f7999bac970109a43to4e7f45066b37a88060058ff012eeacb37c90f7d0, the signed companion NVIDIA/Megatron-LM#8036 revision. Bridge resolves TE through MCore and its dependency-layer check requires their locked TE sources to agree. This advance also includes the 15 MCore main commits between the old gitlink and the companion PR's base; merge the companion PR first.2.20.2+44f31bbe.uv.lockusing uv 0.7.2 in the existingnvcr.io/nvidia/pytorch:26.08-py3base. All 345 package names remain unchanged and no unrelated package versions move. The resolver also normalizes existing platform markers; forcing re-resolution on the unchanged base reproduces those same marker changes.Validation
uv lockanduv lock --checkin the supported PyTorch base container.uv run --no-sync pre-commit run --all-filesin a Bridge CI container..github/scripts/test_docker_dependency_layers.sh, including exact Bridge/MCore TE source equality.git diff --check.needs-more-testsandfull-test-suiterequest the expanded CI matrix.Before ready for review