blob: 289e9127f6d06aba2130b33285f543b51b0c1ebc (
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
|
# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
# 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")
set(ARMNN_LIBS "")
get_filename_component(PARENT_DIR ${PROJECT_SOURCE_DIR} DIRECTORY)
get_filename_component(REPO_DIR ${PARENT_DIR} DIRECTORY)
foreach(armnn_lib ${ARMNN_LIB_NAMES})
find_library(ARMNN_${armnn_lib}
NAMES
${armnn_lib}
HINTS
${ARMNN_LIB_DIR} ${REPO_DIR}
PATHS
${ARMNN_LIB_DIR} ${REPO_DIR}
PATH_SUFFIXES
"lib"
"lib64")
if(ARMNN_${armnn_lib})
message("Found library ${ARMNN_${armnn_lib}}")
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)
endif()
endforeach()
if(NOT ARMNN_LIBS)
message(FATAL_ERROR "Could not find ArmNN libraries ${ARMNN_LIB_NAMES}")
endif()
|
