From 90e569cd4e0d74ef6ddb0e0685f8812293960671 Mon Sep 17 00:00:00 2001
From: Jacob Kahn <jacobkahn1@gmail.com>
Date: Tue, 18 Jul 2023 18:32:17 -0500
Subject: [PATCH] Add cuDNN include path as a common include dir

Some cuDNN setups don't have cuDNN installed in the same place as CUDA -- in this case, headers aren't found --> build failures.
---
 CMakeLists.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 870e67f0a..00c8f2d97 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -209,6 +209,7 @@ set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
 set(COMMON_HEADER_DIRS
   ${PROJECT_SOURCE_DIR}
   ${CUDA_PATH}/include
+  ${CUDNN_INCLUDE_PATH}
   ${CUTLASS_HEADER_DIR}
   ${CUTLASS_EXTENSIONS_DIR}
   ${PROJECT_SOURCE_DIR}/3rdparty/trt_fp8_fmha/src