mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-29 20:14:01 +00:00
60 lines
No EOL
2.9 KiB
CMake
60 lines
No EOL
2.9 KiB
CMake
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
add_library(onnx_proto ${ONNXRUNTIME_ROOT}/core/protobuf/onnx-ml.proto ${ONNXRUNTIME_ROOT}/core/protobuf/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>)
|
|
onnxruntime_protobuf_generate(APPEND_PATH IMPORT_DIRS ${ONNXRUNTIME_ROOT}/core/protobuf TARGET onnx_proto)
|
|
if (WIN32)
|
|
target_compile_options(onnx_proto PRIVATE /wd4146) # unary minus operator applied to unsigned type
|
|
endif()
|
|
# Cpp Tests were added and they require googletest
|
|
# since we have our own copy, try using that
|
|
set(ONNX_SOURCE_ROOT ${PROJECT_SOURCE_DIR}/external/onnx)
|
|
file(GLOB_RECURSE onnx_src
|
|
"${ONNX_SOURCE_ROOT}/onnx/*.h"
|
|
"${ONNX_SOURCE_ROOT}/onnx/*.cc"
|
|
)
|
|
|
|
file(GLOB_RECURSE onnx_exclude_src
|
|
"${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})
|
|
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>)
|
|
target_compile_definitions(onnx PUBLIC $<TARGET_PROPERTY:onnx_proto,INTERFACE_COMPILE_DEFINITIONS> PRIVATE "__ONNX_DISABLE_STATIC_REGISTRATION")
|
|
target_compile_definitions(onnx PUBLIC "ONNX_ML" "ONNX_NAMESPACE=onnx")
|
|
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
|
|
/EHsc # exception handling - C++ may throw, extern "C" will not
|
|
)
|
|
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")
|
|
target_compile_options(onnx_proto PRIVATE "-Wno-unused-parameter")
|
|
endif()
|
|
if(HAS_UNUSED_BUT_SET_VARIABLE)
|
|
target_compile_options(onnx PRIVATE "-Wno-unused-but-set-variable")
|
|
target_compile_options(onnx_proto PRIVATE "-Wno-unused-but-set-variable")
|
|
endif()
|
|
endif()
|
|
|
|
|
|
set_target_properties(onnx PROPERTIES FOLDER "External/ONNX")
|
|
set_target_properties(onnx_proto PROPERTIES FOLDER "External/ONNX") |