From 3a059ce9af6eeb7a449a0652e63e60f99dc1d940 Mon Sep 17 00:00:00 2001
From: Zain Rizvi <ZainRizvi@users.noreply.github.com>
Date: Sat, 14 Sep 2024 05:06:56 -0500
Subject: [PATCH] Use the default runner type (#3285)

* Use the default runner type

During the Linux Amazon 2023 AMI (base OS) migration these jobs had been temporarily shifted to use the new experimental AMI.

Now that the new AMI has become the base AMI, shifting these back to use the default runners

* More runner fixes
---
 .github/workflows/docker-build.yml  | 8 ++++----
 .github/workflows/gpu-hvd-tests.yml | 2 +-
 .github/workflows/gpu-tests.yml     | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index 54db85d7af1..aadccaafb33 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -56,7 +56,7 @@ jobs:
     env:
       REPOSITORY: ${{ github.repository }}
       PR_NUMBER: ${{ github.event.pull_request.number }}
-    runs-on: amz2023.linux.4xlarge
+    runs-on: linux.4xlarge
     steps:
       - name: Clean workspace
         run: |
@@ -116,7 +116,7 @@ jobs:
     env:
       REPOSITORY: ${{ github.repository }}
       PR_NUMBER: ${{ github.event.pull_request.number }}
-    runs-on: amz2023.linux.12xlarge
+    runs-on: linux.12xlarge
     steps:
       - name: Clean workspace
         run: |
@@ -176,7 +176,7 @@ jobs:
     env:
       REPOSITORY: ${{ github.repository }}
       PR_NUMBER: ${{ github.event.pull_request.number }}
-    runs-on: amz2023.linux.4xlarge
+    runs-on: linux.4xlarge
     steps:
       - name: Clean workspace
         run: |
@@ -236,7 +236,7 @@ jobs:
     env:
       REPOSITORY: ${{ github.repository }}
       PR_NUMBER: ${{ github.event.pull_request.number }}
-    runs-on: amz2023.linux.12xlarge
+    runs-on: linux.12xlarge
     steps:
       - name: Clean workspace
         run: |
diff --git a/.github/workflows/gpu-hvd-tests.yml b/.github/workflows/gpu-hvd-tests.yml
index 3f0dd5ad2c6..2017cf8acda 100644
--- a/.github/workflows/gpu-hvd-tests.yml
+++ b/.github/workflows/gpu-hvd-tests.yml
@@ -28,7 +28,7 @@ jobs:
       DOCKER_IMAGE: "pytorch/conda-builder:cuda12.1"
       REPOSITORY: ${{ github.repository }}
       PR_NUMBER: ${{ github.event.pull_request.number }}
-    runs-on: amz2023.linux.8xlarge.nvidia.gpu
+    runs-on: linux.8xlarge.nvidia.gpu
     timeout-minutes: 60
 
     steps:
diff --git a/.github/workflows/gpu-tests.yml b/.github/workflows/gpu-tests.yml
index 094f20ed225..81862e1f67b 100644
--- a/.github/workflows/gpu-tests.yml
+++ b/.github/workflows/gpu-tests.yml
@@ -28,7 +28,7 @@ jobs:
       DOCKER_IMAGE: "pytorch/conda-builder:cuda12.1"
       REPOSITORY: ${{ github.repository }}
       PR_NUMBER: ${{ github.event.pull_request.number }}
-    runs-on: amz2023.linux.8xlarge.nvidia.gpu
+    runs-on: linux.8xlarge.nvidia.gpu
     timeout-minutes: 85
 
     steps: