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

Fix FSDP support with pytorch 2.1.0 #454

Merged
merged 1 commit into from
May 2, 2024
Merged

Fix FSDP support with pytorch 2.1.0 #454

merged 1 commit into from
May 2, 2024

Conversation

JadeCopet
Copy link
Contributor

Fix FSDP code wrote for torch 2.0 but breaking for 2.1.0 as raised by #358 and #450 .

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label May 2, 2024
Copy link
Contributor

@felixkreuk felixkreuk left a comment

Choose a reason for hiding this comment

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

Looks good! Thanks for taking care of the Jade!

@JadeCopet JadeCopet merged commit 795f8dc into main May 2, 2024
3 checks passed
Copy link
Contributor

@simonrouard simonrouard left a comment

Choose a reason for hiding this comment

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

Looks good! Thanks Jade

@JadeCopet JadeCopet deleted the fix_fsdp branch May 2, 2024 08:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants