diff options
Diffstat (limited to 'src/backends/reference/CMakeLists.txt')
| -rw-r--r-- | src/backends/reference/CMakeLists.txt | 53 |
1 files changed, 31 insertions, 22 deletions
diff --git a/src/backends/reference/CMakeLists.txt b/src/backends/reference/CMakeLists.txt index 6852ab0b74..ff253787a1 100644 --- a/src/backends/reference/CMakeLists.txt +++ b/src/backends/reference/CMakeLists.txt @@ -3,30 +3,39 @@ # SPDX-License-Identifier: MIT # -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.hpp - RefTensorHandleFactory.cpp -) +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) - -add_subdirectory(workloads) - -if(BUILD_UNIT_TESTS) - add_subdirectory(test) -endif() |