forked from microsoft/onnxruntime
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Last major set of ORT format model changes (microsoft#5056)
* Add minimal build option to build.py Group some of the build settings so binary size reduction options are all together Make some cmake variable naming more consistent Replace usage of std::hash with murmurhash3 for kernel. std::hash is implementation dependent so can't be used. Add initial doco and ONNX to ORT model conversion script Misc cleanups of minimal build breaks.
- Loading branch information
1 parent
6134994
commit b5c2932
Showing
28 changed files
with
983 additions
and
223 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,100 @@ | ||
| # 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. | ||
|
|
||
| set(ONNX_SOURCE_ROOT ${PROJECT_SOURCE_DIR}/external/onnx) | ||
|
|
||
| add_library(onnx_proto ${ONNX_SOURCE_ROOT}/onnx/onnx-ml.proto ${ONNX_SOURCE_ROOT}/onnx/onnx-operators-ml.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) | ||
|
|
||
| if (WIN32) | ||
| target_compile_options(onnx_proto PRIVATE "/wd4146" "/wd4125" "/wd4456" "/wd4267" "/wd4309") | ||
| else() | ||
| if(HAS_UNUSED_VARIABLE) | ||
| target_compile_options(onnx_proto PRIVATE "-Wno-unused-variable") | ||
| endif() | ||
|
|
||
| if(HAS_UNUSED_BUT_SET_VARIABLE) | ||
| target_compile_options(onnx_proto PRIVATE "-Wno-unused-but-set-variable") | ||
| endif() | ||
| endif() | ||
|
|
||
| # 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}) | ||
| file(GLOB onnx_src CONFIGURE_DEPENDS | ||
| "${ONNX_SOURCE_ROOT}/onnx/defs/data_type_utils.*" | ||
| ) | ||
|
|
||
| if (MSVC) | ||
| SET (CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /Gw /GL") | ||
| SET (CMAKE_CXX_FLAGS_RELWITHDEBINFO "${CMAKE_CXX_FLAGS_RELWITHDEBINFO} /Gw /GL") | ||
| endif() | ||
|
|
||
| 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() | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.