-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[onnxruntime] support 'coreml' feature
- Loading branch information
Showing
7 changed files
with
229 additions
and
11 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,145 @@ | ||
| # Copyright (c) Microsoft Corporation. All rights reserved. | ||
| # Licensed under the MIT License. | ||
|
|
||
| include(FetchContent) | ||
|
|
||
| # Pass to build | ||
| set(ABSL_PROPAGATE_CXX_STD 1) | ||
| set(BUILD_TESTING 0) | ||
| set(ABSL_BUILD_TESTING OFF) | ||
| set(ABSL_BUILD_TEST_HELPERS OFF) | ||
| set(ABSL_USE_EXTERNAL_GOOGLETEST ON) | ||
| if(Patch_FOUND AND WIN32) | ||
| set(ABSL_PATCH_COMMAND ${Patch_EXECUTABLE} --binary --ignore-whitespace -p1 < ${PROJECT_SOURCE_DIR}/patches/abseil/absl_windows.patch) | ||
| else() | ||
| set(ABSL_PATCH_COMMAND "") | ||
| endif() | ||
| if(WIN32 AND NOT Patch_FOUND) | ||
| #see https://github.com/google/re2/issues/425 and https://github.com/google/re2/issues/436 | ||
| set(ABSL_ENABLE_INSTALL ON) | ||
| endif() | ||
| # NB! Advancing Abseil version changes its internal namespace, | ||
| # currently absl::lts_20240116 which affects abseil-cpp.natvis debugger | ||
| # visualization file, that must be adjusted accordingly, unless we eliminate | ||
| # that namespace at build time. | ||
| FetchContent_Declare( | ||
| abseil_cpp | ||
| URL ${DEP_URL_abseil_cpp} | ||
| URL_HASH SHA1=${DEP_SHA1_abseil_cpp} | ||
| PATCH_COMMAND ${ABSL_PATCH_COMMAND} | ||
| FIND_PACKAGE_ARGS NAMES absl | ||
| ) | ||
|
|
||
| onnxruntime_fetchcontent_makeavailable(abseil_cpp) | ||
| FetchContent_GetProperties(abseil_cpp) | ||
| set(ABSEIL_SOURCE_DIR ${abseil_cpp_SOURCE_DIR}) | ||
| # abseil_cpp_SOURCE_DIR is non-empty if we build it from source | ||
| message(STATUS "Abseil source dir:" ${ABSEIL_SOURCE_DIR}) | ||
| # abseil_cpp_VERSION is non-empty if we find a preinstalled ABSL | ||
| if(abseil_cpp_VERSION) | ||
| message(STATUS "Abseil version:" ${abseil_cpp_VERSION}) | ||
| endif() | ||
| if (GDK_PLATFORM) | ||
| # Abseil considers any partition that is NOT in the WINAPI_PARTITION_APP a viable platform | ||
| # for Win32 symbolize code (which depends on dbghelp.lib); this logic should really be flipped | ||
| # to only include partitions that are known to support it (e.g. DESKTOP). As a workaround we | ||
| # tell Abseil to pretend we're building an APP. | ||
| target_compile_definitions(absl_symbolize PRIVATE WINAPI_FAMILY=WINAPI_FAMILY_DESKTOP_APP) | ||
| endif() | ||
|
|
||
| # TODO: since multiple ORT's dependencies depend on Abseil, the list below would vary from version to version. | ||
| # We'd better to not manually manage the list. | ||
| set(ABSEIL_LIBS | ||
| absl::absl_log | ||
| absl::log_internal_log_impl | ||
| absl::log_internal_strip | ||
| absl::log_internal_message | ||
| absl::log_internal_format | ||
| absl::synchronization | ||
| absl::str_format | ||
| absl::flags | ||
| absl::log_internal_globals | ||
| absl::kernel_timeout_internal | ||
| absl::str_format_internal | ||
| absl::hash | ||
| absl::log_internal_append_truncated | ||
| absl::absl_vlog_is_on | ||
| absl::flags_commandlineflag | ||
| absl::time | ||
| absl::symbolize | ||
| absl::graphcycles_internal | ||
| absl::log_internal_conditions | ||
| absl::strings | ||
| absl::malloc_internal | ||
| absl::demangle_internal | ||
| absl::optional | ||
| absl::stacktrace | ||
| absl::base | ||
| absl::demangle_rust | ||
| absl::bad_optional_access | ||
| absl::strings_internal | ||
| absl::debugging_internal | ||
| absl::int128 | ||
| absl::spinlock_wait | ||
| absl::decode_rust_punycode | ||
| absl::raw_logging_internal | ||
| absl::flat_hash_set | ||
| absl::flat_hash_map | ||
| absl::node_hash_map | ||
| absl::node_hash_set | ||
| absl::compare | ||
| absl::base_internal | ||
| absl::nullability | ||
| absl::bounded_utf8_length_sequence | ||
| absl::log_severity | ||
| absl::type_traits | ||
| absl::atomic_hook | ||
| absl::bits | ||
| absl::flags_commandlineflag_internal | ||
| absl::hash_container_defaults | ||
| absl::numeric_representation | ||
| absl::node_slot_policy | ||
| absl::core_headers | ||
| absl::dynamic_annotations | ||
| absl::utf8_for_code_point | ||
| absl::errno_saver | ||
| absl::absl_check | ||
| absl::hash_function_defaults | ||
| absl::function_ref | ||
| absl::city | ||
| absl::low_level_hash | ||
| absl::fixed_array | ||
| absl::variant | ||
| absl::meta | ||
| absl::log_internal_voidify | ||
| absl::log_sink | ||
| absl::log_internal_log_sink_set | ||
| absl::log_sink_registry | ||
| absl::log_entry | ||
| absl::log_globals | ||
| absl::log_internal_nullguard | ||
| absl::examine_stack | ||
| absl::inlined_vector | ||
| absl::log_internal_proto | ||
| absl::strerror | ||
| absl::log_internal_config | ||
| absl::raw_hash_map | ||
| absl::raw_hash_set | ||
| absl::container_memory | ||
| absl::algorithm_container | ||
| absl::span | ||
| absl::log_internal_nullstream | ||
| absl::vlog_config_internal | ||
| absl::flags_reflection | ||
| absl::flags_internal | ||
| absl::flags_config | ||
| absl::fast_type_id | ||
| absl::utility | ||
| absl::time_zone | ||
| absl::civil_time | ||
| absl::string_view | ||
| absl::throw_delegate | ||
| absl::memory | ||
| absl::charset | ||
| absl::endian | ||
| absl::config) |
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,42 @@ | ||
| diff --git a/cmake/onnxruntime_providers_coreml.cmake b/cmake/onnxruntime_providers_coreml.cmake | ||
| index 0aa25a2..1cb768c 100644 | ||
| --- a/cmake/onnxruntime_providers_coreml.cmake | ||
| +++ b/cmake/onnxruntime_providers_coreml.cmake | ||
| @@ -9,7 +9,7 @@ add_compile_definitions(USE_COREML=1) | ||
|
|
||
| # Check if we can build the coremltools code for creating an mlpackage with an mlprogram. | ||
| # The coremltools source requires std::filesystem::path which is only available from iOS 13 on. | ||
| -set(_enable_ML_PROGRAM ON) | ||
| +option(_enable_ML_PROGRAM "..." ON) | ||
| if (IOS AND CMAKE_OSX_DEPLOYMENT_TARGET VERSION_LESS 13.0) | ||
| message(WARNING "CoreML ML Program is not supported on iOS < 13.0. Excluding ML Program support from build.") | ||
| set(_enable_ML_PROGRAM OFF) | ||
| @@ -29,7 +29,9 @@ if (_enable_ML_PROGRAM) | ||
| endif() | ||
|
|
||
| # Compile CoreML proto definition to ${CMAKE_CURRENT_BINARY_DIR}/coreml_proto | ||
| -set(COREML_PROTO_ROOT ${coremltools_SOURCE_DIR}/mlmodel/format) | ||
| +if(NOT DEFINED COREML_PROTO_ROOT) | ||
| + set(COREML_PROTO_ROOT ${coremltools_SOURCE_DIR}/mlmodel/format) | ||
| +endif() | ||
| file(GLOB coreml_proto_srcs "${COREML_PROTO_ROOT}/*.proto") | ||
|
|
||
| onnxruntime_add_static_library(coreml_proto ${coreml_proto_srcs}) | ||
| @@ -62,7 +64,7 @@ endif() | ||
| # Separate source_group for each as the .proto files are in the repo and the .cc/.h files are generated in the build | ||
| # output directory. | ||
| set_target_properties(coreml_proto PROPERTIES FOLDER "External") | ||
| -source_group(TREE ${COREML_PROTO_ROOT} PREFIX coreml_proto FILES ${coreml_proto_srcs}) | ||
| +# source_group(TREE ${COREML_PROTO_ROOT} PREFIX coreml_proto FILES ${coreml_proto_srcs}) | ||
|
|
||
| # filter to the generated .cc/.h files | ||
| get_target_property(coreml_proto_generated_srcs coreml_proto SOURCES) | ||
| @@ -119,7 +121,7 @@ if(_enable_ML_PROGRAM) | ||
| ${onnxruntime_providers_coreml_modelpackage_cc_srcs} | ||
| ) | ||
|
|
||
| - source_group(TREE ${coremltools_SOURCE_DIR} PREFIX coremltools FILES ${coremltools_srcs}) | ||
| + # source_group(TREE ${coremltools_SOURCE_DIR} PREFIX coremltools FILES ${coremltools_srcs}) | ||
| endif() | ||
|
|
||
| # Add CoreML objective c++ source code |
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
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