From a93d86ec83a44723abef19113bf74ef48f20aac5 Mon Sep 17 00:00:00 2001
From: TLescoatTFX <144024179+TLescoatTFX@users.noreply.github.com>
Date: Thu, 5 Mar 2026 02:51:37 +0100
Subject: [PATCH] Fix finding cuDNN (#2890)

* Fix finding cuDNN

* Search $CUDNN_PATH first, not last
---
 cuDNN.cmake | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cuDNN.cmake b/cuDNN.cmake
index 49836b38..5570c28f 100644
--- a/cuDNN.cmake
+++ b/cuDNN.cmake
@@ -39,6 +39,7 @@ message(STATUS "Configuring cuDNN ...")
 find_path(
     _CUDNN_INCLUDE_DIR cudnn.h
     PATHS
+    $ENV{CUDNN_PATH}
     ${CUDA_TOOLKIT_ROOT_DIR}/include
     $ENV{CUDNN_PATH}/include
     $ENV{CUDA_PATH}/include
@@ -48,6 +49,7 @@ find_path(
 find_library(
     _CUDNN_LIBRARY cudnn
     HINTS
+    $ENV{CUDNN_PATH}
     ${CUDA_TOOLKIT_ROOT_DIR}/lib64
     ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
     ${CUDA_TOOLKIT_ROOT_DIR}/lib