diff --git a/cmake/FindONNXRuntime.cmake b/cmake/FindONNXRuntime.cmake index d9c17e44e..4b2890344 100644 --- a/cmake/FindONNXRuntime.cmake +++ b/cmake/FindONNXRuntime.cmake @@ -50,9 +50,10 @@ find_library( PATH_SUFFIXES lib HINTS ${PC_ONNXRuntime_LIBRARY_DIRS}) find_path( - ONNXRuntime_INCLUDE_DIR core/session/onnxruntime_cxx_api.h + ONNXRuntime_INCLUDE_DIR onnxruntime_cxx_api.h PATHS ${ONNXRuntime_ROOT_DIR} - PATH_SUFFIXES onnxruntime include/onnxruntime + PATH_SUFFIXES onnxruntime include include/onnxruntime onnxruntime/core/session + include/onnxruntime/core/session HINTS ${PC_ONNXRuntime_INCLUDE_DIRS}) include(FindPackageHandleStandardArgs)