diff options
Diffstat (limited to 'cmake')
| -rw-r--r-- | cmake/FindONNXRuntime.cmake | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/cmake/FindONNXRuntime.cmake b/cmake/FindONNXRuntime.cmake index f485ecdd..7350ec0d 100644 --- a/cmake/FindONNXRuntime.cmake +++ b/cmake/FindONNXRuntime.cmake @@ -19,7 +19,9 @@ find_library(ORT_LIB onnxruntime CMAKE_FIND_ROOT_PATH_BOTH) -find_path(ORT_INCLUDE onnxruntime/core/session/onnxruntime_cxx_api.h + +find_path(ORT_INCLUDE onnxruntime_cxx_api.h + PATH_SUFFIXES onnxruntime/core/session CMAKE_FIND_ROOT_PATH_BOTH) if(ORT_LIB AND ORT_INCLUDE) |