mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

[FindCUDNN] fix unnecessary message and dll symbol

Browse Source
This commit is contained in:
Stefano Sinigardi
2019-04-18 09:17:31 +02:00
parent ff7d920a5d
commit eebe735f84
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cmake/Modules/FindCUDNN.cmake
View File

@ -60,8 +60,10 @@ endif()
set(CUDNN_INCLUDE_DIRS ${CUDNN_INCLUDE_DIR}) set(CUDNN_INCLUDE_DIRS ${CUDNN_INCLUDE_DIR})
set(CUDNN_LIBRARIES ${CUDNN_LIBRARY}) set(CUDNN_LIBRARIES ${CUDNN_LIBRARY})
message(STATUS "Found cuDNN: v${CUDNN_VERSION} (include: ${CUDNN_INCLUDE_DIR}, library: ${CUDNN_LIBRARY})") if(CUDNN_FOUND)
mark_as_advanced(CUDNN_ROOT_DIR CUDNN_LIBRARY CUDNN_INCLUDE_DIR) message(STATUS "Found cuDNN: v${CUDNN_VERSION} (include: ${CUDNN_INCLUDE_DIR}, library: ${CUDNN_LIBRARY})")
endif()
mark_as_advanced(CUDNN_LIBRARY CUDNN_INCLUDE_DIR)
# Register imported libraries: # Register imported libraries:
# 1. If we can find a Windows .dll file (or if we can find both Debug and # 1. If we can find a Windows .dll file (or if we can find both Debug and
@ -75,7 +77,7 @@ if(WIN32)
endif() endif()
if( CUDNN_FOUND AND NOT TARGET CuDNN::CuDNN ) if( CUDNN_FOUND AND NOT TARGET CuDNN::CuDNN )
if( EXISTS "${CUDNN_LIBRARY_RELEASE_DLL}" ) if( EXISTS "${CUDNN_LIBRARY_DLL}" )
add_library( CuDNN::CuDNN SHARED IMPORTED ) add_library( CuDNN::CuDNN SHARED IMPORTED )
set_target_properties( CuDNN::CuDNN PROPERTIES set_target_properties( CuDNN::CuDNN PROPERTIES
IMPORTED_LOCATION "${CUDNN_LIBRARY_DLL}" IMPORTED_LOCATION "${CUDNN_LIBRARY_DLL}"