96 lines
3.5 KiB
CMake
96 lines
3.5 KiB
CMake
|
vcpkg_check_linkage(ONLY_STATIC_LIBRARY)
|
||
|
|
|
||
|
|
vcpkg_from_github(
|
||
|
|
OUT_SOURCE_PATH SOURCE_PATH
|
||
|
|
REPO onnx/onnx
|
||
|
|
REF v1.10.2
|
||
|
|
SHA512 7519d326cd2b2b13a269ec0d01af07c32115d183dae6e1eaae55f5b23b6c92b2aadbb2b1e555557f4201bbcf921fa563d09d45d7f1d3bd2399c1a94a6ef63303
|
||
|
|
PATCHES
|
||
|
|
fix-cmakelists.patch
|
||
|
|
fix-dependency-protobuf.patch
|
||
|
|
)
|
||
|
|
|
||
|
|
string(COMPARE EQUAL "${VCPKG_CRT_LINKAGE}" "static" USE_STATIC_RUNTIME)
|
||
|
|
|
||
|
|
# ONNX_USE_PROTOBUF_SHARED_LIBS: find the library and check its file extension
|
||
|
|
find_library(PROTOBUF_LIBPATH NAMES protobuf PATHS "${CURRENT_INSTALLED_DIR}/bin" "${CURRENT_INSTALLED_DIR}/lib" REQUIRED)
|
||
|
|
get_filename_component(PROTOBUF_LIBNAME "${PROTOBUF_LIBPATH}" NAME)
|
||
|
|
if(PROTOBUF_LIBNAME MATCHES "${CMAKE_SHARED_LIBRARY_SUFFIX}")
|
||
|
|
set(USE_PROTOBUF_SHARED ON)
|
||
|
|
else()
|
||
|
|
set(USE_PROTOBUF_SHARED OFF)
|
||
|
|
endif()
|
||
|
|
|
||
|
|
vcpkg_check_features(OUT_FEATURE_OPTIONS FEATURE_OPTIONS
|
||
|
|
FEATURES
|
||
|
|
pybind11 BUILD_ONNX_PYTHON
|
||
|
|
)
|
||
|
|
|
||
|
|
# Like protoc, python is required for codegen.
|
||
|
|
vcpkg_find_acquire_program(PYTHON3)
|
||
|
|
|
||
|
|
# PATH for .bat scripts so it can find 'python'
|
||
|
|
get_filename_component(PYTHON_DIR "${PYTHON3}" PATH)
|
||
|
|
vcpkg_add_to_path(PREPEND "${PYTHON_DIR}")
|
||
|
|
|
||
|
|
if("pybind11" IN_LIST FEATURES)
|
||
|
|
# When BUILD_ONNX_PYTHON, we need Development component. Give a hint for FindPython3
|
||
|
|
list(APPEND FEATURE_OPTIONS
|
||
|
|
"-DPython3_ROOT_DIR=${CURRENT_INSTALLED_DIR}"
|
||
|
|
)
|
||
|
|
endif()
|
||
|
|
|
||
|
|
vcpkg_cmake_configure(
|
||
|
|
SOURCE_PATH "${SOURCE_PATH}"
|
||
|
|
OPTIONS
|
||
|
|
${FEATURE_OPTIONS}
|
||
|
|
-DPython3_EXECUTABLE=${PYTHON3}
|
||
|
|
-DONNX_ML=ON
|
||
|
|
-DONNX_GEN_PB_TYPE_STUBS=ON
|
||
|
|
-DONNX_USE_PROTOBUF_SHARED_LIBS=${USE_PROTOBUF_SHARED}
|
||
|
|
-DONNX_USE_LITE_PROTO=OFF
|
||
|
|
-DONNX_USE_MSVC_STATIC_RUNTIME=${USE_STATIC_RUNTIME}
|
||
|
|
-DONNX_BUILD_TESTS=OFF
|
||
|
|
-DONNX_BUILD_BENCHMARKS=OFF
|
||
|
|
MAYBE_UNUSED_VARIABLES
|
||
|
|
ONNX_USE_MSVC_STATIC_RUNTIME
|
||
|
|
)
|
||
|
|
|
||
|
|
if("pybind11" IN_LIST FEATURES)
|
||
|
|
# This target is not in install/export
|
||
|
|
vcpkg_cmake_build(TARGET onnx_cpp2py_export)
|
||
|
|
endif()
|
||
|
|
vcpkg_cmake_install()
|
||
|
|
vcpkg_cmake_config_fixup(CONFIG_PATH lib/cmake/ONNX)
|
||
|
|
|
||
|
|
file(INSTALL "${SOURCE_PATH}/LICENSE" DESTINATION "${CURRENT_PACKAGES_DIR}/share/${PORT}" RENAME copyright)
|
||
|
|
|
||
|
|
file(REMOVE_RECURSE
|
||
|
|
"${CURRENT_PACKAGES_DIR}/debug/include"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/debug/share"
|
||
|
|
# the others are empty
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/backend"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/bin"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/controlflow"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/generator"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/logical"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/math"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/nn"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/object_detection"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/optional"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/quantization"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/reduction"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/rnn"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/sequence"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/traditionalml"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/defs/training"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/examples"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/frontend"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/onnx_cpp2py_export"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/test"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/tools"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/onnx_ml"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/onnx_data"
|
||
|
|
"${CURRENT_PACKAGES_DIR}/include/onnx/onnx_operators_ml"
|
||
|
|
)
|