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

Enable support for Kueue for GKETPUJob #623

Merged
merged 1 commit into from
Aug 13, 2024

Conversation

samos123
Copy link
Contributor

@samos123 samos123 commented Aug 3, 2024

Internal V5E testing capacity requires using Kueue. This PR allows you to specify which Kueue Queue to use. The GPUGKEJob already supported this so moving the shared config to GKEJob.

@samos123 samos123 marked this pull request as ready for review August 3, 2024 06:33
@samos123
Copy link
Contributor Author

Do we need anything else to get this merged? Do we need another review from @ruomingp ?

@markblee markblee added this pull request to the merge queue Aug 13, 2024
Merged via the queue into apple:main with commit 66704ea Aug 13, 2024
4 checks passed
qdavid1 pushed a commit to qdavid1/axlearn that referenced this pull request Dec 11, 2024
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