blob: 44bc46c1e8047f9f4b313ef80de6510335a9ee8b (
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
# 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. This is an environment
# variable
# ONNXRuntime_DIR CMake variable to set the installation root.
#
# Output variable
# ---------------
#
# ONNXRuntime_FOUND True if headers and requested libraries were found
# ONNXRuntime_LIBRARIES Component libraries to be linked.
# ONNXRuntime_INCLUDE_DIRS Include directories.
set(ONNXRuntime_DIR CACHE PATH "Root directory of the ONNX Runtime installation")
# This script is mostly inspired by the guide there:
# https://gitlab.kitware.com/cmake/community/-/wikis/doc/tutorials/How-To-Find-Libraries
# Finding of the DLLs was added so we can install them with the application.
# We adhere to modern CMake standards and also define an import library.
find_library(ONNXRuntime_LIBRARY onnxruntime
HINTS
${ONNXRuntime_DIR}
PATHS
${CMAKE_INSTALL_PREFIX}
PATH_SUFFIXES lib lib64 bin
CMAKE_FIND_ROOT_PATH_BOTH)
if(WIN32)
SET(CMAKE_FIND_LIBRARY_SUFFIXES ".dll" ".DLL")
find_library(ONNXRuntime_RUNTIME onnxruntime
HINTS
${ONNXRuntime_DIR}
PATHS
${CMAKE_INSTALL_PREFIX}
PATH_SUFFIXES lib lib64 bin
CMAKE_FIND_ROOT_PATH_BOTH)
else()
SET(ONNXRuntime_RUNTIME ${ONNXRuntime_LIBRARY})
endif()
find_path(ONNXRuntime_INCLUDE_DIR onnxruntime_cxx_api.h
HINTS
${ONNXRuntime_DIR}/include
${ONNXRuntime_DIR}
PATHS
${CMAKE_INSTALL_PREFIX}/include
PATH_SUFFIXES onnxruntime/core/session
CMAKE_FIND_ROOT_PATH_BOTH)
include(FindPackageHandleStandardArgs)
# handle the QUIETLY and REQUIRED arguments and set ONNXRuntime_FOUND to TRUE
# if all listed variables are TRUE
find_package_handle_standard_args(ONNXRuntime DEFAULT_MSG
ONNXRuntime_LIBRARY ONNXRuntime_INCLUDE_DIR ONNXRuntime_RUNTIME)
if(ONNXRuntime_FOUND)
if(NOT TARGET onnxruntime::onnxruntime)
# Following this quide:
# https://cmake.org/cmake/help/git-stage/guide/importing-exporting/index.html#importing-libraries
add_library(onnxruntime::onnxruntime SHARED IMPORTED)
set_target_properties(onnxruntime::onnxruntime PROPERTIES
IMPORTED_LOCATION "${ONNXRuntime_RUNTIME}"
INTERFACE_INCLUDE_DIRECTORIES "${ONNXRuntime_INCLUDE_DIR}"
IMPORTED_IMPLIB "${ONNXRuntime_LIBRARY}")
endif()
endif()
mark_as_advanced(ONNXRuntime_INCLUDE_DIR ONNXRuntime_LIBRARY ONNXRuntime_RUNTIME)
set(ONNXRuntime_INCLUDE_DIRS ${ONNXRuntime_INCLUDE_DIR})
set(ONNXRuntime_LIBRARIES ${ONNXRuntime_LIBRARY})
|
