diff options
author | Eanna O Cathain <eanna.ocathain@arm.com> | 2022-03-03 15:58:10 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-04-13 14:27:44 +0000 |
commit | 2f0ddb67d8f9267ab600a8a26308cab32f9e16ac (patch) | |
tree | 0eab15a96ebf1e21e60347804a08144380e53800 /samples/common/cmake/find_armnn.cmake | |
parent | bab8fa9a11cf3bfef4b72fb757b81575b6fd75f0 (diff) | |
download | armnn-2f0ddb67d8f9267ab600a8a26308cab32f9e16ac.tar.gz |
MLECO-2492 Add CPP OD example with TFLITE-ArmnnDelegate
Signed-off-by: Dvir Markovich <dvir.markovich@arm.com>
Change-Id: If412c15ba49abe8370a570260b0a8ed8de305b7c
Diffstat (limited to 'samples/common/cmake/find_armnn.cmake')
-rw-r--r-- | samples/common/cmake/find_armnn.cmake | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/samples/common/cmake/find_armnn.cmake b/samples/common/cmake/find_armnn.cmake index 289e9127f6..35f87ebf19 100644 --- a/samples/common/cmake/find_armnn.cmake +++ b/samples/common/cmake/find_armnn.cmake @@ -2,8 +2,13 @@ # SPDX-License-Identifier: MIT # Search for ArmNN built libraries in user-provided path first, then current repository, then system -set(ARMNN_LIB_NAMES "libarmnn.so" - "libarmnnTfLiteParser.so") +if( USE_ARMNN_DELEGATE ) + set(ARMNN_LIB_NAMES "libarmnn.so" + "libarmnnDelegate.so") +else() + set(ARMNN_LIB_NAMES "libarmnn.so" + "libarmnnTfLiteParser.so") +endif() set(ARMNN_LIBS "") @@ -26,7 +31,13 @@ foreach(armnn_lib ${ARMNN_LIB_NAMES}) list(APPEND ARMNN_LIBS ${ARMNN_${armnn_lib}}) get_filename_component(LIB_DIR ${ARMNN_${armnn_lib}} DIRECTORY) get_filename_component(LIB_PARENT_DIR ${LIB_DIR} DIRECTORY) - set(ARMNN_INCLUDE_DIR ${LIB_PARENT_DIR}/include) + if( USE_ARMNN_DELEGATE ) + set(ARMNN_INCLUDE_DIR ${LIB_PARENT_DIR}/include + ${PARENT_DIR}/../delegate/include + ${PARENT_DIR}/../delegate/src) + else() + set(ARMNN_INCLUDE_DIR ${LIB_PARENT_DIR}/include) + endif() endif() endforeach() |