Skip to content

Conversation

@Xiaoming-AMD
Copy link
Collaborator

get_megatron_optimizer will use the ddp_config, while megatron-lm's torch_fsdp not hold the ddp_config

Copy link
Contributor

@wenxie-amd wenxie-amd left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@wenxie-amd wenxie-amd merged commit d0da8db into main Apr 18, 2025
2 checks passed
@Xiaoming-AMD Xiaoming-AMD deleted the dev/xiaoming/fix_fsdp2 branch April 18, 2025 02:28
@Xiaoming-AMD Xiaoming-AMD changed the title Dev/xiaoming/fix fsdp2 fix(torch_fsdp): get ddp_config failed when use torch_fsdp Jun 4, 2025
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