Skip to content

Separate pallas test into a separate test shard #9231

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

Merged
merged 2 commits into from
May 23, 2025
Merged

Conversation

ghpvnist
Copy link
Collaborator

@ghpvnist ghpvnist commented May 21, 2025

Reduces total TPU test time by 7 minutes, but TPU test is no longer a bottleneck. It's still good to add this optimization in case it becomes a bottleneck again.

@ghpvnist ghpvnist requested a review from zhanyong-wan May 23, 2025 05:50
@ghpvnist ghpvnist self-assigned this May 23, 2025
@ghpvnist ghpvnist marked this pull request as ready for review May 23, 2025 05:52
Copy link
Collaborator

@zhanyong-wan zhanyong-wan left a comment

Choose a reason for hiding this comment

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

Thanks!

@ghpvnist ghpvnist merged commit 0d14c8a into pytorch:master May 23, 2025
55 of 58 checks passed
@ghpvnist ghpvnist deleted the ci2 branch May 23, 2025 09:42
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.

2 participants