diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2023-05-03 16:08:11 +0100 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2023-05-05 15:45:35 +0000 |
commit | 26654cb71db2b1e163527f52c3198d9434bb0e37 (patch) | |
tree | c7406ad3edde1e914c9474bbedd4a21c62d612b4 /delegate/cmake | |
parent | 74be13e0d663b9127f630360fffa55993ccce9d6 (diff) | |
download | armnn-26654cb71db2b1e163527f52c3198d9434bb0e37.tar.gz |
IVGCVSW-7423 Add ArmnnDelegatePlugin
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: Ie02021ac56a512598760e4c6d05ef1a80f4aec8d
Diffstat (limited to 'delegate/cmake')
-rw-r--r-- | delegate/cmake/Modules/FindTfLiteAbsl.cmake | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/delegate/cmake/Modules/FindTfLiteAbsl.cmake b/delegate/cmake/Modules/FindTfLiteAbsl.cmake new file mode 100644 index 0000000000..46fab3bc5b --- /dev/null +++ b/delegate/cmake/Modules/FindTfLiteAbsl.cmake @@ -0,0 +1,62 @@ +# +# Copyright © 2023 Arm Ltd and Contributors. All rights reserved. +# SPDX-License-Identifier: MIT +# + +include(FindPackageHandleStandardArgs) +unset(TFLITEABSL_FOUND) + +find_path(TfLite_ABSL_SYNC_HEADERS + NAMES + absl + HINTS + ${TFLITE_LIB_ROOT}/abseil-cpp) + +# First look for the static version of tensorflow lite +find_library(TfLite_LIB NAMES "libtensorflow-lite.a" HINTS ${TFLITE_LIB_ROOT} ${TFLITE_LIB_ROOT}/tensorflow/lite) + +# If the static library was found, gather extra absl libraries for opaque delegate +if (TfLite_LIB MATCHES .a$) + find_library(TfLite_abseil_base_LIB "libabsl_base.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/base) + find_library(TfLite_abseil_log_severity_LIB "libabsl_log_severity.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/base) + find_library(TfLite_abseil_spinlock_wait_LIB "libabsl_spinlock_wait.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/base) + find_library(TfLite_abseil_malloc_internal_LIB "libabsl_malloc_internal.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/base) + find_library(TfLite_abseil_raw_logging_internal_LIB "libabsl_raw_logging_internal.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/base) + find_library(TfLite_abseil_stacktrace_LIB "libabsl_stacktrace.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/debugging) + find_library(TfLite_abseil_debugging_internal_LIB "libabsl_debugging_internal.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/debugging) + find_library(TfLite_abseil_symbolize_LIB "libabsl_symbolize.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/debugging) + find_library(TfLite_abseil_demangle_internal_LIB "libabsl_demangle_internal.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/debugging) + find_library(TfLite_abseil_time_LIB "libabsl_time.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/time) + find_library(TfLite_abseil_time_zone_LIB "libabsl_time_zone.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/time) + find_library(TfLite_abseil_int128_LIB "libabsl_int128.a" PATH + ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/numeric) + + ## Set TFLITEABSL_FOUND + find_package_handle_standard_args(TfLiteAbsl DEFAULT_MSG TfLite_ABSL_SYNC_HEADERS TfLite_abseil_base_LIB + TfLite_abseil_int128_LIB TfLite_abseil_spinlock_wait_LIB + TfLite_abseil_raw_logging_internal_LIB TfLite_abseil_malloc_internal_LIB + TfLite_abseil_symbolize_LIB TfLite_abseil_stacktrace_LIB + TfLite_abseil_demangle_internal_LIB TfLite_abseil_debugging_internal_LIB + TfLite_abseil_time_LIB TfLite_abseil_time_zone_LIB) + + ## Set external variables for usage in CMakeLists.txt + if(TFLITEABSL_FOUND) + set(TfLite_ABSL_SYNC_HEADERS ${TfLite_ABSL_SYNC_HEADERS}) + set(TfLite_Extra_Absl_LIB ${TfLite_abseil_base_LIB} ${TfLite_abseil_int128_LIB} ${TfLite_abseil_spinlock_wait_LIB} + ${TfLite_abseil_raw_logging_internal_LIB} ${TfLite_abseil_malloc_internal_LIB} + ${TfLite_abseil_symbolize_LIB} ${TfLite_abseil_stacktrace_LIB} + ${TfLite_abseil_demangle_internal_LIB} ${TfLite_abseil_debugging_internal_LIB} + ${TfLite_abseil_time_LIB} ${TfLite_abseil_time_zone_LIB}) + endif() +endif() |