-
Notifications
You must be signed in to change notification settings - Fork 531
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Bump composer to 0.24.1 + FSDP config device_mesh deprecation #1487
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should be 24.1?
@mvpatel2000 nah I haven't cut the newest composer release yet. I mean this PR can wait until then, but taking out device mesh completely is 0.24.0 onwards |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Might as well wait for 0.24.1 just to make it one PR, but LGTM
@mvpatel2000 @dakinggg composer v0.24.1 is out so updated this PR. should be good to go |
Bump composer to 0.24.1, and as part of that, we replace parts of code that assume
device_mesh
is a key specified in the fsdp_config dict.Added unit test for correct fsdp_config --> ffn_config device mesh parsing
This also resolves issues we were seeing with not using device mesh in regression tests.
Successfully reran failing regressions, which succeeded.