[DO NOT MERGE] Demonstrate how to explicity opt-in to Amazon 2 AMI

pytorch · Aug 14, 2024 · 4121202 · 4121202
1 parent 09e2690
commit 4121202
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/build-docs.yml b/.github/workflows/build-docs.yml
@@ -17,7 +17,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       job-name: Build doc
-      runner: linux.2xlarge
+      runner: amz2.linux.2xlarge
       repository: pytorch/text
       gpu-arch-type: cpu
       timeout: 120

diff --git a/.github/workflows/test-linux-cpu.yml b/.github/workflows/test-linux-cpu.yml
@@ -20,7 +20,7 @@ jobs:
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
-      runner: linux.12xlarge
+      runner: amz2.linux.12xlarge
       repository: pytorch/text
       script: |
         # Mark Build Directory Safe

diff --git a/.github/workflows/test-linux-gpu.yml b/.github/workflows/test-linux-gpu.yml
@@ -21,7 +21,7 @@ jobs:
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
-      runner: linux.g5.4xlarge.nvidia.gpu
+      runner: amz2.linux.g5.4xlarge.nvidia.gpu
       repository: pytorch/text
       gpu-arch-type: cuda
       gpu-arch-version: ${{ matrix.cuda_arch_version }}