blob: af5093f3f43265725975fbcee58b2392abf29bd2 (
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
# 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.
# Determine architecture string for subdir in nuget package
# This is only relevant when we point `ONNXRuntime_DIR` to the
# extracted nuget package content.
if (CMAKE_SYSTEM_NAME STREQUAL "Linux")
if (CMAKE_SIZEOF_VOID_P GREATER_EQUAL 8)
set(ONNXRuntime_Arch "linux-x64")
else()
message(FATAL_ERROR "32 Bit Linux builds are not supported")
endif()
elseif(CMAKE_SYSTEM_NAME STREQUAL "Windows")
if (CMAKE_SIZEOF_VOID_P GREATER_EQUAL 8)
set(ONNXRuntime_Arch "win-x64")
else()
set(ONNXRuntime_Arch "win-x86")
endif()
endif()
find_library(ONNXRuntime_LIBRARY onnxruntime
HINTS ${ONNXRuntime_DIR}
PATH_SUFFIXES
# For a "normal" installation
"lib" "lib64" "bin"
# For the nuget package
"runtimes/${ONNXRuntime_Arch}/native"
)
if(WIN32)
SET(CMAKE_FIND_LIBRARY_SUFFIXES ".dll" ".DLL")
find_library(ONNXRuntime_RUNTIME onnxruntime
HINTS ${ONNXRuntime_DIR}
PATH_SUFFIXES
# For a "normal" installation
"lib" "lib64" "bin"
# For the nuget package
"runtimes/${ONNXRuntime_Arch}/native"
)
else()
SET(ONNXRuntime_RUNTIME ${ONNXRuntime_LIBRARY})
endif()
find_path(ONNXRuntime_INCLUDE_DIR onnxruntime_cxx_api.h
HINTS ${ONNXRuntime_DIR}
PATH_SUFFIXES
# For .nuget packages
"build/native/include"
# For when the directory structure of the onnx source repo is preserved
"include/onnxruntime/core/session"
# For when we copy the files somewhere
"include"
)
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})
|
