Skip to content

Commit 914a0a5

Browse files
committed
Configure jobs to use better machine
1 parent b821962 commit 914a0a5

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

.github/workflows/build_and_test.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ jobs:
4949
torch-commit: ${{needs.get-torch-commit.outputs.torch_commit}}
5050
timeout-minutes: 240
5151
has_code_changes: ${{ needs.check_code_changes.outputs.has_code_changes }}
52+
runner: linux.24xlarge
5253
secrets:
5354
gcloud-service-key: ${{ secrets.GCLOUD_SERVICE_KEY }}
5455

@@ -73,6 +74,7 @@ jobs:
7374
with:
7475
dev-image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:3.10_cuda_12.3
7576
has_code_changes: ${{ needs.check_code_changes.outputs.has_code_changes }}
77+
runner: linux.24xlarge
7678
secrets:
7779
gcloud-service-key: ${{ secrets.GCLOUD_SERVICE_KEY }}
7880

@@ -84,6 +86,7 @@ jobs:
8486
dev-image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:3.10_tpuvm
8587
timeout-minutes: 120
8688
collect-coverage: false
89+
runner: linux.24xlarge
8790
torch-commit: ${{needs.get-torch-commit.outputs.torch_commit}}
8891
has_code_changes: ${{ needs.check_code_changes.outputs.has_code_changes }}
8992
secrets:

0 commit comments

Comments
 (0)