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
|
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
#
# Setup onnx and onnx_protobuf for a build with onnxruntime_MINIMAL_BUILD enabled.
# We exclude everything but the essentials from the onnx library.
#
if(NOT onnxruntime_MINIMAL_BUILD)
message(FATAL_ERROR "This file should only be included in a minimal build")
endif()
#TODO: if protobuf is a shared lib and onnxruntime_USE_FULL_PROTOBUF is ON, then onnx_proto should be built as a shared lib instead of a static lib. Otherwise any code outside onnxruntime.dll can't use onnx protobuf definitions if they share the protobuf.dll with onnxruntime. For example, if protobuf is a shared lib and onnx_proto is a static lib then onnxruntime_perf_test won't work.
FetchContent_Populate(onnx)
set(ONNX_SOURCE_ROOT ${onnx_SOURCE_DIR})
add_library(onnx_proto ${ONNX_SOURCE_ROOT}/onnx/onnx-ml.proto ${ONNX_SOURCE_ROOT}/onnx/onnx-operators-ml.proto ${ONNX_SOURCE_ROOT}/onnx/onnx-data.proto)
target_include_directories(onnx_proto PUBLIC $<TARGET_PROPERTY:protobuf::libprotobuf,INTERFACE_INCLUDE_DIRECTORIES> "${CMAKE_CURRENT_BINARY_DIR}")
target_compile_definitions(onnx_proto PUBLIC $<TARGET_PROPERTY:protobuf::libprotobuf,INTERFACE_COMPILE_DEFINITIONS>)
set(_src_prefix "onnx/")
onnxruntime_protobuf_generate(NO_SRC_INCLUDES GEN_SRC_PREFIX ${_src_prefix} IMPORT_DIRS ${ONNX_SOURCE_ROOT} TARGET onnx_proto)
# For reference, this would be the full ONNX source include. We only need data_type_utils in this build.
# file(GLOB_RECURSE onnx_src CONFIGURE_DEPENDS
# "${ONNX_SOURCE_ROOT}/onnx/*.h"
# "${ONNX_SOURCE_ROOT}/onnx/*.cc"
# )
# file(GLOB_RECURSE onnx_exclude_src CONFIGURE_DEPENDS
# "${ONNX_SOURCE_ROOT}/onnx/py_utils.h"
# "${ONNX_SOURCE_ROOT}/onnx/proto_utils.h"
# "${ONNX_SOURCE_ROOT}/onnx/backend/test/cpp/*"
# "${ONNX_SOURCE_ROOT}/onnx/test/*"
# "${ONNX_SOURCE_ROOT}/onnx/cpp2py_export.cc"
# )
# list(REMOVE_ITEM onnx_src ${onnx_exclude_src})
set(onnx_src
"${ONNX_SOURCE_ROOT}/onnx/common/common.h"
"${ONNX_SOURCE_ROOT}/onnx/defs/data_type_utils.h"
"${ONNX_SOURCE_ROOT}/onnx/defs/data_type_utils.cc"
)
add_library(onnx ${onnx_src})
add_dependencies(onnx onnx_proto)
target_include_directories(onnx PUBLIC "${ONNX_SOURCE_ROOT}")
target_include_directories(onnx PUBLIC $<TARGET_PROPERTY:onnx_proto,INTERFACE_INCLUDE_DIRECTORIES>)
if (onnxruntime_USE_FULL_PROTOBUF)
target_compile_definitions(onnx PUBLIC "ONNX_ML" "ONNX_NAMESPACE=onnx")
else()
target_compile_definitions(onnx PUBLIC "ONNX_ML" "ONNX_NAMESPACE=onnx" "ONNX_USE_LITE_PROTO")
endif()
if (WIN32)
target_compile_options(onnx PRIVATE
/wd4800 # 'type' : forcing value to bool 'true' or 'false' (performance warning)
/wd4125 # decimal digit terminates octal escape sequence
/wd4100 # 'param' : unreferenced formal parameter
/wd4244 # 'argument' conversion from 'google::protobuf::int64' to 'int', possible loss of data
/wd4996 # 'argument' Using double parameter version instead of single parameter version of SetTotalBytesLimit(). The second parameter is ignored.
)
if (NOT onnxruntime_DISABLE_EXCEPTIONS)
target_compile_options(onnx PRIVATE
/EHsc # exception handling - C++ may throw, extern "C" will not
)
endif()
target_compile_options(onnx_proto PRIVATE
/wd4244 # 'argument' conversion from 'google::protobuf::int64' to 'int', possible loss of data
)
set(onnx_static_library_flags
-IGNORE:4221 # LNK4221: This object file does not define any previously undefined public symbols, so it will not be used by any link operation that consumes this library
)
set_target_properties(onnx PROPERTIES
STATIC_LIBRARY_FLAGS "${onnx_static_library_flags}")
else()
if(HAS_UNUSED_PARAMETER)
target_compile_options(onnx PRIVATE "-Wno-unused-parameter")
endif()
if(HAS_UNUSED_BUT_SET_VARIABLE)
target_compile_options(onnx PRIVATE "-Wno-unused-but-set-variable")
endif()
endif()
|
