Skip to content

Commit fe3f3b2

Browse files
committed
feat(ci): add CUDA 12.5 wheel builds
1 parent 4265a67 commit fe3f3b2

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

.github/workflows/build-wheels-cuda.yaml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ on:
2323
- 12.2.2
2424
- 12.3.2
2525
- 12.4.1
26+
- 12.5.1
2627
test_python:
2728
description: Python version to smoke-test
2829
required: true
@@ -65,7 +66,7 @@ jobs:
6566
$os = @($osInput)
6667
}
6768
if ([string]::IsNullOrWhiteSpace($cudaInput) -or $cudaInput -eq 'all') {
68-
$cuda = @("12.1.1", "12.2.2", "12.3.2", "12.4.1") #, "12.5.1", "12.6.1")
69+
$cuda = @("12.1.1", "12.2.2", "12.3.2", "12.4.1", "12.5.1")
6970
} else {
7071
$cuda = @($cudaInput)
7172
}
@@ -143,9 +144,9 @@ jobs:
143144
$cudaVersion = $env:CUDAVER
144145
$cudaChannel = "nvidia/label/cuda-$cudaVersion"
145146
if ($IsLinux) {
146-
$nvccPackage = "${cudaChannel}::cuda-nvcc_linux-64=$cudaVersion"
147+
$nvccPackage = "${cudaChannel}::cuda-nvcc_linux-64"
147148
} elseif ($IsWindows) {
148-
$nvccPackage = "${cudaChannel}::cuda-nvcc_win-64=$cudaVersion"
149+
$nvccPackage = "${cudaChannel}::cuda-nvcc_win-64"
149150
} else {
150151
throw 'Unsupported CUDA wheel build platform'
151152
}

0 commit comments

Comments
 (0)