[FindCUDNN] fix unnecessary message and dll symbol

pull/2952/head
Stefano Sinigardi 6 years ago
parent ff7d920a5d
commit eebe735f84
  1. 8
      cmake/Modules/FindCUDNN.cmake

@ -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}"

Loading…
Cancel
Save