diff options
| author | Inki Dae <inki.dae@samsung.com> | 2020-12-21 16:48:37 +0900 |
|---|---|---|
| committer | Inki Dae <inki.dae@samsung.com> | 2020-12-21 16:48:37 +0900 |
| commit | 08e363cd3c87e4af2a7947b8f9dd6e9f144e2e95 (patch) | |
| tree | 4d177b2d7aef64bc36d5e2ce3e4a4598eb3aae2c | |
| parent | 8af6849ff5f07041874f6e6bc08113ae71aad897 (diff) | |
| download | armnn-sandbox/daeinki/armnn_20_11_gpu_delegate.tar.gz armnn-sandbox/daeinki/armnn_20_11_gpu_delegate.tar.bz2 armnn-sandbox/daeinki/armnn_20_11_gpu_delegate.zip | |
Add tflite gpu delegate supportsandbox/daeinki/armnn_20_11_gpu_delegate
Change-Id: I5631bc251521e482308a3fcfebe52cf26594e4d9
Signed-off-by: Inki Dae <inki.dae@samsung.com>
| -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 \ |