From 1e30e4114d4cd664efc98f30eac76e7e773bce68 Mon Sep 17 00:00:00 2001 From: Leo Fang Date: Thu, 30 Jan 2025 11:58:36 -0500 Subject: [PATCH] update to use NVKS runners --- .github/workflows/build-docs.yml | 2 +- .github/workflows/test-wheel.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build-docs.yml b/.github/workflows/build-docs.yml index 72ef2474c..d0801870d 100644 --- a/.github/workflows/build-docs.yml +++ b/.github/workflows/build-docs.yml @@ -38,7 +38,7 @@ jobs: # The build stage could fail but we want the CI to keep moving. if: ${{ github.repository_owner == 'nvidia' && !cancelled() }} # WAR: Building the doc currently requires a GPU (NVIDIA/cuda-python#326,327) - runs-on: linux-amd64-gpu-t4-latest-1-testing + runs-on: linux-amd64-gpu-t4-latest-1 #runs-on: ubuntu-latest defaults: run: diff --git a/.github/workflows/test-wheel.yml b/.github/workflows/test-wheel.yml index 7e911640e..c15de07e5 100644 --- a/.github/workflows/test-wheel.yml +++ b/.github/workflows/test-wheel.yml @@ -28,7 +28,7 @@ jobs: if: ${{ github.repository_owner == 'nvidia' && !cancelled() }} runs-on: ${{ (inputs.runner == 'default' && inputs.host-platform == 'linux-64' && 'linux-amd64-gpu-v100-latest-1') || (inputs.runner == 'default' && inputs.host-platform == 'linux-aarch64' && 'linux-arm64-gpu-a100-latest-1') || - (inputs.runner == 'H100' && 'linux-amd64-gpu-h100-latest-1-testing') }} + (inputs.runner == 'H100' && 'linux-amd64-gpu-h100-latest-1') }} # Our self-hosted runners require a container # TODO: use a different (nvidia?) container container: