blob: 22c583a3238718923518132b70ee840a5cddc839 (
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#
# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
# SPDX-License-Identifier: MIT
#
include(FindPackageHandleStandardArgs)
unset(TFLITE_FOUND)
#
# NOTE: this module is used to find the tensorflow lite binary libraries only
# the FindTfLiteSrc.cmake module is used to find the tensorflow lite include directory.
# This is to allow components like the Tensorflow lite parser that have a source dependency
# on tensorflow lite headers but no need to link to the binary libraries to use only the sources
# and not have an artificial dependency on the libraries.
#
# 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 not found then, look for the dynamic library of tensorflow lite
find_library(TfLite_LIB NAMES "libtensorflow_lite_all.so" "libtensorflowlite.so" HINTS ${TFLITE_LIB_ROOT} ${TFLITE_LIB_ROOT}/tensorflow/lite)
# If the static library was found, gather all of its dependencies
if (TfLite_LIB MATCHES .a$)
message("-- Static tensorflow lite library found, using for ArmNN build")
find_library(TfLite_abseilstrings_LIB "libabsl_strings.a"
PATH ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/strings)
find_library(TfLite_abseil_synchronization_LIB "libabsl_synchronization.a"
PATH ${TFLITE_LIB_ROOT}/_deps/abseil-cpp-build/absl/synchronization)
find_library(TfLite_farmhash_LIB "libfarmhash.a"
PATH ${TFLITE_LIB_ROOT}/_deps/farmhash-build)
find_library(TfLite_fftsg_LIB "libfft2d_fftsg.a"
PATH ${TFLITE_LIB_ROOT}/_deps/fft2d-build)
find_library(TfLite_fftsg2d_LIB "libfft2d_fftsg2d.a"
PATH ${TFLITE_LIB_ROOT}/_deps/fft2d-build)
find_library(TfLite_flatbuffers_LIB "libflatbuffers.a"
PATH ${TFLITE_LIB_ROOT}/_deps/flatbuffers-build)
find_library(TfLite_cpuinfo_LIB "libcpuinfo.a" PATH
${TFLITE_LIB_ROOT}/_deps/cpuinfo-build)
find_library(TfLite_clog_LIB "libclog.a" PATH
${TFLITE_LIB_ROOT}/_deps/clog-build)
# All remaining libraries are part of libruy.
find_library(TfLite_ruy_allocator_LIB "libruy_allocator.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_apply_multiplier_LIB "libruy_apply_multiplier.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_blocking_counter_LIB "libruy_blocking_counter.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_block_map_LIB "libruy_block_map.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_context_LIB "libruy_context.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_context_get_ctx_LIB "libruy_context_get_ctx.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_cpuinfo_LIB "libruy_cpuinfo.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_ctx_LIB "libruy_ctx.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_denormal_LIB "libruy_denormal.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_frontend_LIB "libruy_frontend.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_have_built_path_for_avx2_fma_LIB "libruy_have_built_path_for_avx2_fma.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_have_built_path_for_avx512_LIB "libruy_have_built_path_for_avx512.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_have_built_path_for_avx_LIB "libruy_have_built_path_for_avx.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_kernel_arm_LIB "libruy_kernel_arm.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_kernel_avx2_fma_LIB "libruy_kernel_avx2_fma.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_kernel_avx512_LIB "libruy_kernel_avx512.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_kernel_avx_LIB "libruy_kernel_avx.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_pack_arm_LIB "libruy_pack_arm.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_pack_avx2_fma_LIB "libruy_pack_avx2_fma.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_pack_avx512_LIB "libruy_pack_avx512.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_pack_avx_LIB "libruy_pack_avx.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_prepacked_cache_LIB "libruy_prepacked_cache.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_prepare_packed_matrices_LIB "libruy_prepare_packed_matrices.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_system_aligned_alloc_LIB "libruy_system_aligned_alloc.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_threadpool_LIB "libruy_thread_pool.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_trmul_LIB "libruy_trmul.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_tune_LIB "libruy_tune.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_wait_LIB "libruy_wait.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy)
find_library(TfLite_ruy_profiler_LIB "libruy_profiler_instrumentation.a" PATH
${TFLITE_LIB_ROOT}/_deps/ruy-build/ruy/profiler)
## Set TFLITE_FOUND if all libraries are satisfied for static lib
find_package_handle_standard_args(TfLite DEFAULT_MSG TfLite_LIB TfLite_abseilstrings_LIB TfLite_farmhash_LIB TfLite_fftsg_LIB TfLite_fftsg2d_LIB
TfLite_flatbuffers_LIB TfLite_ruy_allocator_LIB TfLite_ruy_apply_multiplier_LIB TfLite_ruy_blocking_counter_LIB
TfLite_ruy_block_map_LIB TfLite_ruy_context_LIB TfLite_ruy_context_get_ctx_LIB TfLite_ruy_cpuinfo_LIB
TfLite_ruy_ctx_LIB TfLite_ruy_denormal_LIB TfLite_ruy_frontend_LIB TfLite_ruy_have_built_path_for_avx2_fma_LIB
TfLite_ruy_have_built_path_for_avx512_LIB TfLite_ruy_have_built_path_for_avx_LIB TfLite_ruy_kernel_arm_LIB
TfLite_ruy_kernel_avx2_fma_LIB TfLite_ruy_kernel_avx512_LIB TfLite_ruy_kernel_avx_LIB TfLite_ruy_pack_arm_LIB
TfLite_ruy_pack_avx2_fma_LIB TfLite_ruy_pack_avx512_LIB TfLite_ruy_pack_avx_LIB TfLite_ruy_prepacked_cache_LIB
TfLite_ruy_prepare_packed_matrices_LIB TfLite_ruy_system_aligned_alloc_LIB TfLite_ruy_threadpool_LIB
TfLite_ruy_trmul_LIB TfLite_ruy_tune_LIB TfLite_ruy_wait_LIB TfLite_ruy_profiler_LIB TfLite_cpuinfo_LIB TfLite_clog_LIB
TfLite_abseil_synchronization_LIB)
# Set external variables for usage in CMakeLists.txt
if (TFLITE_FOUND)
# WARNING! The order of these libraries is critical. Moving them
# around will result in linker errors in DelegateUnitTests.
set(TfLite_LIB ${TfLite_LIB} ${TfLite_abseilstrings_LIB} ${TfLite_farmhash_LIB} ${TfLite_fftsg_LIB} ${TfLite_fftsg2d_LIB} ${TfLite_ruy_threadpool_LIB}
${TfLite_flatbuffers_LIB} ${TfLite_ruy_allocator_LIB} ${TfLite_ruy_apply_multiplier_LIB} ${TfLite_ruy_blocking_counter_LIB}
${TfLite_ruy_frontend_LIB} ${TfLite_ruy_trmul_LIB} ${TfLite_ruy_block_map_LIB} ${TfLite_ruy_context_LIB}
${TfLite_ruy_context_get_ctx_LIB} ${TfLite_ruy_cpuinfo_LIB} ${TfLite_ruy_ctx_LIB} ${TfLite_ruy_denormal_LIB}
${TfLite_ruy_have_built_path_for_avx2_fma_LIB} ${TfLite_ruy_have_built_path_for_avx512_LIB}
${TfLite_ruy_have_built_path_for_avx_LIB} ${TfLite_ruy_kernel_arm_LIB} ${TfLite_ruy_kernel_avx2_fma_LIB}
${TfLite_ruy_kernel_avx512_LIB} ${TfLite_ruy_kernel_avx_LIB} ${TfLite_ruy_pack_arm_LIB}
${TfLite_ruy_pack_avx2_fma_LIB} ${TfLite_ruy_pack_avx512_LIB} ${TfLite_ruy_pack_avx_LIB} ${TfLite_ruy_prepacked_cache_LIB}
${TfLite_ruy_prepare_packed_matrices_LIB} ${TfLite_ruy_system_aligned_alloc_LIB}
${TfLite_ruy_tune_LIB} ${TfLite_ruy_wait_LIB} ${TfLite_ruy_profiler_LIB}
${TfLite_cpuinfo_LIB} ${TfLite_clog_LIB} ${TfLite_abseil_synchronization_LIB})
endif ()
elseif (TfLite_LIB MATCHES .so$)
message("-- Dynamic tensorflow lite library found, using for ArmNN build")
find_package_handle_standard_args(TfLite DEFAULT_MSG TfLite_LIB)
## Set external variables for usage in CMakeLists.txt
if (TFLITE_FOUND)
set(TfLite_LIB ${TfLite_LIB})
endif ()
else()
message(FATAL_ERROR "Could not find a tensorflow lite library to use")
endif()
|
