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