Add ONNXRuntime cmake module

1 files changed, 31 insertions, 0 deletions

diff --git a/cmake/FindONNXRuntime.cmake b/cmake/FindONNXRuntime.cmake
new file mode 100644
index 00000000..733bdb36
--- /dev/null
+++ b/cmake/FindONNXRuntime.cmake
@@ -0,0 +1,31 @@
+# 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           Variable indicating that ONNXRuntime has been
+#                               found.
+#   ONNXRuntime_LIBRARIES       Library implementing ONNXRuntime
+#   ONNXRuntime_INCLUDE_DIRS    Headers for ONNXRuntime
+
+find_library(ORT_LIB onnxruntime
+    CMAKE_FIND_ROOT_PATH_BOTH)
+find_path(ORT_INCLUDE onnxruntime/core/session/onnxruntime_cxx_api.h
+    CMAKE_FIND_ROOT_PATH_BOTH)
+
+if(ORT_LIB AND ORT_INCLUDE)
+    set(ONNXRuntime_FOUND TRUE)
+    # For CMake output only
+    set(ONNXRuntime_LIBRARIES "${ORT_LIB}" CACHE STRING "ONNX Runtime libraries")
+    set(ONNXRuntime_INCLUDE_DIRS "${ORT_INCLUDE}" CACHE STRING "ONNX Runtime include path")
+endif()