Skip to content

Qwen3-Next support on Blackwell - #1216

Merged
adityavavreNVDA merged 7 commits into
mainfrom
avavre/qwen3next_blackwell_support
Nov 12, 2025
Merged

Qwen3-Next support on Blackwell#1216
adityavavreNVDA merged 7 commits into
mainfrom
avavre/qwen3next_blackwell_support

Conversation

@adityavavreNVDA

Copy link
Copy Markdown
Contributor

What does this PR do ?

This PR adds changes to support Qwen3-Next on Blackwell. JIT fuser on blackwell causes hang issues with Qwen3-Next as stated here: NVIDIA/Megatron-LM#2058
To mitigate this we add a similar functionality in megatron-bridge to disable JIT fuser.
Below are the B200 training logs for Qwen3-Next:
image

Signed-off-by: Aditya Vavre <avavre@nvidia.com>
@copy-pr-bot

copy-pr-bot Bot commented Nov 7, 2025

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@adityavavreNVDA

Copy link
Copy Markdown
Contributor Author

/ok to test d3966c6

Comment thread src/megatron/bridge/recipes/qwen/qwen3_next.py Outdated
Comment thread src/megatron/bridge/training/config.py Outdated
@ananthsub ananthsub added the r0.2.0 Cherry-pick label for r0.2.0 release branch label Nov 8, 2025
Signed-off-by: Aditya Vavre <avavre@nvidia.com>
Signed-off-by: Aditya Vavre <avavre@nvidia.com>
@adityavavreNVDA

Copy link
Copy Markdown
Contributor Author

/ok to test 4cb31ed

@adityavavreNVDA

Copy link
Copy Markdown
Contributor Author

/ok to test b525087

ananthsub
ananthsub previously approved these changes Nov 10, 2025
@ananthsub
ananthsub enabled auto-merge (squash) November 11, 2025 00:53
@adityavavreNVDA

Copy link
Copy Markdown
Contributor Author

/ok to test 62f62ba

@adityavavreNVDA

Copy link
Copy Markdown
Contributor Author

/ok to test 7e3b487

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.

2 participants