diff options
Diffstat (limited to 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 488 |
1 files changed, 488 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt new file mode 100644 index 0000000000..d09e549106 --- /dev/null +++ b/CMakeLists.txt @@ -0,0 +1,488 @@ +cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake. +project(armnn) + +set(additional_cmake_files) +list(APPEND additional_cmake_files + cmake/Utils.cmake + cmake/GlobalConfig.cmake + cmake/AddDllCopyCommands.cmake) + +foreach(cmake_file ${additional_cmake_files}) + include(${cmake_file}) +endforeach() + +if (BUILD_TESTS) + add_subdirectory(tests) +endif() + +# Include the additional cmake files in their own target so that they will appear nicely in IDEs +add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files}) + +set(armnnUtils_sources) +list(APPEND armnnUtils_sources + src/armnnUtils/GraphTopologicalSort.hpp + src/armnnUtils/Logging.hpp + src/armnnUtils/Permute.hpp + src/armnnUtils/Logging.cpp + src/armnnUtils/Permute.cpp + ) +if(BUILD_CAFFE_PARSER) + list(APPEND armnnUtils_sources + src/armnnUtils/ParserPrototxtFixture.hpp + ) +endif() + +add_library_ex(armnnUtils STATIC ${armnnUtils_sources}) + +if(BUILD_CAFFE_PARSER) + # ArmNN Parser source files required for all build options + set(armnn_caffe_parser_sources) + list(APPEND armnn_caffe_parser_sources + include/armnnCaffeParser/ICaffeParser.hpp + src/armnnCaffeParser/CaffeParser.hpp + src/armnnCaffeParser/CaffeParser.cpp + ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc + ) + # The generated Caffe protobuf .cc file is not warning clean and we can't fix them. + if(COMPILER_IS_GNU_LIKE) + set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion") + endif() + + add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources}) + set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}") + + target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils) + + target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY}) + + target_link_libraries(armnnCaffeParser armnn) + target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES}) +endif() + +# ArmNN source files required for all build options +list(APPEND armnn_sources + include/armnn/ArmNN.hpp + include/armnn/Descriptors.hpp + include/armnn/DescriptorsFwd.hpp + include/armnn/IRuntime.hpp + include/armnn/INetwork.hpp + include/armnn/Tensor.hpp + include/armnn/TensorFwd.hpp + include/armnn/Types.hpp + include/armnn/TypesUtils.hpp + include/armnn/Utils.hpp + include/armnn/LayerSupport.hpp + include/armnn/Version.hpp + src/armnn/backends/ClWorkloadFactory.hpp + src/armnn/backends/ClWorkloadFactory.cpp + src/armnn/backends/ClLayerSupport.cpp + src/armnn/backends/ClLayerSupport.hpp + src/armnn/backends/CpuTensorHandleFwd.hpp + src/armnn/backends/CpuTensorHandle.hpp + src/armnn/backends/CpuTensorHandle.cpp + src/armnn/backends/RefWorkloadFactory.cpp + src/armnn/backends/RefWorkloadFactory.hpp + src/armnn/backends/RefLayerSupport.cpp + src/armnn/backends/RefLayerSupport.hpp + src/armnn/backends/MakeWorkloadHelper.hpp + src/armnn/backends/NeonWorkloadFactory.cpp + src/armnn/backends/NeonWorkloadFactory.hpp + src/armnn/backends/NeonLayerSupport.cpp + src/armnn/backends/NeonLayerSupport.hpp + src/armnn/backends/Workload.hpp + src/armnn/backends/WorkloadDataFwd.hpp + src/armnn/backends/WorkloadData.hpp + src/armnn/backends/WorkloadData.cpp + src/armnn/backends/WorkloadFactory.hpp + src/armnn/backends/WorkloadFactory.cpp + src/armnn/backends/WorkloadInfo.hpp + src/armnn/backends/MemCopyWorkload.cpp + src/armnn/backends/MemCopyWorkload.hpp + src/armnn/backends/RefWorkloads/Broadcast.hpp + src/armnn/backends/RefWorkloads/Broadcast.cpp + src/armnn/backends/RefWorkloads/RefMergerUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefConstantUint8Workload.hpp + src/armnn/backends/RefWorkloads/Addition.hpp + src/armnn/backends/RefWorkloads/ConvImpl.hpp + src/armnn/backends/RefWorkloads/RefResizeBilinearUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefMultiplicationUint8Workload.hpp + src/armnn/backends/RefWorkloads/FullyConnected.cpp + src/armnn/backends/RefWorkloads/RefFullyConnectedFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefSoftmaxFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefBaseConstantWorkload.hpp + src/armnn/backends/RefWorkloads/RefConvolution2dUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefSplitterUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefActivationFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefBatchNormalizationUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefResizeBilinearUint8Workload.hpp + src/armnn/backends/RefWorkloads/Multiplication.cpp + src/armnn/backends/RefWorkloads/RefMultiplicationUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefL2NormalizationFloat32Workload.hpp + src/armnn/backends/RefWorkloads/Multiplication.hpp + src/armnn/backends/RefWorkloads/RefActivationUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefBaseConstantWorkload.cpp + src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefResizeBilinearFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefBatchNormalizationFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefPooling2dFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp + src/armnn/backends/RefWorkloads/RefMergerUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefFullyConnectedFloat32Workload.hpp + src/armnn/backends/RefWorkloads/Softmax.hpp + src/armnn/backends/RefWorkloads/RefMergerFloat32Workload.hpp + src/armnn/backends/RefWorkloads/Addition.cpp + src/armnn/backends/RefWorkloads/RefFakeQuantizationFloat32Workload.cpp + src/armnn/backends/RefWorkloads/TensorBufferArrayView.hpp + src/armnn/backends/RefWorkloads/ResizeBilinear.cpp + src/armnn/backends/RefWorkloads/RefBatchNormalizationFloat32Workload.hpp + src/armnn/backends/RefWorkloads/Splitter.hpp + src/armnn/backends/RefWorkloads/RefFullyConnectedUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefSoftmaxUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefReshapeFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dUint8Workload.hpp + src/armnn/backends/RefWorkloads/FullyConnected.hpp + src/armnn/backends/RefWorkloads/RefFloorFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefPooling2dUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefFloorFloat32Workload.cpp + src/armnn/backends/RefWorkloads/ConvImpl.cpp + src/armnn/backends/RefWorkloads/RefSoftmaxFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefSoftmaxUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefReshapeUint8Workload.hpp + src/armnn/backends/RefWorkloads/Activation.cpp + src/armnn/backends/RefWorkloads/RefResizeBilinearFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefReshapeUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefMultiplicationFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefL2NormalizationFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefConvolution2dFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefBatchNormalizationUint8Workload.hpp + src/armnn/backends/RefWorkloads/RefConvolution2dUint8Workload.cpp + src/armnn/backends/RefWorkloads/ResizeBilinear.hpp + src/armnn/backends/RefWorkloads/RefNormalizationFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefSplitterFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefActivationUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefPooling2dUint8Workload.hpp + src/armnn/backends/RefWorkloads/BatchNormImpl.hpp + src/armnn/backends/RefWorkloads/RefSplitterUint8Workload.cpp + src/armnn/backends/RefWorkloads/Activation.hpp + src/armnn/backends/RefWorkloads/RefPooling2dFloat32Workload.cpp + src/armnn/backends/RefWorkloads/Merger.hpp + src/armnn/backends/RefWorkloads/RefSplitterFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefConstantFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefActivationFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefMultiplicationFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefReshapeFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefNormalizationFloat32Workload.cpp + src/armnn/backends/RefWorkloads/Softmax.cpp + src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefConstantUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefConstantFloat32Workload.cpp + src/armnn/backends/RefWorkloads/Pooling2d.cpp + src/armnn/backends/RefWorkloads/RefAdditionFloat32Workload.cpp + src/armnn/backends/RefWorkloads/RefConvolution2dFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefAdditionFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefMergerFloat32Workload.cpp + src/armnn/backends/RefWorkloads/Pooling2d.hpp + src/armnn/backends/RefWorkloads/RefFullyConnectedUint8Workload.cpp + src/armnn/backends/RefWorkloads/RefFakeQuantizationFloat32Workload.hpp + src/armnn/backends/RefWorkloads/RefPermuteWorkload.hpp + src/armnn/backends/RefWorkloads/RefPermuteWorkload.cpp + src/armnn/InternalTypes.hpp + src/armnn/InternalTypes.cpp + src/armnn/LayerFwd.hpp + src/armnn/Layer.hpp + src/armnn/Layer.cpp + src/armnn/LayersFwd.hpp + src/armnn/Layers.hpp + src/armnn/Layers.cpp + src/armnn/Runtime.hpp + src/armnn/Runtime.cpp + src/armnn/Descriptors.cpp + src/armnn/LoadedNetwork.hpp + src/armnn/LoadedNetwork.cpp + src/armnn/Exceptions.cpp + src/armnn/Graph.hpp + src/armnn/Graph.cpp + src/armnn/Network.hpp + src/armnn/Network.cpp + src/armnn/backends/OutputHandler.hpp + src/armnn/backends/OutputHandler.cpp + src/armnn/Profiling.cpp + src/armnn/Tensor.cpp + src/armnn/Utils.cpp + src/armnn/LayerSupport.cpp + src/armnn/LayerSupportCommon.hpp + src/armnn/optimizations/All.hpp + src/armnn/optimizations/MovePermuteUp.hpp + src/armnn/optimizations/Optimization.hpp + src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp + src/armnn/optimizations/OptimizeInversePermutes.hpp + src/armnn/optimizations/PermuteAsReshape.hpp + src/armnn/optimizations/SquashEqualSiblings.hpp + src/armnn/Optimizer.hpp + src/armnn/Optimizer.cpp + ) + +if(ARMCOMPUTENEON) + # Additionally include source files for ARM Compute NEON backend + list(APPEND armnn_sources + src/armnn/backends/NeonWorkloads/NeonActivationFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonActivationFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonActivationUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonActivationUint8Workload.hpp + src/armnn/backends/NeonWorkloads/NeonAdditionFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonAdditionFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonBaseConstantWorkload.hpp + src/armnn/backends/NeonWorkloads/NeonBaseMergerWorkload.hpp + src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp + src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonConstantFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonConstantFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonConstantUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonConstantUint8Workload.hpp + src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.cpp + src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp + src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp + src/armnn/backends/NeonWorkloads/NeonFloorFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonFloorFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonFullyConnectedFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonFullyConnectedFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonMergerFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonMergerFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonMergerUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonMergerUint8Workload.hpp + src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonNormalizationFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonNormalizationFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonPermuteWorkload.cpp + src/armnn/backends/NeonWorkloads/NeonPermuteWorkload.hpp + src/armnn/backends/NeonWorkloads/NeonPooling2dBaseWorkload.cpp + src/armnn/backends/NeonWorkloads/NeonPooling2dBaseWorkload.hpp + src/armnn/backends/NeonWorkloads/NeonPooling2dFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonPooling2dFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonPooling2dUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonPooling2dUint8Workload.hpp + src/armnn/backends/NeonWorkloads/NeonReshapeFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonReshapeFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonReshapeUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonReshapeUint8Workload.hpp + src/armnn/backends/NeonWorkloads/NeonSoftmaxFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonSoftmaxFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonSoftmaxUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonSoftmaxUint8Workload.hpp + src/armnn/backends/NeonWorkloads/NeonSplitterFloat32Workload.cpp + src/armnn/backends/NeonWorkloads/NeonSplitterFloat32Workload.hpp + src/armnn/backends/NeonWorkloads/NeonSplitterUint8Workload.cpp + src/armnn/backends/NeonWorkloads/NeonSplitterUint8Workload.hpp + src/armnn/backends/NeonWorkloadUtils.cpp + src/armnn/backends/NeonWorkloadUtils.hpp + src/armnn/backends/NeonTensorHandle.hpp) +endif() +if(ARMCOMPUTECL) + # Additionally include source files for ARM Compute OpenCL backend + list(APPEND armnn_sources + src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClActivationUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClActivationUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClAdditionFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClAdditionFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClBaseConstantWorkload.cpp + src/armnn/backends/ClWorkloads/ClBaseConstantWorkload.hpp + src/armnn/backends/ClWorkloads/ClBaseMergerWorkload.hpp + src/armnn/backends/ClWorkloads/ClBatchNormalizationFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClBatchNormalizationFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClConstantFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClConstantFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClConstantUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClConstantUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClConvolution2dFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClConvolution2dFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClConvolution2dUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionHelper.hpp + src/armnn/backends/ClWorkloads/ClFloorFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClFloorFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClL2NormalizationFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClL2NormalizationFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClMergerFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClMergerFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClMergerUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClMergerUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClMultiplicationFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClMultiplicationFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClNormalizationFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClNormalizationFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClPermuteWorkload.cpp + src/armnn/backends/ClWorkloads/ClPermuteWorkload.hpp + src/armnn/backends/ClWorkloads/ClPooling2dBaseWorkload.cpp + src/armnn/backends/ClWorkloads/ClPooling2dBaseWorkload.hpp + src/armnn/backends/ClWorkloads/ClPooling2dFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClPooling2dFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClPooling2dUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClPooling2dUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClReshapeFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClReshapeFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClReshapeUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClReshapeUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClResizeBilinearFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClResizeBilinearFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClSoftmaxFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClSoftmaxFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp + src/armnn/backends/ClWorkloads/ClSplitterFloat32Workload.cpp + src/armnn/backends/ClWorkloads/ClSplitterFloat32Workload.hpp + src/armnn/backends/ClWorkloads/ClSplitterUint8Workload.cpp + src/armnn/backends/ClWorkloads/ClSplitterUint8Workload.hpp + src/armnn/backends/ClWorkloadUtils.hpp + src/armnn/backends/ClTensorHandle.hpp) +endif() +# Files shared by all ARM Compute backends +if(ARMCOMPUTENEON OR ARMCOMPUTECL) + list(APPEND armnn_sources + src/armnn/backends/ArmComputeTensorUtils.hpp + src/armnn/backends/ArmComputeTensorUtils.cpp + src/armnn/backends/ArmComputeUtils.hpp) +endif() + +# Files used for Streamline-based profiling backend +if(PROFILING_BACKEND_STREAMLINE) + list(APPEND armnn_sources + ${GATOR_ROOT}/annotate/streamline_annotate.h + ${GATOR_ROOT}/annotate/streamline_annotate.c) +endif() + +add_library_ex(armnn SHARED ${armnn_sources}) +target_include_directories(armnn PRIVATE src/armnn) +target_include_directories(armnn PRIVATE src/armnnUtils) +target_link_libraries(armnn armnnUtils) +target_link_libraries(armnn ${CMAKE_DL_LIBS}) + +install(TARGETS armnn DESTINATION ${CMAKE_INSTALL_PREFIX}/lib) +if(BUILD_CAFFE_PARSER) + install(TARGETS armnnCaffeParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib) +endif() +install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_PREFIX}/include) + +target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY}) + +if(ARMCOMPUTENEON OR ARMCOMPUTECL) + target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES}) +endif() +if(ARMCOMPUTECL) + target_link_libraries(armnn ${OPENCL_LIBRARIES}) +endif() + +if(PROFILING_BACKEND_STREAMLINE) + target_link_libraries(armnn pthread) +endif() + + +if(BUILD_UNIT_TESTS) + set(unittest_sources) + list(APPEND unittest_sources + src/armnn/test/UnitTests.cpp + src/armnn/test/UnitTests.hpp + src/armnn/test/EndToEndTest.cpp + src/armnn/test/UtilsTests.cpp + src/armnn/test/GraphTests.cpp + src/armnn/test/RuntimeTests.cpp + src/armnn/test/CreateWorkload.hpp + src/armnn/test/TensorTest.cpp + src/armnn/test/TensorHelpers.hpp + src/armnn/test/Network_test.cpp + src/armnn/test/GraphUtils.hpp + src/armnn/backends/test/IsLayerSupportedTest.cpp + src/armnn/backends/test/IsLayerSupportedTestImpl.hpp + src/armnn/backends/test/Reference.cpp + src/armnn/backends/test/WorkloadDataValidation.cpp + src/armnn/backends/test/TensorCopyUtils.hpp + src/armnn/backends/test/TensorCopyUtils.cpp + src/armnn/backends/test/LayerTests.hpp + src/armnn/backends/test/LayerTests.cpp + src/armnn/backends/test/Conv2dTestImpl.hpp + src/armnn/backends/test/ActivationTestImpl.hpp + src/armnn/backends/test/ActivationFixture.hpp + src/armnn/backends/test/Pooling2dTestImpl.hpp + src/armnn/backends/test/ReshapeTestImpl.hpp + src/armnn/backends/test/PermuteTestImpl.hpp + src/armnn/backends/test/FullyConnectedTestImpl.hpp + src/armnn/backends/test/SplitterTestImpl.hpp + src/armnn/backends/test/NormTestImpl.hpp + src/armnn/backends/test/BatchNormTestImpl.hpp + src/armnn/backends/test/WorkloadTestUtils.hpp + src/armnn/backends/test/CreateWorkloadRef.cpp + src/armnn/backends/test/QuantizeHelper.hpp) + + if(ARMCOMPUTENEON) + list(APPEND unittest_sources + src/armnn/backends/test/ArmComputeNeon.cpp + src/armnn/backends/test/CreateWorkloadNeon.cpp + src/armnn/test/CreateWorkloadClNeon.hpp) + endif() + + if(ARMCOMPUTECL) + list(APPEND unittest_sources + src/armnn/backends/test/ArmComputeCl.cpp + src/armnn/backends/test/CreateWorkloadCl.cpp + src/armnn/test/CreateWorkloadClNeon.hpp) + endif() + + if(ARMCOMPUTENEON OR ARMCOMPUTECL) + list(APPEND unittest_sources + src/armnn/backends/test/MemCopyTests.cpp) + endif() + + if(BUILD_CAFFE_PARSER) + list(APPEND unittest_sources + src/armnnCaffeParser/test/TestAdd.cpp + src/armnnCaffeParser/test/TestConcat.cpp + src/armnnCaffeParser/test/TestDropout.cpp + src/armnnCaffeParser/test/TestInputs.cpp + src/armnnCaffeParser/test/TestMul.cpp + src/armnnCaffeParser/test/TestPooling2d.cpp + src/armnnCaffeParser/test/TestInPlace.cpp + src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp + src/armnnCaffeParser/test/TestSplit.cpp + ) + endif() + + add_executable_ex(UnitTests ${unittest_sources}) + target_include_directories(UnitTests PRIVATE src/armnn) + target_include_directories(UnitTests PRIVATE src/armnnUtils) + + CHECK_INCLUDE_FILE(valgrind/memcheck.h VALGRIND_FOUND) + + if(VALGRIND_FOUND) + target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1") + endif() + + target_link_libraries(UnitTests armnn) + target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT}) + target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY}) + + if(BUILD_CAFFE_PARSER) + target_link_libraries(UnitTests armnnCaffeParser) + endif() + + addDllCopyCommands(UnitTests) +endif() + |