• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
2# SPDX-License-Identifier: MIT
3# Search for ArmNN built libraries in user-provided path first, then current repository, then system
4
5set(ARMNN_LIB_NAMES "libarmnn.so"
6    "libarmnnTfLiteParser.so")
7
8set(ARMNN_LIBS "")
9
10get_filename_component(PARENT_DIR ${PROJECT_SOURCE_DIR} DIRECTORY)
11get_filename_component(REPO_DIR ${PARENT_DIR} DIRECTORY)
12
13foreach(armnn_lib ${ARMNN_LIB_NAMES})
14    find_library(ARMNN_${armnn_lib}
15        NAMES
16            ${armnn_lib}
17        HINTS
18            ${ARMNN_LIB_DIR} ${REPO_DIR}
19        PATHS
20            ${ARMNN_LIB_DIR} ${REPO_DIR}
21        PATH_SUFFIXES
22            "lib"
23            "lib64")
24    if(ARMNN_${armnn_lib})
25        message("Found library ${ARMNN_${armnn_lib}}")
26        list(APPEND ARMNN_LIBS ${ARMNN_${armnn_lib}})
27        get_filename_component(LIB_DIR ${ARMNN_${armnn_lib}} DIRECTORY)
28        get_filename_component(LIB_PARENT_DIR ${LIB_DIR} DIRECTORY)
29        set(ARMNN_INCLUDE_DIR ${LIB_PARENT_DIR}/include)
30    endif()
31endforeach()
32
33if(NOT ARMNN_LIBS)
34    message(FATAL_ERROR "Could not find ArmNN libraries ${ARMNN_LIB_NAMES}")
35endif()
36