Turn off NCCL_STATIC/CUDNN_STATIC when OF_CUDA_LINK_DYNAMIC_LIBRARY i…

…s ON (#5243)
Oneflow-Inc · Jun 19, 2021 · 360ba56 · 360ba56
1 parent 9c1b19c
commit 360ba56
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cmake/third_party/FindCUDNN.cmake b/cmake/third_party/FindCUDNN.cmake
@@ -14,7 +14,7 @@ include(FindPackageHandleStandardArgs)
 set(CUDNN_ROOT_DIR "" CACHE PATH "Folder contains NVIDIA cuDNN")
 
 option(CUDNN_STATIC "Look for static cuDNN" ON)
-if(BUILD_SHARED_LIBS)
+if(OF_CUDA_LINK_DYNAMIC_LIBRARY)
    set(CUDNN_STATIC OFF)
 endif()
 if (CUDNN_STATIC)

diff --git a/cmake/third_party/nccl.cmake b/cmake/third_party/nccl.cmake
@@ -1,6 +1,9 @@
 include (ExternalProject)
 
 option(NCCL_STATIC "" ON)
+if(OF_CUDA_LINK_DYNAMIC_LIBRARY)
+   set(NCCL_STATIC OFF)
+endif()
 
 set(NCCL_INSTALL_DIR ${THIRD_PARTY_DIR}/nccl)
 set(NCCL_INCLUDE_DIR ${NCCL_INSTALL_DIR}/include)