fix: add cuda suffix for win build, add cuda 12 support (#1019)

* fix: add cuda suffix for win build, add cuda 12 support

* resolve comment

* fix ci
main
Eric 2023-12-11 15:23:25 +08:00 committed by GitHub
parent c92b9e11c3
commit eac6edee96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 3 deletions

View File

@ -26,7 +26,8 @@ jobs:
container: ${{ matrix.container }}
strategy:
matrix:
binary: [aarch64-apple-darwin, x86_64-manylinux2014, x86_64-manylinux2014-cuda117, x86_64-pc-windows-msvc]
binary: [aarch64-apple-darwin, x86_64-manylinux2014, x86_64-manylinux2014-cuda117,
x86_64-windows-msvc-cuda117, x86_64-windows-msvc-cuda122]
include:
- os: macos-latest
target: aarch64-apple-darwin
@ -42,9 +43,16 @@ jobs:
build_args: --features cuda
- os: windows-latest
target: x86_64-pc-windows-msvc
binary: x86_64-pc-windows-msvc
binary: x86_64-windows-msvc-cuda117
ext: .exe
build_args: --features cuda
windows_cuda: '11.7.1'
- os: windows-latest
target: x86_64-pc-windows-msvc
binary: x86_64-windows-msvc-cuda122
ext: .exe
build_args: --features cuda
windows_cuda: '12.2.0'
env:
SCCACHE_GHA_ENABLED: true
@ -93,7 +101,7 @@ jobs:
- name: Install CUDA toolkit for Windows
uses: Jimver/cuda-toolkit@v0.2.11
with:
cuda: '11.7.1'
cuda: ${{ matrix.windows_cuda }}
method: 'network'
sub-packages: '["nvcc", "cudart", "cublas", "cublas_dev", "thrust", "visual_studio_integration"]'
if: runner.os == 'Windows'