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 }}
|
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'
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue