Skip to content

Support DDP overlap for models with repeated parameters - #2837

Merged
ericharper merged 3 commits into
NVIDIA:mainfrom
deepakn94:dnarayanan/ddp_overlap_with_repeated_modules
Jan 14, 2026
Merged

Support DDP overlap for models with repeated parameters#2837
ericharper merged 3 commits into
NVIDIA:mainfrom
deepakn94:dnarayanan/ddp_overlap_with_repeated_modules

Conversation

@deepakn94

Copy link
Copy Markdown
Contributor

No description provided.

Comment thread megatron/core/distributed/param_and_grad_buffer.py
… parameters)

Signed-off-by: Deepak Narayanan <dnarayanan@nvidia.com>
@deepakn94
deepakn94 force-pushed the dnarayanan/ddp_overlap_with_repeated_modules branch from facecf4 to 59f5560 Compare January 7, 2026 21:51
@deepakn94
deepakn94 force-pushed the dnarayanan/ddp_overlap_with_repeated_modules branch from d039292 to ffecdc9 Compare January 8, 2026 04:04
@deepakn94
deepakn94 force-pushed the dnarayanan/ddp_overlap_with_repeated_modules branch from ffecdc9 to 5e5adba Compare January 12, 2026 22:11
Signed-off-by: Deepak Narayanan <dnarayanan@nvidia.com>
@ericharper
ericharper enabled auto-merge January 14, 2026 05:15
@deepakn94
deepakn94 removed the request for review from Phlip79 January 14, 2026 07:20
@ericharper
ericharper added this pull request to the merge queue Jan 14, 2026
Merged via the queue into NVIDIA:main with commit aa3f105 Jan 14, 2026
45 of 47 checks passed
@deepakn94
deepakn94 deleted the dnarayanan/ddp_overlap_with_repeated_modules branch January 14, 2026 16:26
daiyaanarfeen pushed a commit to daiyaanarfeen/Megatron-LM that referenced this pull request Feb 23, 2026
Signed-off-by: Deepak Narayanan <dnarayanan@nvidia.com>
yangbofun pushed a commit to xlm-research/Megatron-LM that referenced this pull request May 22, 2026
Signed-off-by: Deepak Narayanan <dnarayanan@nvidia.com>
terminator123 pushed a commit to 021ai/Megatron-LM that referenced this pull request Aug 3, 2026
Signed-off-by: Deepak Narayanan <dnarayanan@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants