Skip to content

Commit bfa1d5b

Browse files
nWEIdiaatalman
authored andcommitted
Add missing changes to cu121/cu124
1 parent ec4ee83 commit bfa1d5b

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

manywheel/build_cuda.sh

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,11 @@ if [[ $CUDA_VERSION == "12.1" || $CUDA_VERSION == "12.4" ]]; then
151151
DEPS_LIST+=(
152152
"/usr/local/cuda/lib64/libcudnn_adv.so.9"
153153
"/usr/local/cuda/lib64/libcudnn_cnn.so.9"
154+
"/usr/local/cuda/lib64/libcudnn_graph.so.9"
154155
"/usr/local/cuda/lib64/libcudnn_ops.so.9"
156+
"/usr/local/cuda/lib64/libcudnn_engines_runtime_compiled.so.9"
157+
"/usr/local/cuda/lib64/libcudnn_engines_precompiled.so.9"
158+
"/usr/local/cuda/lib64/libcudnn_heuristic.so.9"
155159
"/usr/local/cuda/lib64/libcudnn.so.9"
156160
"/usr/local/cuda/lib64/libcublas.so.12"
157161
"/usr/local/cuda/lib64/libcublasLt.so.12"
@@ -163,7 +167,11 @@ if [[ $CUDA_VERSION == "12.1" || $CUDA_VERSION == "12.4" ]]; then
163167
DEPS_SONAME+=(
164168
"libcudnn_adv.so.9"
165169
"libcudnn_cnn.so.9"
170+
"libcudnn_graph.so.9"
166171
"libcudnn_ops.so.9"
172+
"libcudnn_engines_runtime_compiled.so.9"
173+
"libcudnn_engines_precompiled.so.9"
174+
"libcudnn_heuristic.so.9"
167175
"libcudnn.so.9"
168176
"libcublas.so.12"
169177
"libcublasLt.so.12"

0 commit comments

Comments
 (0)