From 574fb92ac67356df07a80880d89417f05f026cdf Mon Sep 17 00:00:00 2001 From: Valery Chernov Date: Wed, 2 Feb 2022 08:45:08 +0300 Subject: [PATCH] roll back CudaNN home check --- cmake/CMakeLists.txt | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cmake/CMakeLists.txt b/cmake/CMakeLists.txt index fee00dee0b551..8ff5b5d706435 100644 --- a/cmake/CMakeLists.txt +++ b/cmake/CMakeLists.txt @@ -807,6 +807,10 @@ if (onnxruntime_DISABLE_OPTIONAL_TYPE) add_compile_definitions(DISABLE_OPTIONAL_TYPE) endif() +if (onnxruntime_USE_CUDA AND "${onnxruntime_CUDNN_HOME}" STREQUAL "") + message(FATAL_ERROR "onnxruntime_CUDNN_HOME required for onnxruntime_USE_CUDA") +endif() + get_filename_component(ONNXRUNTIME_ROOT "${ONNXRUNTIME_ROOT}" ABSOLUTE) get_filename_component(ORTTRAINING_ROOT "${ORTTRAINING_ROOT}" ABSOLUTE) get_filename_component(REPO_ROOT "${REPO_ROOT}" ABSOLUTE)