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
|
find_package(onnxruntime QUIET)
if(${onnxruntime_FOUND} AND NOT onnxruntime_INCLUDE_DIR)
find_path(onnxruntime_INCLUDE_DIR NAMES onnxruntime_cxx_api.h
HINTS
ENV onnxruntime_ROOT
ENV onnxruntime_ROOT_DIR
${CMAKE_INSTALL_PREFIX}/include
${KDE4_INCLUDE_DIR}
PATH_SUFFIXES onnxruntime
)
include(FindPackageHandleStandardArgs)
find_package_handle_standard_args(onnxruntime DEFAULT_MSG onnxruntime_INCLUDE_DIR TRUE)
set(onnxruntime_LIBRARIES onnxruntime)
get_filename_component(onnxruntime_INSTALL_PREFIX "${onnxruntime_INCLUDE_DIR}/../../" ABSOLUTE)
find_library(onnxruntime_LIBRARY onnxruntime PATHS "${onnxruntime_INSTALL_PREFIX}/lib")
add_library(${onnxruntime_LIBRARIES} SHARED IMPORTED)
set_target_properties(${onnxruntime_LIBRARIES} PROPERTIES IMPORTED_LOCATION "${onnxruntime_LIBRARY}")
set_target_properties(${onnxruntime_LIBRARIES} PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${onnxruntime_INCLUDE_DIR}")
endif()
if(NOT ${onnxruntime_FOUND})
message(STATUS "ONNXrt not found. Attempt to install...")
EXECUTE_PROCESS( COMMAND uname -m COMMAND tr -d '\n' OUTPUT_VARIABLE ARCH )
message( STATUS "Detected architecture: ${ARCH}" )
if ("${ARCH}" STREQUAL "x86_64")
set(ARCH "x64")
endif()
set(ORT_VERSION "1.16.3")
set(ORT_URL_BASE "https://github.com/microsoft/onnxruntime/releases/download")
set(ORT_TARNAME "onnxruntime-linux-${ARCH}-${ORT_VERSION}")
set(ORT_DEST "${CMAKE_CURRENT_BINARY_DIR}/external/${ORT_TARNAME}.tgz")
set(ORT_URL "${ORT_URL_BASE}/v${ORT_VERSION}/${ORT_TARNAME}.tgz")
# https://cmake.org/cmake/help/latest/policy/CMP0135.html
#
# CMP0135 is for solving re-building and re-downloading.
# The NEW policy suppresses warnings for some CMake versions.
if(POLICY CMP0135)
cmake_policy(SET CMP0135 NEW)
endif()
set(onnxruntime_DIR "${CMAKE_INSTALL_PREFIX}/external")
file(DOWNLOAD ${ORT_URL} ${ORT_DEST}
EXPECTED_HASH SHA256=b072f989d6315ac0e22dcb4771b083c5156d974a3496ac3504c77f4062eb248e
)
execute_process(COMMAND ${CMAKE_COMMAND} -E tar xzf ${ORT_DEST})
execute_process(COMMAND ${CMAKE_COMMAND} -E copy_directory ${ORT_TARNAME} ${onnxruntime_DIR})
set(onnxruntime_INCLUDE_DIR "${onnxruntime_DIR}/include")
set(onnxruntime_LIBRARY_DIR "${onnxruntime_DIR}/lib")
set(onnxruntime_LIBRARIES onnxruntime)
add_library(${onnxruntime_LIBRARIES} SHARED IMPORTED GLOBAL)
set_target_properties(${onnxruntime_LIBRARIES} PROPERTIES IMPORTED_LOCATION ${onnxruntime_LIBRARY_DIR}/libonnxruntime.so)
set(onnxruntime_FOUND TRUE)
endif()
|
