From 728cc03ad91e58127b2e54ea2d66944e19c10dc2 Mon Sep 17 00:00:00 2001
From: WolframRhodium <WolframRhodium@users.noreply.github.com>
Date: Mon, 23 Oct 2023 07:11:00 +0800
Subject: [PATCH] vstrt/{trtexec}/CMakeLists.txt: linking cuDNN

---
 vstrt/CMakeLists.txt         | 2 +-
 vstrt/trtexec/CMakeLists.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/vstrt/CMakeLists.txt b/vstrt/CMakeLists.txt
index 1185c64..a4de3d0 100644
--- a/vstrt/CMakeLists.txt
+++ b/vstrt/CMakeLists.txt
@@ -39,7 +39,7 @@ endif()
 
 if (USE_NVINFER_PLUGIN)
     add_definitions(-DUSE_NVINFER_PLUGIN)
-    find_library(CUDNN_LIB cudnn HINTS ${CUDNN_ROOT_DIR} PATH_SUFFIXES lib)
+    find_library(CUDNN_LIB cudnn HINTS ${CUDNN_HOME} PATH_SUFFIXES lib lib/x64)
     if (USE_NVINFER_PLUGIN_STATIC)
         target_link_libraries(vstrt PRIVATE nvinfer_plugin_static ${CUDNN_LIB} CUDA::cublas)
 
diff --git a/vstrt/trtexec/CMakeLists.txt b/vstrt/trtexec/CMakeLists.txt
index d04208e..1282795 100644
--- a/vstrt/trtexec/CMakeLists.txt
+++ b/vstrt/trtexec/CMakeLists.txt
@@ -29,7 +29,7 @@ target_link_directories(trtexec PRIVATE
     ${TENSORRT_HOME}/lib
 )
 
-find_library(CUDNN_LIB cudnn HINTS ${CUDNN_ROOT_DIR} PATH_SUFFIXES lib)
+find_library(CUDNN_LIB cudnn HINTS ${CUDNN_HOME} PATH_SUFFIXES lib lib/x64)
 
 target_link_libraries(trtexec PRIVATE
     CUDA::cudart_static