1111
1212env :
1313 PY_VER : " 3.14"
14- CUDA_VER : " 13.1.0"
1514 LOCAL_CTK : " 1"
1615 GPU : " a100"
1716 DRIVER : " latest"
1817
1918jobs :
19+ coverage-vars :
20+ runs-on : ubuntu-latest
21+ outputs :
22+ CUDA_VER : ${{ steps.get-vars.outputs.cuda_ver }}
23+ steps :
24+ - name : Checkout ${{ github.event.repository.name }}
25+ uses : actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
26+ - name : Get CUDA version
27+ id : get-vars
28+ run : |
29+ cuda_ver=$(yq '.cuda.build.version' ci/versions.yml)
30+ echo "cuda_ver=$cuda_ver" >> $GITHUB_OUTPUT
31+
2032 coverage-linux :
2133 name : Coverage (Linux)
34+ needs : [coverage-vars]
2235 runs-on : " linux-amd64-gpu-a100-latest-1"
2336 permissions :
2437 id-token : write
2942 env :
3043 HOST_PLATFORM : " linux-64"
3144 ARCH : " x86_64"
45+ CUDA_VER : ${{ needs.coverage-vars.outputs.CUDA_VER }}
3246 # Our self-hosted runners require a container
3347 # TODO: use a different (nvidia?) container
3448 container :
@@ -164,6 +178,7 @@ jobs:
164178 # Build Windows wheels on GitHub-hosted runner (has VS, no GPU)
165179 build-wheel-windows :
166180 name : Build Wheels (Windows)
181+ needs : [coverage-vars]
167182 runs-on : windows-2022
168183 permissions :
169184 contents : read
@@ -173,6 +188,7 @@ jobs:
173188 env :
174189 HOST_PLATFORM : " win-64"
175190 CUDA_PYTHON_COVERAGE : " 1"
191+ CUDA_VER : ${{ needs.coverage-vars.outputs.CUDA_VER }}
176192 steps :
177193 - name : Checkout ${{ github.event.repository.name }}
178194 uses : actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
@@ -231,7 +247,7 @@ jobs:
231247 # Run coverage tests on self-hosted GPU runner (no VS needed, installs pre-built wheels)
232248 coverage-windows :
233249 name : Coverage (Windows)
234- needs : [build-wheel-windows]
250+ needs : [coverage-vars, build-wheel-windows]
235251 runs-on : " windows-amd64-gpu-a100-latest-1"
236252 permissions :
237253 id-token : write
@@ -240,6 +256,7 @@ jobs:
240256 HOST_PLATFORM : " win-64"
241257 ARCH : " amd64"
242258 CUDA_PYTHON_COVERAGE : " 1"
259+ CUDA_VER : ${{ needs.coverage-vars.outputs.CUDA_VER }}
243260 defaults :
244261 run :
245262 shell : bash --noprofile --norc -xeuo pipefail {0}
0 commit comments