Skip to content

Commit 2a11961

Browse files
cryoscryos
committed
Get CUDA build version from JSON, docs and checks, full matrix
1 parent d2d9def commit 2a11961

4 files changed

Lines changed: 111 additions & 10 deletions

File tree

‎.github/workflows/build.yml‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@ jobs:
2222
matrix:
2323
python-version:
2424
- "3.9"
25+
- "3.10"
26+
- "3.11"
2527
- "3.12"
2628
- "3.13"
2729
name: py${{ matrix.python-version }}

‎.github/workflows/ci.yml‎

Lines changed: 54 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,7 @@
11
name: "CI"
22

33
concurrency:
4-
group: ${{ github.workflow }}-${{
5-
github.ref_name == 'main' && format('ci-main-build-test-{0}', github.run_id) ||
6-
format('ci-pr-build-test-on-{0}-against-branch-{1}', github.event_name, github.ref_name)
7-
}}
4+
group: ${{ github.workflow }}-${{ github.ref }}-${{ github.event_name }}
85
cancel-in-progress: true
96

107
on:
@@ -14,22 +11,39 @@ on:
1411
- "main"
1512

1613
jobs:
14+
ci-vars:
15+
runs-on: ubuntu-latest
16+
outputs:
17+
CUDA_BUILD_VER: ${{ steps.get-vars.outputs.cuda_build_ver }}
18+
steps:
19+
- name: Checkout repository
20+
uses: actions/checkout@v4
21+
with:
22+
fetch-depth: 0
23+
- name: Get CUDA build version
24+
id: get-vars
25+
run: |
26+
cuda_build_ver=$(jq -r .cuda.build.version versions.json)
27+
echo "cuda_build_ver=$cuda_build_ver" >> $GITHUB_OUTPUT
28+
1729
build:
30+
needs:
31+
- ci-vars
1832
strategy:
1933
fail-fast: false
2034
matrix:
2135
host-platform:
2236
- linux-64
2337
- linux-aarch64
2438
- win-64
25-
name: Build ${{ matrix.host-platform }}, CUDA 12.8.0
39+
name: Build ${{ matrix.host-platform }}, CUDA ${{ needs.ci-vars.outputs.CUDA_BUILD_VER }}
2640
if: ${{ github.repository_owner == 'nvidia' }}
2741
secrets: inherit
2842
uses:
2943
./.github/workflows/build.yml
3044
with:
3145
host-platform: ${{ matrix.host-platform }}
32-
cuda-version: "12.8.0"
46+
cuda-version: ${{ needs.ci-vars.outputs.CUDA_BUILD_VER }}
3347

3448
test-linux:
3549
strategy:
@@ -43,14 +57,15 @@ jobs:
4357
permissions:
4458
contents: read # This is required for actions/checkout
4559
needs:
60+
- ci-vars
4661
- build
4762
secrets: inherit
4863
uses:
4964
./.github/workflows/test-wheel-linux.yml
5065
with:
5166
build-type: pull-request
5267
host-platform: ${{ matrix.host-platform }}
53-
build-ctk-ver: 12.8.0
68+
build-ctk-ver: ${{ needs.ci-vars.outputs.CUDA_BUILD_VER }}
5469

5570
test-windows:
5671
strategy:
@@ -63,11 +78,42 @@ jobs:
6378
permissions:
6479
contents: read # This is required for actions/checkout
6580
needs:
81+
- ci-vars
6682
- build
6783
secrets: inherit
6884
uses:
6985
./.github/workflows/test-wheel-windows.yml
7086
with:
7187
build-type: pull-request
7288
host-platform: ${{ matrix.host-platform }}
73-
build-ctk-ver: 12.8.0
89+
build-ctk-ver: ${{ needs.ci-vars.outputs.CUDA_BUILD_VER }}
90+
91+
doc:
92+
name: Docs
93+
if: ${{ github.repository_owner == 'nvidia' }}
94+
# Sets permissions of the GITHUB_TOKEN to allow deployment to GitHub Pages
95+
permissions:
96+
id-token: write
97+
contents: write
98+
pull-requests: write
99+
needs:
100+
- ci-vars
101+
- build
102+
secrets: inherit
103+
uses:
104+
./.github/workflows/build-docs.yml
105+
with:
106+
build-ctk-ver: ${{ needs.ci-vars.outputs.CUDA_BUILD_VER }}
107+
108+
checks:
109+
name: Check job status
110+
permissions:
111+
checks: read
112+
needs:
113+
- build
114+
- test-linux
115+
- test-windows
116+
- doc
117+
secrets: inherit
118+
uses:
119+
./.github/workflows/status-check.yml

‎.github/workflows/test-wheel-linux.yml‎

Lines changed: 48 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,13 +58,59 @@ jobs:
5858
#
5959
export MATRICES="
6060
pull-request:
61-
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
61+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
62+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
6263
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
64+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
65+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
66+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
67+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
68+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
69+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
70+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
71+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
72+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
73+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
74+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
75+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
76+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
77+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
78+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
79+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
80+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
81+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
82+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
83+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
6384
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
85+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
6486
${special_runner}
6587
nightly:
66-
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
88+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
89+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
90+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
91+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
92+
- { ARCH: ${ARCH}, PY_VER: '3.9', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
93+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
94+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
95+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
96+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
97+
- { ARCH: ${ARCH}, PY_VER: '3.10', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
98+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
99+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
100+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
101+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
102+
- { ARCH: ${ARCH}, PY_VER: '3.11', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
103+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
104+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
105+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
106+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
107+
- { ARCH: ${ARCH}, PY_VER: '3.12', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
108+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '11.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'earliest' }
109+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '11.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
110+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '12.0.1', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'latest' }
67111
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '12.8.0', LOCAL_CTK: '0', GPU: ${gpu}, DRIVER: 'latest' }
112+
- { ARCH: ${ARCH}, PY_VER: '3.13', CUDA_VER: '12.8.0', LOCAL_CTK: '1', GPU: ${gpu}, DRIVER: 'earliest' }
113+
${special_runner}
68114
"
69115
70116
# Use the nightly matrix for branch tests

‎versions.json‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
{
2+
"cuda": {
3+
"build": {
4+
"version": "12.8.0"
5+
}
6+
}
7+
}

0 commit comments

Comments
 (0)