aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/CMakeLists.txt
blob: ff253787a1e8ae965bb9281f3b70de88fda5b244 (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
36
37
38
39
40
41
#
# Copyright © 2017 Arm Ltd. All rights reserved.
# SPDX-License-Identifier: MIT
#

if(ARMCOMPUTEREF)
    list(APPEND armnnRefBackend_sources
        RefBackend.cpp
        RefBackend.hpp
        RefBackendId.hpp
        RefTensorHandle.hpp
        RefTensorHandle.cpp
        RefLayerSupport.cpp
        RefLayerSupport.hpp
        RefMemoryManager.hpp
        RefMemoryManager.cpp
        RefRegistryInitializer.cpp
        RefWorkloadFactory.cpp
        RefWorkloadFactory.hpp
        RefTensorHandleFactory.cpp
        RefTensorHandleFactory.hpp
    )

    add_subdirectory(workloads)

    if(BUILD_UNIT_TESTS)
        add_subdirectory(test)
    endif()

else()
    list(APPEND armnnRefBackend_sources
        RefBackendId.hpp
        RefLayerSupport.cpp
        RefLayerSupport.hpp
    )
endif()

add_library(armnnRefBackend OBJECT ${armnnRefBackend_sources})
target_include_directories(armnnRefBackend PRIVATE ${PROJECT_SOURCE_DIR}/src/armnn)
target_include_directories(armnnRefBackend PRIVATE ${PROJECT_SOURCE_DIR}/src/armnnUtils)
target_include_directories(armnnRefBackend PRIVATE ${PROJECT_SOURCE_DIR}/src/backends)