From a6da3a4922b46277dfab81c4378ada0e70368f55 Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Wed, 2 Sep 2026 16:15:40 -0700 Subject: [PATCH] [CI] Re-enable GB300 jobs (#37522) --- .github/workflows/_pr-test-stage.yml | 2 -- .github/workflows/release-whl-deepgemm.yml | 7 +++---- .github/workflows/rerun-test.yml | 6 ++---- 3 files changed, 5 insertions(+), 10 deletions(-) diff --git a/.github/workflows/_pr-test-stage.yml b/.github/workflows/_pr-test-stage.yml index c29402e0b..ae4165c75 100644 --- a/.github/workflows/_pr-test-stage.yml +++ b/.github/workflows/_pr-test-stage.yml @@ -100,14 +100,12 @@ jobs: run: name: ${{ inputs.self_name }} (${{ matrix.partition }}) # Runs on schedule / parallel-dispatch / non-failed PR with main_package or sgl_kernel changes. - # Temporarily skip the broken GB300 runner before GitHub tries to allocate it. # `runner_filter` is matched here rather than in each caller's job `if`: a # caller that declares its runners as a matrix cannot, since job-level `if` # sees github/needs/vars/inputs but not `matrix`. Callers that never declare # the input read back null and match every runner_config. if: | always() && - inputs.runner_config != '4-gpu-gb300' && (fromJson(inputs.caller_inputs).runner_filter == null || fromJson(inputs.caller_inputs).runner_filter == '' || fromJson(inputs.caller_inputs).runner_filter == 'all' || diff --git a/.github/workflows/release-whl-deepgemm.yml b/.github/workflows/release-whl-deepgemm.yml index b10c6a1d9..b321bc946 100644 --- a/.github/workflows/release-whl-deepgemm.yml +++ b/.github/workflows/release-whl-deepgemm.yml @@ -194,10 +194,9 @@ jobs: # - arch_label: sm120 # runner: 1-gpu-5090 # wheel_arch: x86_64 - # Temporarily disabled while the GB300 runner is unavailable. - # - arch_label: sm100-aarch64 - # runner: 4-gpu-gb300 - # wheel_arch: aarch64 + - arch_label: sm100-aarch64 + runner: 4-gpu-gb300 + wheel_arch: aarch64 runs-on: ${{ matrix.runner }} timeout-minutes: 120 steps: diff --git a/.github/workflows/rerun-test.yml b/.github/workflows/rerun-test.yml index 78d7be44f..cb0fda3de 100644 --- a/.github/workflows/rerun-test.yml +++ b/.github/workflows/rerun-test.yml @@ -87,8 +87,7 @@ permissions: jobs: rerun-test-cuda: - # Temporarily reject reruns targeting the broken GB300 runner. - if: inputs.mode == 'cuda' && inputs.runs_on != '4-gpu-gb300' + if: inputs.mode == 'cuda' runs-on: ${{ inputs.runs_on }} timeout-minutes: 120 permissions: @@ -200,8 +199,7 @@ jobs: if: failure() rerun-test-multimodal-gen: - # Temporarily reject reruns targeting the broken GB300 runner. - if: inputs.mode == 'multimodal_gen' && inputs.runs_on != '4-gpu-gb300' + if: inputs.mode == 'multimodal_gen' runs-on: ${{ inputs.runs_on }} timeout-minutes: 120 permissions: