8000 Update build_cuda.sh · pytorch/pytorch@24546be · GitHub
[go: up one dir, main page]

Skip to content

Commit 24546be

Browse files
authored
Update build_cuda.sh
1 parent 37e7ab4 commit 24546be

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.ci/manywheel/build_cuda.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,7 @@ if [[ $CUDA_VERSION == "12.4" || $CUDA_VERSION == "12.6" ]]; then
184184
'$ORIGIN/../../nvidia/cusparse/lib'
185185
'$ORIGIN/../../cusparselt/lib'
186186
'$ORIGIN/../../nvidia/nccl/lib'
187+
'$ORIGIN/../../nvidia/nvjitlink/lib'
187188
'$ORIGIN/../../nvidia/nvtx/lib'
188189
)
189190
CUDA_RPATHS=$(IFS=: ; echo "${CUDA_RPATHS[*]}")

0 commit comments

Comments
 (0)
0