| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
|
|
| include(FindPackageHandleStandardArgs) |
|
|
| set(CUDNN_ROOT $ENV{CUDNN_ROOT_DIR} CACHE PATH "Folder containing NVIDIA cuDNN") |
| if (DEFINED $ENV{CUDNN_ROOT_DIR}) |
| message(WARNING "CUDNN_ROOT_DIR is deprecated. Please set CUDNN_ROOT instead.") |
| endif() |
| list(APPEND CUDNN_ROOT $ENV{CUDNN_ROOT_DIR} ${CUDA_TOOLKIT_ROOT_DIR}) |
|
|
| |
| list(APPEND CMAKE_PREFIX_PATH ${CUDNN_ROOT}) |
|
|
| set(CUDNN_INCLUDE_DIR $ENV{CUDNN_INCLUDE_DIR} CACHE PATH "Folder containing NVIDIA cuDNN header files") |
|
|
| find_path(CUDNN_INCLUDE_PATH cudnn.h |
| HINTS ${CUDNN_INCLUDE_DIR} |
| PATH_SUFFIXES cuda/include cuda include) |
|
|
| option(CUDNN_STATIC "Look for static CUDNN" OFF) |
| if (CUDNN_STATIC) |
| set(CUDNN_LIBNAME "libcudnn_static.a") |
| else() |
| set(CUDNN_LIBNAME "cudnn") |
| endif() |
|
|
| set(CUDNN_LIBRARY $ENV{CUDNN_LIBRARY} CACHE PATH "Path to the cudnn library file (e.g., libcudnn.so)") |
| if (CUDNN_LIBRARY MATCHES ".*cudnn_static.a" AND NOT CUDNN_STATIC) |
| message(WARNING "CUDNN_LIBRARY points to a static library (${CUDNN_LIBRARY}) but CUDNN_STATIC is OFF.") |
| endif() |
|
|
| find_library(CUDNN_LIBRARY_PATH ${CUDNN_LIBNAME} |
| PATHS ${CUDNN_LIBRARY} |
| PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64) |
|
|
| find_package_handle_standard_args(CUDNN DEFAULT_MSG CUDNN_LIBRARY_PATH CUDNN_INCLUDE_PATH) |
|
|
| mark_as_advanced(CUDNN_ROOT CUDNN_INCLUDE_DIR CUDNN_LIBRARY) |
|
|
