Skip to content

FlashInfer sampling#2456

Merged
tdene merged 23 commits into
NVIDIA:mainfrom
tdene:tde/flashinfer_sampling
May 5, 2026
Merged

FlashInfer sampling#2456
tdene merged 23 commits into
NVIDIA:mainfrom
tdene:tde/flashinfer_sampling

Conversation

@tdene

@tdene tdene commented Dec 2, 2025

Copy link
Copy Markdown
Contributor

What does this PR do ?

⚠️ For major changes (either in lines of code or in its impact), please make sure to first share discuss a design-doc with the team.

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]
Loading

Pre-checks

  • I want this PR in a versioned release and have added the appropriate Milestone (e.g., Core 0.8)
  • I have added relevant unit tests
  • I have added relevant functional tests
  • I have added proper typing to my code Typing guidelines
  • I have added relevant documentation
  • I have run the autoformatter.sh on my PR

Code review

The following process is enforced via the CODEOWNERS file for changes into megatron/core. For changes outside of megatron/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

  1. Attach the Expert Review label when your PR is ready for review.
  2. GitHub auto-assigns expert reviewers based on your changes. They will get notified and pick up your PR soon.

⚠️ Only proceed to the next step once all reviewers have approved, merge-conflict are resolved and the CI is passing.
Final Review might get declined if these requirements are not fulfilled.

(Step 3): Final Review

  1. Add Final Review label
  2. GitHub auto-assigns final reviewers based on your changes. They will get notified and pick up your PR soon.

(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, select Cherry-pick to 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.com or zijiey@nvidia.com.

Merging your PR

Any member of core-adlr and core-nemo will be able to merge your PR.

@copy-pr-bot

copy-pr-bot Bot commented Dec 2, 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.

@copy-pr-bot

copy-pr-bot Bot commented Mar 18, 2026

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.

@tdene tdene force-pushed the tde/flashinfer_sampling branch 6 times, most recently from dd26f3c to 84f84de Compare April 12, 2026 20:59
@tdene tdene force-pushed the tde/flashinfer_sampling branch 9 times, most recently from 8f8fede to 39cd9ff Compare April 19, 2026 17:04
@tdene tdene force-pushed the tde/flashinfer_sampling branch from 310037b to 329a85d Compare April 23, 2026 14:01
@tdene tdene marked this pull request as ready for review April 28, 2026 14:07
@tdene tdene requested review from a team as code owners April 28, 2026 14:07
@svcnvidia-nemo-ci svcnvidia-nemo-ci requested a review from a team April 28, 2026 14:07
@tdene tdene force-pushed the tde/flashinfer_sampling branch from 329a85d to f8549dc Compare April 28, 2026 14:10
tdene and others added 3 commits May 4, 2026 08:39
Co-Authored-By: Keshav Santhanam <ksanthanam@nvidia.com>
Co-Authored-By: Keshav Santhanam <ksanthanam@nvidia.com>
@tdene tdene force-pushed the tde/flashinfer_sampling branch from d1cfd43 to de5c56d Compare May 4, 2026 13:48
@tdene tdene enabled auto-merge May 4, 2026 14:06
@tdene tdene added this pull request to the merge queue May 4, 2026
@svcnvidia-nemo-ci

Copy link
Copy Markdown

🔄 Merge queue validation started!

You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25329525317

@github-merge-queue github-merge-queue Bot removed this pull request from the merge queue due to failed status checks May 4, 2026
@tdene tdene requested a review from a team as a code owner May 4, 2026 19:20
@svcnvidia-nemo-ci svcnvidia-nemo-ci removed the Approved All necessary approvals have been made label May 4, 2026
@svcnvidia-nemo-ci svcnvidia-nemo-ci added the Approved All necessary approvals have been made label May 4, 2026
@tdene

tdene commented May 4, 2026

Copy link
Copy Markdown
Contributor Author

/ok to test e580bde

@tdene tdene enabled auto-merge May 4, 2026 19:54
@tdene tdene added this pull request to the merge queue May 4, 2026
@svcnvidia-nemo-ci

Copy link
Copy Markdown

🔄 Merge queue validation started!

You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25344830114

@svcnvidia-nemo-ci

Copy link
Copy Markdown

🔄 Merge queue validation started!

You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25346979513

@github-merge-queue github-merge-queue Bot removed this pull request from the merge queue due to failed status checks May 5, 2026
@tdene tdene added this pull request to the merge queue May 5, 2026
@svcnvidia-nemo-ci

Copy link
Copy Markdown

🔄 Merge queue validation started!

You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25351891254

@svcnvidia-nemo-ci

Copy link
Copy Markdown

🔄 Merge queue validation started!

You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25355425973

Merged via the queue into NVIDIA:main with commit 878228f May 5, 2026
68 of 71 checks passed
Andron00e pushed a commit to Andron00e/Megatron-LM that referenced this pull request May 28, 2026
…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.
yhgalaxy pushed a commit to yhgalaxy/Megatron-LM that referenced this pull request Jun 17, 2026
Co-authored-by: Keshav Santhanam <ksanthanam@nvidia.com>
Signed-off-by: yhgalaxy <yhgalaxy@outlook.com>
jon-barker pushed a commit to jon-barker/Megatron-LM that referenced this pull request Jul 10, 2026
Co-authored-by: Keshav Santhanam <ksanthanam@nvidia.com>
Signed-off-by: Jon Barker <jbarker@aws-cmh-slurm-1-vscode-02.cm.cluster>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Approved All necessary approvals have been made complexity: high

Projects

None yet

Development

Successfully merging this pull request may close these issues.

8 participants