From 026fb3ca1ee755fff724a9cc840a81cc4bdb8a04 Mon Sep 17 00:00:00 2001 From: Stephan Gocht Date: Thu, 6 Apr 2023 17:56:20 +0200 Subject: [PATCH] Fix compilation error when CUDNN_HOME is defined. (#15348) --- cmake/onnxruntime_providers.cmake | 1 + 1 file changed, 1 insertion(+) diff --git a/cmake/onnxruntime_providers.cmake b/cmake/onnxruntime_providers.cmake index 39e00e2592..5497dc62f2 100644 --- a/cmake/onnxruntime_providers.cmake +++ b/cmake/onnxruntime_providers.cmake @@ -491,6 +491,7 @@ if (onnxruntime_USE_CUDA) target_link_libraries(onnxruntime_providers_cuda PRIVATE cublasLt cublas cudnn curand cufft ${ABSEIL_LIBS} ${ONNXRUNTIME_PROVIDERS_SHARED} Boost::mp11 safeint_interface) if(onnxruntime_CUDNN_HOME) target_include_directories(onnxruntime_providers_cuda PRIVATE ${onnxruntime_CUDNN_HOME}/include) + target_link_directories(onnxruntime_providers_cuda PRIVATE ${onnxruntime_CUDNN_HOME}/lib) endif() if (onnxruntime_USE_FLASH_ATTENTION)