-
Notifications
You must be signed in to change notification settings - Fork 3.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
replaces cmake CUDA finder with FindCUDAToolkit
- Loading branch information
Showing
3 changed files
with
95 additions
and
172 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -4,6 +4,7 @@ on: | |
push: | ||
branches: | ||
- master | ||
- cuda-cmake | ||
paths: | ||
- ".github/workflows/build_cpp.yml" | ||
- "**.cpp" | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,76 @@ | ||
# Find the CUDNN libraries | ||
# | ||
# The following variables are optionally searched for defaults | ||
# CUDNN_ROOT: Base directory where CUDNN is found | ||
# CUDNN_INCLUDE_DIR: Directory where CUDNN header is searched for | ||
# CUDNN_LIBRARY: Directory where CUDNN library is searched for | ||
# CUDNN_STATIC: Are we looking for a static library? (default: no) | ||
# | ||
# The following are set after configuration is done: | ||
# CUDNN_FOUND | ||
# CUDNN_INCLUDE_PATH | ||
# CUDNN_LIBRARY_PATH | ||
# | ||
|
||
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}) | ||
|
||
# Compatible layer for CMake <3.12. CUDNN_ROOT will be accounted in for searching paths and libraries for CMake >=3.12. | ||
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) | ||
|
||
if(CUDNN_FOUND) | ||
# Get cuDNN version | ||
if(EXISTS ${CUDNN_INCLUDE_PATH}/cudnn_version.h) | ||
file(READ ${CUDNN_INCLUDE_PATH}/cudnn_version.h CUDNN_HEADER_CONTENTS) | ||
else() | ||
file(READ ${CUDNN_INCLUDE_PATH}/cudnn.h CUDNN_HEADER_CONTENTS) | ||
endif() | ||
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}") | ||
# Assemble cuDNN version | ||
if(NOT CUDNN_VERSION_MAJOR) | ||
set(CUDNN_VERSION "?") | ||
else() | ||
set(CUDNN_VERSION | ||
"${CUDNN_VERSION_MAJOR}.${CUDNN_VERSION_MINOR}.${CUDNN_VERSION_PATCH}") | ||
endif() | ||
endif() |