diff options
| author | Stanislaw Halik <sthalik@misaki.pl> | 2021-08-03 08:04:30 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-08-03 08:04:30 +0000 |
| commit | 6a5954197ef5671b5cc9159377e83ef15bf43373 (patch) | |
| tree | c4ea500f2b42636464b844bd831b6382f931e0ba /cmake | |
| parent | fef1936f3c320ec026a0fdf65f103494385e15e3 (diff) | |
| parent | 1e3d62e04f78e2d0e22b3ceb67aaf20a03f18bbc (diff) | |
Merge pull request #1306 from gdolle/tracker-neuralnet-cmake
Tracker neuralnet update onnxrt cmake (linux)
Diffstat (limited to 'cmake')
| -rw-r--r-- | cmake/FindONNXRuntime.cmake | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/cmake/FindONNXRuntime.cmake b/cmake/FindONNXRuntime.cmake new file mode 100644 index 00000000..7350ec0d --- /dev/null +++ b/cmake/FindONNXRuntime.cmake @@ -0,0 +1,42 @@ +# FindONNXRuntime +# =============== +# +# Find an ONNX Runtime installation. +# ONNX Runtime is a cross-platform inference and training machine-learning +# accelerator. +# +# Input variables +# --------------- +# +# ONNXRuntime_ROOT Set root installation. +# +# Output variable +# --------------- +# +# ONNXRuntime_FOUND True if headers and requested libraries were found +# ONNXRuntime_LIBRARIES Component libraries to be linked. +# ONNXRuntime_INCLUDE_DIRS Include directories. + +find_library(ORT_LIB onnxruntime + CMAKE_FIND_ROOT_PATH_BOTH) + +find_path(ORT_INCLUDE onnxruntime_cxx_api.h + PATH_SUFFIXES onnxruntime/core/session + CMAKE_FIND_ROOT_PATH_BOTH) + +if(ORT_LIB AND ORT_INCLUDE) + set(ONNXRuntime_FOUND TRUE) + set(ONNXRuntime_INCLUDE_DIRS "${ORT_INCLUDE}") + + if(NOT TARGET onnxruntime) + add_library(onnxruntime UNKNOWN IMPORTED) + set_target_properties(onnxruntime PROPERTIES + IMPORTED_LOCATION "${ORT_LIB}" + INTERFACE_INCLUDE_DIRECTORIES "${ORT_INCLUDE}" + INTERFACE_LINK_LIBRARIES "onnxruntime") + list(APPEND ONNXRuntime_LIBRARIES onnxruntime) + endif() +endif() + +unset(ORT_LIB CACHE) +unset(ORT_INCLUDE CACHE) |