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 cimain
parent
c92b9e11c3
commit
eac6edee96
|
|
@ -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'
|
||||
|
|
|
|||
Loading…
Reference in New Issue