-
Notifications
You must be signed in to change notification settings - Fork 344
Prepare for running on GKE runners #1681
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
Conversation
9759789
to
66313b0
Compare
66313b0
to
94cbd45
Compare
94cbd45
to
716fb30
Compare
716fb30
to
12ea975
Compare
12ea975
to
2518186
Compare
This PR is testing the use of GKE cloud runners.
2518186
to
af0d7e1
Compare
@@ -67,6 +67,7 @@ jobs: | |||
with: | |||
device_type: tpu | |||
device_name: v4-8 | |||
# cloud_runner: linux-x86-ct4p-240-4tpu |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
what are these comments for? Generally we should not comment out code, its quite confusing
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this intentional - is the cloud_runner a cpu runner? why is it under the tpu_unit_tests?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There are three groups of cloud runners, each one with 1 tag. This one is the tag for TPU runners. You can see them in the runners page. This PR added a new parameter to run_tests_internal (cloud_runner). Basically, once we want to enable this we just need to uncomment these lines. The reason I'm doing this PR now is to allow Rohan to build upon it to enable CPU cloud runners first, because this part should already work.
This PR is refactoring the code to be able to use GKE cloud runners. One of the current requirements of this framework is using containers.
Checklist
Before submitting this PR, please make sure (put X in square brackets):