blob: 46fab3bc5bba3e2bf6fb429230fda26815ce655c [file] [log] [blame]
#
# 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()