From fd176d6c5c3caf929375af1b75fdb83fac481da0 Mon Sep 17 00:00:00 2001 From: Zain Rizvi Date: Fri, 13 Sep 2024 16:54:17 -0500 Subject: [PATCH] More runner fixes --- .github/workflows/gpu-hvd-tests.yml | 2 +- .github/workflows/gpu-tests.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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: