mirror of https://github.com/AlexeyAB/darknet.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
3.3 KiB
94 lines
3.3 KiB
# Distributed under the OSI-approved BSD 3-Clause License. |
|
# Copyright Stefano Sinigardi |
|
|
|
#.rst: |
|
# FindCUDNN |
|
# -------- |
|
# |
|
# Result Variables |
|
# ^^^^^^^^^^^^^^^^ |
|
# |
|
# This module will set the following variables in your project:: |
|
# |
|
# ``CUDNN_FOUND`` |
|
# True if CUDNN found on the local system |
|
# |
|
# ``CUDNN_INCLUDE_DIRS`` |
|
# Location of CUDNN header files. |
|
# |
|
# ``CUDNN_LIBRARIES`` |
|
# The CUDNN libraries. |
|
# |
|
|
|
include(FindPackageHandleStandardArgs) |
|
|
|
if(NOT CUDNN_INCLUDE_DIR) |
|
find_path(CUDNN_INCLUDE_DIR cudnn.h |
|
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn} $ENV{CUDNN} |
|
PATH_SUFFIXES cuda/include include) |
|
endif() |
|
|
|
if(NOT CUDNN_LIBRARY) |
|
find_library(CUDNN_LIBRARY cudnn |
|
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn} $ENV{CUDNN} |
|
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64) |
|
endif() |
|
|
|
find_package_handle_standard_args( |
|
CUDNN DEFAULT_MSG CUDNN_INCLUDE_DIR CUDNN_LIBRARY) |
|
|
|
if(CUDNN_FOUND) |
|
file(READ ${CUDNN_INCLUDE_DIR}/cudnn.h CUDNN_HEADER_CONTENTS) |
|
string(REGEX MATCH "define CUDNN_MAJOR * +([0-9]+)" |
|
CUDNN_VERSION_MAJOR "${CUDNN_HEADER_CONTENTS}") |
|
string(REGEX REPLACE "define CUDNN_MAJOR * +([0-9]+)" "\\1" |
|
CUDNN_VERSION_MAJOR "${CUDNN_VERSION_MAJOR}") |
|
string(REGEX MATCH "define CUDNN_MINOR * +([0-9]+)" |
|
CUDNN_VERSION_MINOR "${CUDNN_HEADER_CONTENTS}") |
|
string(REGEX REPLACE "define CUDNN_MINOR * +([0-9]+)" "\\1" |
|
CUDNN_VERSION_MINOR "${CUDNN_VERSION_MINOR}") |
|
string(REGEX MATCH "define CUDNN_PATCHLEVEL * +([0-9]+)" |
|
CUDNN_VERSION_PATCH "${CUDNN_HEADER_CONTENTS}") |
|
string(REGEX REPLACE "define CUDNN_PATCHLEVEL * +([0-9]+)" "\\1" |
|
CUDNN_VERSION_PATCH "${CUDNN_VERSION_PATCH}") |
|
if(NOT CUDNN_VERSION_MAJOR) |
|
set(CUDNN_VERSION "?") |
|
else() |
|
set(CUDNN_VERSION "${CUDNN_VERSION_MAJOR}.${CUDNN_VERSION_MINOR}.${CUDNN_VERSION_PATCH}") |
|
endif() |
|
endif() |
|
|
|
set(CUDNN_INCLUDE_DIRS ${CUDNN_INCLUDE_DIR}) |
|
set(CUDNN_LIBRARIES ${CUDNN_LIBRARY}) |
|
if(CUDNN_FOUND) |
|
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: |
|
# 1. If we can find a Windows .dll file (or if we can find both Debug and |
|
# Release libraries), we will set appropriate target properties for these. |
|
# 2. However, for most systems, we will only register the import location and |
|
# include directory. |
|
|
|
# Look for dlls, for Release and Debug libraries. |
|
if(WIN32) |
|
string( REPLACE ".lib" ".dll" CUDNN_LIBRARY_DLL "${CUDNN_LIBRARY}" ) |
|
endif() |
|
|
|
if( CUDNN_FOUND AND NOT TARGET CuDNN::CuDNN ) |
|
if( EXISTS "${CUDNN_LIBRARY_DLL}" ) |
|
add_library( CuDNN::CuDNN SHARED IMPORTED ) |
|
set_target_properties( CuDNN::CuDNN PROPERTIES |
|
IMPORTED_LOCATION "${CUDNN_LIBRARY_DLL}" |
|
IMPORTED_IMPLIB "${CUDNN_LIBRARY}" |
|
INTERFACE_INCLUDE_DIRECTORIES "${CUDNN_INCLUDE_DIR}" |
|
IMPORTED_LINK_INTERFACE_LANGUAGES "C" ) |
|
else() |
|
add_library( CuDNN::CuDNN UNKNOWN IMPORTED ) |
|
set_target_properties( CuDNN::CuDNN PROPERTIES |
|
IMPORTED_LOCATION "${CUDNN_LIBRARY}" |
|
INTERFACE_INCLUDE_DIRECTORIES "${CUDNN_INCLUDE_DIR}" |
|
IMPORTED_LINK_INTERFACE_LANGUAGES "C" ) |
|
endif() |
|
endif()
|
|
|