More runner fixes

ZainRizvi · ZainRizvi · commit fd176d6c5c3c · 2024-09-13T16:54:17.000-05:00
diff --git 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
@@ -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:
