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

Add fsdp tests back to GPU CI #5674

Merged
merged 6 commits into from
Oct 7, 2023
Merged

Conversation

vanbasten23
Copy link
Collaborator

@vanbasten23 vanbasten23 commented Oct 4, 2023

fsdp test was broken in one of the tf pin updates.
This pr tests if it's possible to reenable it in the GPU CI and add it back if so.

@vanbasten23 vanbasten23 changed the title [DO_NOT_REVIEW] Check fsdp test in GPU CI Add fsdp tests back to GPU CI Oct 6, 2023
@vanbasten23 vanbasten23 marked this pull request as ready for review October 6, 2023 20:57
Copy link
Collaborator

@alanwaketan alanwaketan left a comment

Choose a reason for hiding this comment

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

LGTM. Do we have a bug to track the memory regression?

@vanbasten23
Copy link
Collaborator Author

LGTM. Do we have a bug to track the memory regression?

Just cc'ed you in the bug. Thanks Jiewen!

@vanbasten23
Copy link
Collaborator Author

Thanks for the review!

@vanbasten23 vanbasten23 force-pushed the updatePinSlightly20230829 branch from e570496 to 80803f4 Compare October 7, 2023 04:12
@vanbasten23 vanbasten23 merged commit b84f798 into master Oct 7, 2023
17 checks passed
qihqi pushed a commit that referenced this pull request Oct 10, 2023
* use manfei's change

* use the new flag

* reduce data size

* reduce batch size

* keep reducing batch size to 64

* remove comments
zpcore pushed a commit that referenced this pull request Oct 19, 2023
* use manfei's change

* use the new flag

* reduce data size

* reduce batch size

* keep reducing batch size to 64

* remove comments
ghpvnist pushed a commit to ghpvnist/xla that referenced this pull request Oct 31, 2023
* use manfei's change

* use the new flag

* reduce data size

* reduce batch size

* keep reducing batch size to 64

* remove comments
chunnienc pushed a commit to chunnienc/xla that referenced this pull request Dec 14, 2023
* use manfei's change

* use the new flag

* reduce data size

* reduce batch size

* keep reducing batch size to 64

* remove comments
golechwierowicz pushed a commit that referenced this pull request Jan 12, 2024
* use manfei's change

* use the new flag

* reduce data size

* reduce batch size

* keep reducing batch size to 64

* remove comments
bhavya01 pushed a commit that referenced this pull request Apr 22, 2024
* use manfei's change

* use the new flag

* reduce data size

* reduce batch size

* keep reducing batch size to 64

* remove comments
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