Skip to content

Commit bc04e81

Browse files
authored
[CUDA] Not link CUDNN sub libs (microsoft#23656)
### Description Do not link cuDNN sub libs. Before: ``` objdump -p libonnxruntime_providers_cuda.so | grep NEEDED NEEDED libcublasLt.so.12 NEEDED libcublas.so.12 NEEDED libcurand.so.10 NEEDED libcufft.so.11 NEEDED libcudart.so.12 NEEDED libcudnn.so.9 NEEDED libcudnn_adv.so.9 NEEDED libcudnn_ops.so.9 NEEDED libcudnn_cnn.so.9 NEEDED libcudnn_graph.so.9 NEEDED libcudnn_engines_runtime_compiled.so.9 NEEDED libcudnn_engines_precompiled.so.9 NEEDED libcudnn_heuristic.so.9 NEEDED libdl.so.2 NEEDED librt.so.1 NEEDED libnvrtc.so.12 NEEDED libpthread.so.0 NEEDED libstdc++.so.6 NEEDED libm.so.6 NEEDED libgcc_s.so.1 NEEDED libc.so.6 NEEDED ld-linux-x86-64.so.2 ``` After: ``` $ objdump -p libonnxruntime_providers_cuda.so | grep NEEDED NEEDED libcublasLt.so.12 NEEDED libcublas.so.12 NEEDED libcurand.so.10 NEEDED libcufft.so.11 NEEDED libcudart.so.12 NEEDED libcudnn.so.9 NEEDED libnvrtc.so.12 NEEDED libstdc++.so.6 NEEDED libm.so.6 NEEDED libgcc_s.so.1 NEEDED libc.so.6 NEEDED ld-linux-x86-64.so.2 ``` ### Motivation and Context Avoid direct dependency on cuDNN sub libraries, which were introduced in microsoft#19470. microsoft#23643
1 parent 8b77007 commit bc04e81

File tree

2 files changed

+0
-53
lines changed

2 files changed

+0
-53
lines changed

cmake/external/cuDNN.cmake

Lines changed: 0 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -67,43 +67,3 @@ target_link_libraries(
6767
INTERFACE
6868
CUDNN::cudnn
6969
)
70-
71-
if(CUDNN_MAJOR_VERSION EQUAL 8)
72-
find_cudnn_library(cudnn_adv_infer)
73-
find_cudnn_library(cudnn_adv_train)
74-
find_cudnn_library(cudnn_cnn_infer)
75-
find_cudnn_library(cudnn_cnn_train)
76-
find_cudnn_library(cudnn_ops_infer)
77-
find_cudnn_library(cudnn_ops_train)
78-
79-
target_link_libraries(
80-
CUDNN::cudnn_all
81-
INTERFACE
82-
CUDNN::cudnn_adv_train
83-
CUDNN::cudnn_ops_train
84-
CUDNN::cudnn_cnn_train
85-
CUDNN::cudnn_adv_infer
86-
CUDNN::cudnn_cnn_infer
87-
CUDNN::cudnn_ops_infer
88-
)
89-
elseif(CUDNN_MAJOR_VERSION EQUAL 9)
90-
find_cudnn_library(cudnn_cnn)
91-
find_cudnn_library(cudnn_adv)
92-
find_cudnn_library(cudnn_graph)
93-
find_cudnn_library(cudnn_ops)
94-
find_cudnn_library(cudnn_engines_runtime_compiled)
95-
find_cudnn_library(cudnn_engines_precompiled)
96-
find_cudnn_library(cudnn_heuristic)
97-
98-
target_link_libraries(
99-
CUDNN::cudnn_all
100-
INTERFACE
101-
CUDNN::cudnn_adv
102-
CUDNN::cudnn_ops
103-
CUDNN::cudnn_cnn
104-
CUDNN::cudnn_graph
105-
CUDNN::cudnn_engines_runtime_compiled
106-
CUDNN::cudnn_engines_precompiled
107-
CUDNN::cudnn_heuristic
108-
)
109-
endif()

setup.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -213,19 +213,6 @@ def run(self):
213213
"libcufft.so.10",
214214
"libcufft.so.11",
215215
"libcurand.so.10",
216-
"libcudnn_adv_infer.so.8",
217-
"libcudnn_adv_train.so.8",
218-
"libcudnn_cnn_infer.so.8",
219-
"libcudnn_cnn_train.so.8",
220-
"libcudnn_ops_infer.so.8",
221-
"libcudnn_ops_train.so.8",
222-
"libcudnn_adv.so.9",
223-
"libcudnn_cnn.so.9",
224-
"libcudnn_engines_precompiled.so.9",
225-
"libcudnn_engines_runtime_compiled.so.9",
226-
"libcudnn_graph.so.9",
227-
"libcudnn_heuristic.so.9",
228-
"libcudnn_ops.so.9",
229216
"libnvJitLink.so.12",
230217
"libnvrtc.so.11.2", # A symlink to libnvrtc.so.11.8.89
231218
"libnvrtc.so.12",

0 commit comments

Comments
 (0)