Skip to content

Commit 4df9867

Browse files
committed
add nvrtc
1 parent b79e034 commit 4df9867

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

.github/workflows/paddle-windows-build.yml

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,13 @@ jobs:
4343
cuda: ${{ matrix.cuda }}
4444
# Windows sub-packages are kind of different, so we need to handle them separately
4545
# https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
46-
sub-packages: ${{ matrix.os == 'ubuntu-22.04' && '["nvcc"]' || '["nvcc", "cublas", "cublas_dev", "cudart", "cufft", "cufft_dev", "curand", "curand_dev", "cusparse", "cusparse_dev", "cusolver", "cusolver_dev"]' }}
47-
non-cuda-sub-packages: ${{ matrix.os == 'ubuntu-22.04' && '["libcublas", "libcublas-dev", "libcufft", "libcufft-dev", "libcurand", "libcurand-dev", "libcusparse", "libcusparse-dev", "libcusolver", "libcusolver-dev"]' || '[]' }}
46+
sub-packages: ${{ matrix.os == 'ubuntu-22.04' &&
47+
'["nvcc"]' ||
48+
'["nvcc", "cublas", "cublas_dev", "cudart", "cufft", "cufft_dev", "curand", "curand_dev", "cusparse", "cusparse_dev", "cusolver", "cusolver_dev", "nvrtc", "nvrtc_dev"]'
49+
}}
50+
non-cuda-sub-packages: ${{ matrix.os == 'ubuntu-22.04' &&
51+
'["libcublas", "libcublas-dev", "libcufft", "libcufft-dev", "libcurand", "libcurand-dev", "libcusparse", "libcusparse-dev", "libcusolver", "libcusolver-dev", "libnvrtc", "libnvrtc-dev"]' ||
52+
'[]' }}
4853
log-file-suffix: '${{matrix.os}}.txt'
4954

5055
- name: Download cuDNN

0 commit comments

Comments
 (0)