Skip to content

Keep FSDP's and DDP's finish_grad_sync API identical - #3070

Merged
chtruong814 merged 1 commit into
NVIDIA:mainfrom
deepakn94:dnarayanan/fsdp_api_fix
Jan 24, 2026
Merged

Keep FSDP's and DDP's finish_grad_sync API identical#3070
chtruong814 merged 1 commit into
NVIDIA:mainfrom
deepakn94:dnarayanan/fsdp_api_fix

Conversation

@deepakn94

@deepakn94 deepakn94 commented Jan 24, 2026

Copy link
Copy Markdown
Contributor

This will fix the breaking functional test pipeline

Signed-off-by: Deepak Narayanan <dnarayanan@nvidia.com>
@chtruong814
chtruong814 enabled auto-merge January 24, 2026 21:21
@deepakn94
deepakn94 removed the request for review from a team January 24, 2026 21:33
@chtruong814
chtruong814 disabled auto-merge January 24, 2026 21:38
@chtruong814
chtruong814 merged commit 3a7d74d into NVIDIA:main Jan 24, 2026
46 of 50 checks passed
daiyaanarfeen pushed a commit to daiyaanarfeen/Megatron-LM that referenced this pull request Feb 23, 2026
Signed-off-by: Deepak Narayanan <dnarayanan@nvidia.com>
@deepakn94
deepakn94 deleted the dnarayanan/fsdp_api_fix branch March 3, 2026 23:57
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.

3 participants