Skip to content
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

Merged
merged 11 commits into from
Aug 27, 2024

Conversation

snarayan21
Copy link
Contributor

@snarayan21 snarayan21 commented Aug 27, 2024

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.

Copy link
Collaborator

@mvpatel2000 mvpatel2000 left a 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?

@snarayan21
Copy link
Contributor Author

@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

Copy link
Collaborator

@dakinggg dakinggg left a 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

@snarayan21
Copy link
Contributor Author

@mvpatel2000 @dakinggg composer v0.24.1 is out so updated this PR. should be good to go

@snarayan21 snarayan21 changed the title Bump composer to 0.24.0 + FSDP config device_mesh deprecation Bump composer to 0.24.1 + FSDP config device_mesh deprecation Aug 27, 2024
@snarayan21 snarayan21 merged commit 249db93 into main Aug 27, 2024
9 checks passed
@mvpatel2000 mvpatel2000 deleted the saaketh/composer_bump_0240_ony branch August 28, 2024 01:53
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