[dev] fix git history for dev pull main 260122 by FDecaYed · Pull Request #3094 · NVIDIA/Megatron-LM

FDecaYed · 2026-01-27T06:10:28Z

What does this PR do ?

The dev pull main PR
#3045
was merged with squashing, thus git history is not merged. creating this simple PR to merge git history

Also fixed a simple merge error that caused duplicate code

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

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

Feel free to message or comment the @mcore-oncall to help accelerate your merge into main. The less complex your PR is, the faster it will be approved and merged!

(Step 1): Add PR label `Expert Review`

(Step 2): Collect the expert reviewers reviews

Attach the Expert Review label when your PR is ready for review.
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

Add Final Review label
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.

…up (NVIDIA#2634) Co-authored-by: Teodor-Dumitru Ene <teodord.ene@gmail.com>

Signed-off-by: Robin Zhang <robinz@nvidia.com> Co-authored-by: Jared Casper <155158+jaredcasper@users.noreply.github.com>

Signed-off-by: oliver könig <okoenig@nvidia.com>

…NVIDIA#2599)

…VIDIA#2671)

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Signed-off-by: Maanu Grover <maanug@nvidia.com>

Co-authored-by: William Dykas <wdykas@cw-pdx-cs-001-vscode-02.cm.cluster> Co-authored-by: root <root@gpu-h100-0371.cm.cluster> Co-authored-by: root <root@gpu-h100-0159.cm.cluster>

Co-authored-by: Philip Petrakian <pgpetrak@gmail.com>

…IA#2131) Signed-off-by: Jennifer Chen <jennifchen@nvidia.com> Co-authored-by: Jared Casper <155158+jaredcasper@users.noreply.github.com>

… scale (NVIDIA#2445) Signed-off-by: Antoni-Joan Solergibert <asolergibert@nvidia.com> Co-authored-by: Deepak Narayanan <2724038+deepakn94@users.noreply.github.com>

Signed-off-by: oliver könig <okoenig@nvidia.com>

Signed-off-by: Lifu Zhang <lifuz@login-lyris01.lyris.clusters.nvidia.com> Co-authored-by: Lifu Zhang <lifuz@login-lyris01.lyris.clusters.nvidia.com>

Co-authored-by: Shanmugam Ramasamy <shanmugamr@cw-dfw-cs-001-login-01.cm.cluster>

Signed-off-by: Keshav Santhanam <ksanthanam@nvidia.com> Co-authored-by: Kan Zhu <kanz@nvidia.com>

…DIA#2743)

…DIA#3019)

Signed-off-by: Cory Ye <cye@nvidia.com>

…VIDIA#2897) Signed-off-by: Keshav Santhanam <ksanthanam@nvidia.com>

…ntion/CrossAttention (NVIDIA#2761)

…footprint (NVIDIA#2572) Signed-off-by: Jimmy Zhang <jiemingz@nvidia.com> Signed-off-by: Jieming Zhang <jiemingz@nvidia.com> Co-authored-by: Robin Zhang <robinz@nvidia.com>

)

… memory footprint (NVIDIA#2572)" This reverts commit 8baf014.

Co-authored-by: root <root@gpu-h100-0348.cm.cluster> Co-authored-by: root <root@gpu-h100-0193.cm.cluster> Co-authored-by: root <root@gpu-h100-0082.cm.cluster> Co-authored-by: root <root@gpu-h100-0495.cm.cluster> Co-authored-by: William Dykas <wdykas@cw-pdx-cs-001-vscode-02.cm.cluster> Co-authored-by: root <root@gpu-h100-0213.cm.cluster> Co-authored-by: root <root@gpu-h100-0435.cm.cluster> Co-authored-by: root <root@gpu-h100-0188.cm.cluster> Co-authored-by: root <root@gpu-h100-0032.cm.cluster> Co-authored-by: root <root@gpu-h100-0023.cm.cluster> Co-authored-by: root <root@gpu-h100-0368.cm.cluster> Co-authored-by: root <root@gpu-h100-0203.cm.cluster> Co-authored-by: root <root@gpu-h100-0229.cm.cluster> Co-authored-by: root <root@gpu-h100-0123.cm.cluster> Co-authored-by: root <root@gpu-h100-0217.cm.cluster> Co-authored-by: root <root@gpu-h100-0496.cm.cluster> Co-authored-by: root <root@gpu-h100-0022.cm.cluster> Co-authored-by: root <root@gpu-h100-0176.cm.cluster> Co-authored-by: root <root@gpu-h100-0261.cm.cluster> Co-authored-by: root <root@gpu-h100-0029.cm.cluster> Co-authored-by: root <root@gpu-h100-0215.cm.cluster> Co-authored-by: Jon Barker <jbarker@nvidia.com>

Signed-off-by: oliver könig <okoenig@nvidia.com>

Co-authored-by: Yashaswi Karnati <144376261+yashaswikarnati@users.noreply.github.com>

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Signed-off-by: oliver könig <okoenig@nvidia.com>

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

copy-pr-bot · 2026-01-27T06:10:33Z

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.

FDecaYed · 2026-01-27T06:17:34Z

/ok to test da56650

Phlip79 and others added 30 commits December 17, 2025 19:19

pip install uv during GH action (NVIDIA#2695)

9288125

chore: rotate oncall schedule

32b9ee4

Don't delete svcnvidia-nemo-ci team from oncall (NVIDIA#2703)

ff4a622

RL: Rollouts should be distributed over the regular data parallel gro…

3d1a5c8

…up (NVIDIA#2634) Co-authored-by: Teodor-Dumitru Ene <teodord.ene@gmail.com>

Use pull_request_target and don't use uv (NVIDIA#2702)

d321026

Optimize TE cudagraph input memory (NVIDIA#2392)

94b4759

Signed-off-by: Robin Zhang <robinz@nvidia.com> Co-authored-by: Jared Casper <155158+jaredcasper@users.noreply.github.com>

ci(fix): Pin gojq to stable version (NVIDIA#2480)

c7e5489

Signed-off-by: oliver könig <okoenig@nvidia.com>

NVLS - fused reduce-scatter + residual + rms-norm + all-gather kernel (…

f19b59e

…NVIDIA#2599)

Default UVM level to 0. (NVIDIA#2450)

0170e70

docs: improve documentation organization and add additional guides (N…

0b13c98

…VIDIA#2671)

Revert "Default UVM level to 0. (NVIDIA#2450)" (NVIDIA#2713)

d81b37b

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Add missing imports in no-triton fallback (NVIDIA#2711)

a6f822c

Signed-off-by: Maanu Grover <maanug@nvidia.com>

Fixes for NVIDIA#2450. (NVIDIA#2714)

1503c33

Add RL parameter to set parallel generation tasks (NVIDIA#2712)

1a2257b

Refit prep 3 (NVIDIA#2708)

30694e0

Co-authored-by: William Dykas <wdykas@cw-pdx-cs-001-vscode-02.cm.cluster> Co-authored-by: root <root@gpu-h100-0371.cm.cluster> Co-authored-by: root <root@gpu-h100-0159.cm.cluster>

chore: Add cudagraph codeowners (NVIDIA#2720)

000c2e2

[docs] Add developer section to docs (NVIDIA#2717)

703bc36

Fix UVM argument for RL (NVIDIA#2722)

7f471d7

[dcos] Update docs title to Megatron Core (NVIDIA#2729)

ddf691d

remove fp16 assert in moe_grouped_gemm & EP (NVIDIA#2495)

4193f3a

Co-authored-by: Philip Petrakian <pgpetrak@gmail.com>

Improve ModelOpt paths & add more Nemotron/hybrid model support (NVID…

a057662

…IA#2131) Signed-off-by: Jennifer Chen <jennifchen@nvidia.com> Co-authored-by: Jared Casper <155158+jaredcasper@users.noreply.github.com>

Add options to improve data loader initialization time, especially at…

cfd980b

… scale (NVIDIA#2445) Signed-off-by: Antoni-Joan Solergibert <asolergibert@nvidia.com> Co-authored-by: Deepak Narayanan <2724038+deepakn94@users.noreply.github.com>

Update copy-pr-bot.yaml [skip ci]

1c67e7e

ci: Fix copy-pr-bot update (NVIDIA#2736)

8ea3b8d

Signed-off-by: oliver könig <okoenig@nvidia.com>

Add oncall to all new PRs (NVIDIA#2734)

5b1ef07

Hsdp register submesh fix lifuz mirror (NVIDIA#2467)

cc1b0b5

Signed-off-by: Lifu Zhang <lifuz@login-lyris01.lyris.clusters.nvidia.com> Co-authored-by: Lifu Zhang <lifuz@login-lyris01.lyris.clusters.nvidia.com>

Adding stop word support (NVIDIA#2685)

1febe9f

Co-authored-by: Shanmugam Ramasamy <shanmugamr@cw-dfw-cs-001-login-01.cm.cluster>

Fix oncall assign (NVIDIA#2737)

a477766

Add support for non-decode CUDA graphs for Mamba models (NVIDIA#2474)

f5d4c3a

Signed-off-by: Keshav Santhanam <ksanthanam@nvidia.com> Co-authored-by: Kan Zhu <kanz@nvidia.com>

Update sequence packing case when dummy PackedSeqParams are used (NVI…

0a77122

…DIA#2743)

mathemakitten and others added 18 commits January 21, 2026 15:38

Make attn mask inversion in-place instead of allocating it again (NVI…

28c7221

…DIA#3019)

[Megatron-FSDP] Fix incorrect gradient scaling target. (NVIDIA#3023)

ba456fd

Signed-off-by: Cory Ye <cye@nvidia.com>

Ensure that last prefill chunk is handled correctly by Mamba models (N…

096dbeb

…VIDIA#2897) Signed-off-by: Keshav Santhanam <ksanthanam@nvidia.com>

Replaces ModuleSpec with Protocols for some of the inputs to SelfAtte…

90e685b

…ntion/CrossAttention (NVIDIA#2761)

Update oncall schedule (NVIDIA#3017)

dd72aee

Various CUDA graph improvements on capture time, replay time, memory …

8baf014

…footprint (NVIDIA#2572) Signed-off-by: Jimmy Zhang <jiemingz@nvidia.com> Signed-off-by: Jieming Zhang <jiemingz@nvidia.com> Co-authored-by: Robin Zhang <robinz@nvidia.com>

chore: rotate oncall schedule

bbbedbb

Add script for batch running CI tests across distinct nodes (NVIDIA#3047

8c768ea

)

Revert "Various CUDA graph improvements on capture time, replay time,…

c8caeb2

… memory footprint (NVIDIA#2572)" This reverts commit 8baf014.

Catch case of negative tokens to generate (NVIDIA#2985)

91ebe29

ci: Disable broken GRPO tests

27168e2

Signed-off-by: oliver könig <okoenig@nvidia.com>

Sync GitHub and Slack teams (NVIDIA#3037)

f85ad03

Support custom Router implementations in MoELayer (NVIDIA#2891)

7857383

Co-authored-by: Yashaswi Karnati <144376261+yashaswikarnati@users.noreply.github.com>

ci: Remove Github transition comment from CI (NVIDIA#2881)

cfa240a

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

ci: Override N_REPEAT (NVIDIA#3051)

8db8d08

Signed-off-by: oliver könig <okoenig@nvidia.com>

ci: Skip test_precision_aware_optimizer (NVIDIA#3062)

420aa6a

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

Merge branch 'main' into deyuf/dev_pull_main_260122_fix_git

da56650

FDecaYed requested review from a team as code owners January 27, 2026 06:10

copy-pr-bot bot temporarily deployed to nemo-ci January 27, 2026 06:17 Inactive

ko3n1g added this to the Core 0.16 milestone Jan 27, 2026

copy-pr-bot bot temporarily deployed to nemo-ci January 27, 2026 06:17 Inactive

copy-pr-bot bot temporarily deployed to test January 27, 2026 06:18 Inactive

yanring approved these changes Jan 27, 2026

View reviewed changes

ko3n1g merged commit 08357d8 into NVIDIA:dev Jan 27, 2026
42 of 44 checks passed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Comments

[dev] fix git history for dev pull main 260122#3094

[dev] fix git history for dev pull main 260122#3094
ko3n1g merged 207 commits intoNVIDIA:devfrom
FDecaYed:deyuf/dev_pull_main_260122_fix_git

FDecaYed commented Jan 27, 2026 •

edited

Loading

Uh oh!

copy-pr-bot bot commented Jan 27, 2026

Uh oh!

FDecaYed commented Jan 27, 2026

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

20 participants

Comments

Conversation

FDecaYed commented Jan 27, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

What does this PR do ?

Contribution process

Pre-checks

Code review

(Step 1): Add PR label Expert Review

(Step 2): Collect the expert reviewers reviews

(Step 3): Final Review

(Optional Step 4): Cherry-pick into release branch

Merging your PR

Uh oh!

copy-pr-bot bot commented Jan 27, 2026

Uh oh!

FDecaYed commented Jan 27, 2026

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

20 participants

FDecaYed commented Jan 27, 2026 •

edited

Loading

(Step 1): Add PR label `Expert Review`