mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-03 23:49:44 +00:00
97 lines
4.2 KiB
CMake
97 lines
4.2 KiB
CMake
# 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 ${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)
|
|
|
|
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})
|
|
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()
|
|
|