diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 08daac949..7ad0f17e3 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -830,10 +830,7 @@ jobs: ((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true')) ) ) - # `1-gpu-h100-h200` is a shared label advertised by both the 1-GPU H100 and - # 1-GPU H200 runner pools, so this job can land on either. Multimodal_gen - # jobs in pr-test-multimodal-gen.yml continue to pin to `1-gpu-h100`. - runs-on: 1-gpu-h100-h200 + runs-on: 1-gpu-h100 timeout-minutes: 240 strategy: fail-fast: false diff --git a/.github/workflows/rerun-test.yml b/.github/workflows/rerun-test.yml index f918f5f81..1241763f3 100644 --- a/.github/workflows/rerun-test.yml +++ b/.github/workflows/rerun-test.yml @@ -14,7 +14,6 @@ on: type: choice options: - 1-gpu-h100 - - 1-gpu-h100-h200 - 1-gpu-5090 - 2-gpu-h100 - 4-gpu-h100 diff --git a/scripts/ci/utils/slash_command_handler.py b/scripts/ci/utils/slash_command_handler.py index 059f4e86e..76e48233d 100644 --- a/scripts/ci/utils/slash_command_handler.py +++ b/scripts/ci/utils/slash_command_handler.py @@ -478,7 +478,7 @@ CUDA_SUITE_TO_RUNNER = { "stage-a-test-1-gpu-small": "1-gpu-5090", "stage-a-test-cpu": "ubuntu-latest", "stage-b-test-1-gpu-small": "1-gpu-5090", - "stage-b-test-1-gpu-large": "1-gpu-h100-h200", + "stage-b-test-1-gpu-large": "1-gpu-h100", "stage-b-test-2-gpu-large": "2-gpu-h100", "stage-b-test-4-gpu-b200": "4-gpu-b200", "stage-c-test-4-gpu-h100": "4-gpu-h100",