FlashInfer sampling#2456
Conversation
b8558aa to
3c9fa91
Compare
|
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. |
dd26f3c to
84f84de
Compare
8f8fede to
39cd9ff
Compare
310037b to
329a85d
Compare
329a85d to
f8549dc
Compare
Co-Authored-By: Keshav Santhanam <ksanthanam@nvidia.com>
Co-Authored-By: Keshav Santhanam <ksanthanam@nvidia.com>
d1cfd43 to
de5c56d
Compare
|
🔄 Merge queue validation started! You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25329525317 |
|
/ok to test e580bde |
|
🔄 Merge queue validation started! You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25344830114 |
|
🔄 Merge queue validation started! You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25346979513 |
|
🔄 Merge queue validation started! You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25351891254 |
|
🔄 Merge queue validation started! You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25355425973 |
…ync-2026-05 Upstream tip: 932d9ee (Add periodic GPU sniff tests, 2026-05-07). Conflict resolution: - 13 modify/delete conflicts in .github/* and skills/*: kept deletion (matches policy from c037427 removing upstream-NVIDIA CI machinery not relevant to this fork). - AGENTS.md content conflict: kept fork version (skills/contributing sections from upstream don't apply since the directories were removed). - New upstream additions matching the same removal policy were dropped: .claude/settings.json, .github/workflows/nightly-sync-main-to-dev.yml, skills/{cicd,linting-and-formatting,nightly-sync,run-on-slurm,testing, update-golden-values}/SKILL.md. - SECURITY.md (new from upstream): kept (generic security policy). Verified post-merge: - pretrain_gpt.py logging-patch hook intact (lines 405-408). - megatron/core/optimizer/__init__.py emerging_optimizers + muon logic intact. - megatron/training/arguments.py custom muon flags intact (lines 2297-2309). Notable upstream changes pulled in: - Removed legacy transformer + legacy GPT modules (NVIDIA#4207, NVIDIA#4322). - Docker base image bump to 26.04-py3 (NVIDIA#4611). - Inference fixes (vLLM grouped GEMM NVIDIA#4566, FlashInfer sampling NVIDIA#2456, EP sync NVIDIA#4607, MoE dispatcher fixes NVIDIA#4576). - Gradient corruption fix with layerwise param all-gather overlap (NVIDIA#4609). - Hybrid model + Flextron + GPU sniff tests + named validation sets + fault injection + InJob restart.
Co-authored-by: Keshav Santhanam <ksanthanam@nvidia.com> Signed-off-by: yhgalaxy <yhgalaxy@outlook.com>
Co-authored-by: Keshav Santhanam <ksanthanam@nvidia.com> Signed-off-by: Jon Barker <jbarker@aws-cmh-slurm-1-vscode-02.cm.cluster>
What does this PR do ?
Contribution process
flowchart LR A[Pre-checks] --> B[PR Tests] subgraph Code Review/Approval C1[Expert Review] --> C2[Final Review] end B --> C1 C2 --> D[Merge]Pre-checks
Core 0.8)Code review
The following process is enforced via the CODEOWNERS file for changes into
megatron/core. For changes outside ofmegatron/core, it is up to the PR author whether or not to tag the Final Reviewer team.For MRs into `main` branch
(Step 1): Add PR label
Expert Review(Step 2): Collect the expert reviewers reviews
Expert Reviewlabel when your PR is ready for review.Final Review might get declined if these requirements are not fulfilled.
(Step 3): Final Review
Final Reviewlabel(Optional Step 4): Cherry-pick into release branch
If this PR also needs to be merged into
core_r*release branches, after this PR has been merged, selectCherry-pickto open a new PR into the release branch.For MRs into `dev` branch
The proposed review process for `dev` branch is under active discussion.MRs are mergable after one approval by either
eharper@nvidia.comorzijiey@nvidia.com.Merging your PR
Any member of core-adlr and
core-nemowill be able to merge your PR.