Skip to content

Use TE quantization modes for first/last bf16 layers - #715

Merged
maanug-nv merged 4 commits into
mainfrom
maanug/port-mlm-3886
Oct 27, 2025
Merged

Use TE quantization modes for first/last bf16 layers#715
maanug-nv merged 4 commits into
mainfrom
maanug/port-mlm-3886

Conversation

@maanug-nv

Copy link
Copy Markdown
Contributor

Port MLM 3886
Make use of this TE feature for multiple quantization modes: NVIDIA/TransformerEngine#2043

@maanug-nv
maanug-nv requested a review from a team as a code owner September 22, 2025 23:35
@ananthsub
ananthsub requested a review from erhoo82 September 23, 2025 00:27
@maanug-nv
maanug-nv marked this pull request as draft October 20, 2025 23:57
@copy-pr-bot

copy-pr-bot Bot commented Oct 20, 2025

Copy link
Copy Markdown

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.

@maanug-nv

Copy link
Copy Markdown
Contributor Author

nothing that #1006 will require changes in this PR to re-enable comm overlap for 47b and 56b

@ananthsub ananthsub added the r0.2.0 Cherry-pick label for r0.2.0 release branch label Oct 22, 2025
Signed-off-by: Maanu Grover <maanug@nvidia.com>
Signed-off-by: Maanu Grover <maanug@nvidia.com>
Signed-off-by: Maanu Grover <maanug@nvidia.com>
Signed-off-by: Maanu Grover <maanug@nvidia.com>
@maanug-nv
maanug-nv force-pushed the maanug/port-mlm-3886 branch from efdf868 to 07cb8e5 Compare October 27, 2025 18:01
@maanug-nv
maanug-nv marked this pull request as ready for review October 27, 2025 18:02
@maanug-nv
maanug-nv requested a review from ananthsub October 27, 2025 18:02
@maanug-nv
maanug-nv removed request for a team and erhoo82 October 27, 2025 18:02
@maanug-nv

Copy link
Copy Markdown
Contributor Author

Closes #748

@maanug-nv maanug-nv self-assigned this Oct 27, 2025
@maanug-nv maanug-nv linked an issue Oct 27, 2025 that may be closed by this pull request
@maanug-nv
maanug-nv enabled auto-merge (squash) October 27, 2025 18:18
@maanug-nv
maanug-nv merged commit d0209a5 into main Oct 27, 2025
28 checks passed
@maanug-nv
maanug-nv deleted the maanug/port-mlm-3886 branch October 27, 2025 22:18
chtruong814 pushed a commit that referenced this pull request Oct 27, 2025
Signed-off-by: Maanu Grover <maanug@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
paul-gibbons pushed a commit to paul-gibbons/Megatron-Bridge that referenced this pull request Oct 29, 2025
Signed-off-by: Maanu Grover <maanug@nvidia.com>
Signed-off-by: Paul Gibbons <pgibbons@nvidia.com>
paul-gibbons pushed a commit to paul-gibbons/Megatron-Bridge that referenced this pull request Oct 29, 2025
Signed-off-by: Maanu Grover <maanug@nvidia.com>
Signed-off-by: Paul Gibbons <pgibbons@nvidia.com>
nv-mollys pushed a commit that referenced this pull request Oct 31, 2025
Signed-off-by: Maanu Grover <maanug@nvidia.com>
Signed-off-by: mollys <mollys@mollys.nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

r0.2.0 Cherry-pick label for r0.2.0 release branch

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Comm overlap broken with Nemotronh 47b, 56b

2 participants