|
|
|
@ -72,14 +72,14 @@ if(CUDNN_INCLUDE_DIR) |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
string(REGEX MATCH "define CUDNN_MAJOR ([0-9]+)" _ "${CUDNN_H_CONTENTS}") |
|
|
|
|
set(CUDNN_MAJOR_VERSION ${CMAKE_MATCH_1} CACHE INTERNAL "") |
|
|
|
|
set(CUDNN_VERSION_MAJOR ${CMAKE_MATCH_1} CACHE INTERNAL "") |
|
|
|
|
string(REGEX MATCH "define CUDNN_MINOR ([0-9]+)" _ "${CUDNN_H_CONTENTS}") |
|
|
|
|
set(CUDNN_MINOR_VERSION ${CMAKE_MATCH_1} CACHE INTERNAL "") |
|
|
|
|
set(CUDNN_VERSION_MINOR ${CMAKE_MATCH_1} CACHE INTERNAL "") |
|
|
|
|
string(REGEX MATCH "define CUDNN_PATCHLEVEL ([0-9]+)" _ "${CUDNN_H_CONTENTS}") |
|
|
|
|
set(CUDNN_PATCH_VERSION ${CMAKE_MATCH_1} CACHE INTERNAL "") |
|
|
|
|
set(CUDNN_VERSION_PATCH ${CMAKE_MATCH_1} CACHE INTERNAL "") |
|
|
|
|
|
|
|
|
|
set(CUDNN_VERSION |
|
|
|
|
"${CUDNN_MAJOR_VERSION}.${CUDNN_MINOR_VERSION}.${CUDNN_PATCH_VERSION}" |
|
|
|
|
"${CUDNN_VERSION_MAJOR}.${CUDNN_VERSION_MINOR}.${CUDNN_VERSION_PATCH}" |
|
|
|
|
CACHE |
|
|
|
|
STRING |
|
|
|
|
"cuDNN version" |
|
|
|
|