From 0677f3e21cc3379e68d517dc80a8c12e5df0c608 Mon Sep 17 00:00:00 2001 From: Cuntian Liu Date: Thu, 11 Feb 2021 03:38:34 +0800 Subject: [PATCH] Merge pull request #19452 from ctuu:patch-1 * fix: findCUDNN script * Update FindCUDNN.cmake --- cmake/FindCUDNN.cmake | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cmake/FindCUDNN.cmake b/cmake/FindCUDNN.cmake index 90d8b2e..195781b 100644 --- a/cmake/FindCUDNN.cmake +++ b/cmake/FindCUDNN.cmake @@ -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" -- 2.7.4