From b0a65634d2527488824d1dacdd5a877d54eb7869 Mon Sep 17 00:00:00 2001
From: Guillaume Dollé <dolle.guillaume@gmail.com>
Date: Sun, 1 Aug 2021 03:49:35 +0200
Subject: Add ONNXRuntime cmake module

---
 cmake/FindONNXRuntime.cmake | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 cmake/FindONNXRuntime.cmake

(limited to 'cmake/FindONNXRuntime.cmake')

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()
-- 
cgit v1.2.3