aboutsummaryrefslogtreecommitdiff
path: root/ports/onnx/portfile.cmake
blob: 1c565ab0d5f30271200f64bbd38934e84cb34fd1 (plain)
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
90
91
92
93
94
95
96
97
98
# uwp: LOAD_LIBRARY_SEARCH_DEFAULT_DIRS undefined identifier
vcpkg_fail_port_install(ON_TARGET "uwp")
vcpkg_check_linkage(ONLY_STATIC_LIBRARY)

vcpkg_from_github(
    OUT_SOURCE_PATH SOURCE_PATH
    REPO onnx/onnx
    REF v1.9.0
    SHA512 a3eecc74ce4f22524603fb86367d21c87a143ba27eef93ef4bd2e2868c2cadeb724b84df58a429286e7824adebdeba7fa059095b7ab29df8dcea8777bd7f4101
    PATCHES
        fix-cmakelists.patch
        wrap-onnxifi-targets.patch
)

if(VCPKG_TARGET_IS_WINDOWS)
    string(COMPARE EQUAL "${VCPKG_CRT_LINKAGE}" "static" USE_STATIC_RUNTIME)
    list(APPEND PLATFORM_OPTIONS
        -DONNX_USE_MSVC_STATIC_RUNTIME=${USE_STATIC_RUNTIME}
    )
endif()

# 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 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} ${PLATFORM_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
        -DONNXIFI_ENABLE_EXT=OFF
        -DONNX_BUILD_TESTS=OFF
        -DONNX_BUILD_BENCHMARKS=OFF
)

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/onnx_ml"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/onnx_data"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/onnx_operators_ml"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/onnx_cpp2py_export"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/backend"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/tools"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/test"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/bin"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/examples"
                    "${CURRENT_PACKAGES_DIR}/include/onnx/frontend"
                    "${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/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"
)