blob: 733bdb3623023745019478b67429702e305921e9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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()
|
