From f3bab88e1ef0f07d36f4c0c91326a39ea69e1df1 Mon Sep 17 00:00:00 2001
From: Leo Fang <leo80042@gmail.com>
Date: Tue, 7 Apr 2026 19:51:17 -0400
Subject: [PATCH 1/4] Allow matrix.FLAVOR to override runner OS prefix
 (default: linux)

---
 .github/workflows/test-wheel-linux.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test-wheel-linux.yml b/.github/workflows/test-wheel-linux.yml
index cbdae6e7def..df3c3938871 100644
--- a/.github/workflows/test-wheel-linux.yml
+++ b/.github/workflows/test-wheel-linux.yml
@@ -75,7 +75,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix: ${{ fromJSON(needs.compute-matrix.outputs.MATRIX) }}
-    runs-on: "linux-${{ matrix.ARCH }}-gpu-${{ matrix.GPU }}-${{ matrix.DRIVER }}-${{ matrix.GPU_COUNT }}"
+    runs-on: "${{ matrix.FLAVOR || 'linux' }}-${{ matrix.ARCH }}-gpu-${{ matrix.GPU }}-${{ matrix.DRIVER }}-${{ matrix.GPU_COUNT }}"
     # The build stage could fail but we want the CI to keep moving.
     if: ${{ github.repository_owner == 'nvidia' && !cancelled() }}
     # Our self-hosted runners require a container

From 105f10c5d48fa8310557fd14fdfa6cbcf92b2369 Mon Sep 17 00:00:00 2001
From: Leo Fang <leo80042@gmail.com>
Date: Tue, 7 Apr 2026 19:52:34 -0400
Subject: [PATCH 2/4] Add WSL runners to Linux test matrix

---
 ci/test-matrix.yml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ci/test-matrix.yml b/ci/test-matrix.yml
index b1adcee61aa..ee6f8bfe30a 100644
--- a/ci/test-matrix.yml
+++ b/ci/test-matrix.yml
@@ -60,6 +60,9 @@ linux:
     - { ARCH: 'amd64', PY_VER: '3.13',  CUDA_VER: '13.2.0', LOCAL_CTK: '1', GPU: 'h100',       GPU_COUNT: '1', DRIVER: 'latest' }
     - { ARCH: 'amd64', PY_VER: '3.14',  CUDA_VER: '13.2.0', LOCAL_CTK: '1', GPU: 't4',         GPU_COUNT: '2', DRIVER: 'latest' }
     - { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '13.2.0', LOCAL_CTK: '1', GPU: 'h100',       GPU_COUNT: '2', DRIVER: 'latest' }
+    # WSL runners
+    - { ARCH: 'amd64', PY_VER: '3.11',  CUDA_VER: '12.9.1', LOCAL_CTK: '0', GPU: 't4',         GPU_COUNT: '1', DRIVER: 'latest', FLAVOR: 'wsl' }
+    - { ARCH: 'amd64', PY_VER: '3.12',  CUDA_VER: '13.2.0', LOCAL_CTK: '0', GPU: 'rtx4090',    GPU_COUNT: '1', DRIVER: 'latest', FLAVOR: 'wsl' }
   nightly: []
 
 windows:

From e7fb7ef81ad00a8df3d21a02d73975e2775a12fb Mon Sep 17 00:00:00 2001
From: Leo Fang <leo80042@gmail.com>
Date: Tue, 7 Apr 2026 19:55:17 -0400
Subject: [PATCH 3/4] Remove WSL runners comment

---
 ci/test-matrix.yml | 1 -
 1 file changed, 1 deletion(-)

diff --git a/ci/test-matrix.yml b/ci/test-matrix.yml
index ee6f8bfe30a..c4682606073 100644
--- a/ci/test-matrix.yml
+++ b/ci/test-matrix.yml
@@ -60,7 +60,6 @@ linux:
     - { ARCH: 'amd64', PY_VER: '3.13',  CUDA_VER: '13.2.0', LOCAL_CTK: '1', GPU: 'h100',       GPU_COUNT: '1', DRIVER: 'latest' }
     - { ARCH: 'amd64', PY_VER: '3.14',  CUDA_VER: '13.2.0', LOCAL_CTK: '1', GPU: 't4',         GPU_COUNT: '2', DRIVER: 'latest' }
     - { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '13.2.0', LOCAL_CTK: '1', GPU: 'h100',       GPU_COUNT: '2', DRIVER: 'latest' }
-    # WSL runners
     - { ARCH: 'amd64', PY_VER: '3.11',  CUDA_VER: '12.9.1', LOCAL_CTK: '0', GPU: 't4',         GPU_COUNT: '1', DRIVER: 'latest', FLAVOR: 'wsl' }
     - { ARCH: 'amd64', PY_VER: '3.12',  CUDA_VER: '13.2.0', LOCAL_CTK: '0', GPU: 'rtx4090',    GPU_COUNT: '1', DRIVER: 'latest', FLAVOR: 'wsl' }
   nightly: []

From afbd15d390bc2bea390323f53af053b00094e8ce Mon Sep 17 00:00:00 2001
From: Leo Fang <leo80042@gmail.com>
Date: Tue, 7 Apr 2026 19:59:45 -0400
Subject: [PATCH 4/4] Show FLAVOR in job name when set

---
 .github/workflows/test-wheel-linux.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test-wheel-linux.yml b/.github/workflows/test-wheel-linux.yml
index df3c3938871..5d5cef36fa8 100644
--- a/.github/workflows/test-wheel-linux.yml
+++ b/.github/workflows/test-wheel-linux.yml
@@ -70,7 +70,7 @@ jobs:
           echo "OLD_BRANCH=${OLD_BRANCH}" >> "$GITHUB_OUTPUT"
 
   test:
-    name: py${{ matrix.PY_VER }}, ${{ matrix.CUDA_VER }}, ${{ (matrix.LOCAL_CTK == '1' && 'local') || 'wheels' }}, ${{ matrix.GPU }}${{ matrix.GPU_COUNT != '1' && format('(x{0})', matrix.GPU_COUNT) || '' }}
+    name: py${{ matrix.PY_VER }}, ${{ matrix.CUDA_VER }}, ${{ (matrix.LOCAL_CTK == '1' && 'local') || 'wheels' }}, ${{ matrix.GPU }}${{ matrix.GPU_COUNT != '1' && format('(x{0})', matrix.GPU_COUNT) || '' }}${{ matrix.FLAVOR && format(', {0}', matrix.FLAVOR) || '' }}
     needs: compute-matrix
     strategy:
       fail-fast: false