diff options
| -rw-r--r-- | delegate/CMakeLists.txt | 2 | ||||
| -rw-r--r-- | packaging/armnn.spec | 7 |
2 files changed, 8 insertions, 1 deletions
diff --git a/delegate/CMakeLists.txt b/delegate/CMakeLists.txt index 38f7bd10e..c2d25ac6e 100644 --- a/delegate/CMakeLists.txt +++ b/delegate/CMakeLists.txt @@ -70,7 +70,7 @@ target_include_directories(armnnDelegate find_package(TfLite REQUIRED MODULE) target_link_libraries(armnnDelegate - ${TfLite_LIB}) + ${TfLite_LIB} pthread dl) # Various tflite header files are not warning clean # We can't change compilation flags on header files directly, so we need to add them to an interface library first diff --git a/packaging/armnn.spec b/packaging/armnn.spec index fb22624f9..37045470c 100644 --- a/packaging/armnn.spec +++ b/packaging/armnn.spec @@ -39,6 +39,7 @@ BuildRequires: boost-devel %if 0%{?TF_LITE_SUPPORT} == 1 BuildRequires: tensorflow-lite-devel BuildRequires: tensorflow-lite-flatbuf-schema +BuildRequires: tensorflow2-lite-devel %endif %endif @@ -118,10 +119,16 @@ cmake -DARMCOMPUTENEON=1 \ -DBUILD_TF_LITE_PARSER=1 \ -DTF_LITE_GENERATED_PATH=./externals/tflite/usr/include/tensorflow/lite/schema \ -DTF_LITE_SCHEMA_PATH=./externals/tflite/usr/share/tensorflow/lite/schema \ + -DTensorflow_INCLUDE_DIR=/usr/include/ \ + -DTfLite_INCLUDE_DIR=/usr/include/tensorflow2/ \ + -DTensorflow_LIB=/usr/lib/ \ + -DTfLite_LIB=/usr/lib/libtensorflow2-lite.a \ + -DTfLite_Schema_INCLUDE_PATH=./externals/tflite/usr/share/tensorflow/lite/schema \ -DFLATBUFFERS_INCLUDE_PATH=/usr/include/ \ -DFLATC=/usr/bin/flatc \ %endif # TF_LITE_SUPPORT -DBUILD_TESTS=1 \ + -DBUILD_ARMNN_TFLITE_DELEGATE=1 \ -DARMCOMPUTE_CORE_LIBRARY_DEBUG=%{_libdir}/libarm_compute_core.so \ -DARMCOMPUTE_CORE_LIBRARY_RELEASE=%{_libdir}/libarm_compute_core.so \ -DARMCOMPUTE_LIBRARY_DEBUG=%{_libdir}/libarm_compute.so \ |