From eac6edee9614b5ece144e06e2e630e5cc6229458 Mon Sep 17 00:00:00 2001 From: Eric Date: Mon, 11 Dec 2023 15:23:25 +0800 Subject: [PATCH] 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 --- .github/workflows/release.yml | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 39dd532..87b0d03 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -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'