CMakeLists.txt

   1 #
   2 # Copyright © 2017 Arm Ltd. All rights reserved.
   3 # SPDX-License-Identifier: MIT
   4 #
   5 cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake
   6 project(armnn)
   7
   8 set(additional_cmake_files)
   9 list(APPEND additional_cmake_files
  10     cmake/Utils.cmake
  11     cmake/GlobalConfig.cmake
  12     cmake/AddDllCopyCommands.cmake)
  13
  14 foreach(cmake_file ${additional_cmake_files})
  15     include(${cmake_file})
  16 endforeach()
  17
  18 if (DYNAMIC_BACKEND_PATHS)
  19     # It's expected to have the format: DYNAMIC_BACKEND_PATHS="PATH_1:PATH_2...:PATH_N"
  20     add_definitions('-DDYNAMIC_BACKEND_PATHS="${DYNAMIC_BACKEND_PATHS}"')
  21 endif()
  22
  23 include(GNUInstallDirs)
  24
  25 add_subdirectory(samples)
  26 add_subdirectory(src/armnnTfLiteParser)
  27 add_subdirectory(src/armnnSerializer)
  28
  29 if (BUILD_TESTS)
  30     add_subdirectory(tests)
  31 endif()
  32
  33 # Include the additional cmake files in their own target so that they will appear nicely in IDEs
  34 add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files})
  35
  36 set(armnnUtils_sources)
  37 list(APPEND armnnUtils_sources
  38     src/armnnUtils/GraphTopologicalSort.hpp
  39     src/armnnUtils/Half.hpp
  40     src/armnnUtils/Logging.hpp
  41     src/armnnUtils/Logging.cpp
  42     src/armnnUtils/Permute.hpp
  43     src/armnnUtils/Permute.cpp
  44     src/armnnUtils/DataLayoutIndexed.cpp
  45     src/armnnUtils/DataLayoutIndexed.hpp
  46     src/armnnUtils/DotSerializer.cpp
  47     src/armnnUtils/DotSerializer.hpp
  48     src/armnnUtils/HeapProfiling.cpp
  49     src/armnnUtils/HeapProfiling.hpp
  50     src/armnnUtils/LeakChecking.cpp
  51     src/armnnUtils/LeakChecking.hpp
  52     src/armnnUtils/ModelAccuracyChecker.cpp
  53     src/armnnUtils/ModelAccuracyChecker.hpp
  54     src/armnnUtils/CsvReader.cpp
  55     src/armnnUtils/CsvReader.hpp
  56     src/armnnUtils/FloatingPointConverter.cpp
  57     src/armnnUtils/FloatingPointConverter.hpp
  58     src/armnnUtils/VerificationHelpers.hpp
  59     src/armnnUtils/VerificationHelpers.cpp
  60     src/armnnUtils/ParserHelper.hpp
  61     src/armnnUtils/ParserHelper.cpp
  62     src/armnnUtils/ParserPrototxtFixture.hpp
  63     src/armnnUtils/PrototxtConversions.hpp
  64     src/armnnUtils/PrototxtConversions.cpp
  65     src/armnnUtils/TensorIOUtils.hpp
  66     src/armnnUtils/TensorUtils.hpp
  67     src/armnnUtils/TensorUtils.cpp
  68     )
  69
  70 add_library_ex(armnnUtils STATIC ${armnnUtils_sources})
  71 target_include_directories(armnnUtils PRIVATE src/backends)
  72
  73 if(BUILD_CAFFE_PARSER)
  74     # ArmNN Parser source files required for all build options
  75     set(armnn_caffe_parser_sources)
  76     list(APPEND armnn_caffe_parser_sources
  77         include/armnnCaffeParser/ICaffeParser.hpp
  78         src/armnnCaffeParser/RecordByRecordCaffeParser.hpp
  79         src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
  80         src/armnnCaffeParser/CaffeParser.hpp
  81         src/armnnCaffeParser/CaffeParser.cpp
  82         ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc
  83         )
  84     # The generated Caffe protobuf .cc file is not warning clean and we can't fix them.
  85     if(COMPILER_IS_GNU_LIKE)
  86         set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
  87     endif()
  88
  89     add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})
  90     set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")
  91
  92     target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)
  93
  94     target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
  95
  96     target_link_libraries(armnnCaffeParser armnn)
  97     target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})
  98
  99 endif()
 100
 101 if(BUILD_ONNX_PARSER)
 102     set(armnn_onnx_parser_sources)
 103     list(APPEND armnn_onnx_parser_sources
 104         include/armnnOnnxParser/IOnnxParser.hpp
 105         src/armnnOnnxParser/OnnxParser.hpp
 106         src/armnnOnnxParser/OnnxParser.cpp
 107         ${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc
 108         )
 109     # The generated onnx protobuf .cc files are not warning clean and we can't fix them.
 110     if(COMPILER_IS_GNU_LIKE)
 111         set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
 112     endif()
 113
 114     add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})
 115
 116     target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)
 117
 118     target_link_libraries(armnnOnnxParser armnn)
 119
 120     # Protobuf
 121     target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})
 122 endif()
 123
 124 if(BUILD_TF_PARSER)
 125     set(armnn_tf_parser_sources)
 126     list(APPEND armnn_tf_parser_sources
 127         include/armnnTfParser/ITfParser.hpp
 128         src/armnnTfParser/TfParser.hpp
 129         src/armnnTfParser/TfParser.cpp
 130         ${TF_PROTOBUFS}
 131         )
 132     # The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.
 133     if(COMPILER_IS_GNU_LIKE)
 134         set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
 135     endif()
 136
 137     add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})
 138
 139     target_include_directories(armnnTfParser PRIVATE src/armnnUtils)
 140
 141     target_link_libraries(armnnTfParser armnn)
 142
 143     # Protobuf (use the specific version tensorflow wants)
 144     target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})
 145 endif()
 146
 147 if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
 148     if(NOT BUILD_ARMNN_SERIALIZER)
 149         message(ERROR, "In order to build the ArmNN Quantization Tool you must set BUILD_ARMNN_SERIALIZER = YES")
 150     endif()
 151
 152     set(armnn_quantizer_sources)
 153     list(APPEND armnn_quantizer_sources
 154         src/armnnQuantizer/QuantizationDataSet.hpp
 155         src/armnnQuantizer/QuantizationDataSet.cpp
 156         src/armnnQuantizer/QuantizationInput.hpp
 157         src/armnnQuantizer/QuantizationInput.cpp
 158         src/armnnQuantizer/CommandLineProcessor.hpp
 159         src/armnnQuantizer/CommandLineProcessor.cpp
 160         )
 161
 162     add_library_ex(armnnQuantizer SHARED ${armnn_quantizer_sources})
 163
 164     target_include_directories(armnnQuantizer PRIVATE include/armnnDeserializer)
 165     target_include_directories(armnnQuantizer PRIVATE src/armnnUtils)
 166     target_include_directories(armnnQuantizer PRIVATE src/armnn)
 167
 168     include_directories(SYSTEM "${FLATBUFFERS_INCLUDE_PATH}")
 169     set_target_properties(armnnQuantizer PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR})
 170
 171     target_link_libraries(armnnQuantizer
 172             armnn
 173             ${Boost_SYSTEM_LIBRARY}
 174             ${Boost_PROGRAM_OPTIONS_LIBRARY}
 175             ${Boost_FILESYSTEM_LIBRARY}
 176             ${Boost_LOG_LIBRARY}
 177             ${Boost_THREAD_LIBRARY} )
 178
 179     add_executable_ex(ArmnnQuantizer
 180         src/armnnQuantizer/ArmNNQuantizerMain.cpp)
 181
 182     target_include_directories(ArmnnQuantizer PRIVATE include/armnnDeserializer)
 183     target_include_directories(ArmnnQuantizer PRIVATE src/armnn)
 184
 185     target_link_libraries(ArmnnQuantizer
 186             ${Boost_SYSTEM_LIBRARY}
 187             ${Boost_PROGRAM_OPTIONS_LIBRARY}
 188             ${Boost_FILESYSTEM_LIBRARY}
 189             ${Boost_LOG_LIBRARY}
 190             ${Boost_THREAD_LIBRARY} )
 191
 192     target_link_libraries(ArmnnQuantizer
 193             armnnQuantizer
 194             armnnSerializer
 195             armnn
 196             armnnUtils
 197             ${FLATBUFFERS_LIBRARY})
 198
 199     if(Threads_FOUND AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android)))
 200         target_link_libraries(ArmnnQuantizer pthread)
 201     endif()
 202
 203 endif()
 204
 205
 206 list(APPEND armnn_sources
 207     include/armnn/ArmNN.hpp
 208     include/armnn/BackendHelper.hpp
 209     include/armnn/BackendId.hpp
 210     include/armnn/Conversion.hpp
 211     include/armnn/Deprecated.hpp
 212     include/armnn/Descriptors.hpp
 213     include/armnn/DescriptorsFwd.hpp
 214     include/armnn/Exceptions.hpp
 215     include/armnn/ILayerSupport.hpp
 216     include/armnn/ILayerVisitor.hpp
 217     include/armnn/INetwork.hpp
 218     include/armnn/IProfiler.hpp
 219     include/armnn/IRuntime.hpp
 220     include/armnn/LayerSupport.hpp
 221     include/armnn/LayerVisitorBase.hpp
 222     include/armnn/LstmParams.hpp
 223     include/armnn/MemorySources.hpp
 224     include/armnn/NetworkFwd.hpp
 225     include/armnn/Optional.hpp
 226     include/armnn/QuantizedLstmParams.hpp
 227     include/armnn/Tensor.hpp
 228     include/armnn/TensorFwd.hpp
 229     include/armnn/Types.hpp
 230     include/armnn/TypesUtils.hpp
 231     include/armnn/Utils.hpp
 232     include/armnn/Version.hpp
 233     src/armnn/layers/LayerCloneBase.hpp
 234     src/armnn/layers/LayerWithParameters.hpp
 235     src/armnn/layers/AbsLayer.hpp
 236     src/armnn/layers/AbsLayer.cpp
 237     src/armnn/layers/ActivationLayer.hpp
 238     src/armnn/layers/ActivationLayer.cpp
 239     src/armnn/layers/AdditionLayer.hpp
 240     src/armnn/layers/AdditionLayer.cpp
 241     src/armnn/layers/ArgMinMaxLayer.hpp
 242     src/armnn/layers/ArgMinMaxLayer.cpp
 243     src/armnn/layers/BatchNormalizationLayer.hpp
 244     src/armnn/layers/BatchNormalizationLayer.cpp
 245     src/armnn/layers/BatchToSpaceNdLayer.hpp
 246     src/armnn/layers/BatchToSpaceNdLayer.cpp
 247     src/armnn/layers/ConcatLayer.hpp
 248     src/armnn/layers/ConcatLayer.cpp
 249     src/armnn/layers/ConstantLayer.hpp
 250     src/armnn/layers/ConstantLayer.cpp
 251     src/armnn/layers/Convolution2dLayer.hpp
 252     src/armnn/layers/Convolution2dLayer.cpp
 253     src/armnn/layers/ConvertFp16ToFp32Layer.hpp
 254     src/armnn/layers/ConvertFp16ToFp32Layer.cpp
 255     src/armnn/layers/ConvertFp32ToFp16Layer.hpp
 256     src/armnn/layers/ConvertFp32ToFp16Layer.cpp
 257     src/armnn/layers/DebugLayer.hpp
 258     src/armnn/layers/DebugLayer.cpp
 259     src/armnn/layers/DepthToSpaceLayer.hpp
 260     src/armnn/layers/DepthToSpaceLayer.cpp
 261     src/armnn/layers/DepthwiseConvolution2dLayer.hpp
 262     src/armnn/layers/DepthwiseConvolution2dLayer.cpp
 263     src/armnn/layers/DequantizeLayer.hpp
 264     src/armnn/layers/DequantizeLayer.cpp
 265     src/armnn/layers/DetectionPostProcessLayer.hpp
 266     src/armnn/layers/DetectionPostProcessLayer.cpp
 267     src/armnn/layers/ElementwiseBaseLayer.hpp
 268     src/armnn/layers/ElementwiseBaseLayer.cpp
 269     src/armnn/layers/EqualLayer.hpp
 270     src/armnn/layers/EqualLayer.cpp
 271     src/armnn/layers/FakeQuantizationLayer.hpp
 272     src/armnn/layers/FakeQuantizationLayer.cpp
 273     src/armnn/layers/FloorLayer.hpp
 274     src/armnn/layers/FloorLayer.cpp
 275     src/armnn/layers/FullyConnectedLayer.hpp
 276     src/armnn/layers/FullyConnectedLayer.cpp
 277     src/armnn/layers/GatherLayer.cpp
 278     src/armnn/layers/GatherLayer.hpp
 279     src/armnn/layers/GreaterLayer.cpp
 280     src/armnn/layers/GreaterLayer.hpp
 281     src/armnn/layers/InputLayer.hpp
 282     src/armnn/layers/InputLayer.cpp
 283     src/armnn/layers/L2NormalizationLayer.hpp
 284     src/armnn/layers/L2NormalizationLayer.cpp
 285     src/armnn/layers/LstmLayer.cpp
 286     src/armnn/layers/LstmLayer.hpp
 287     src/armnn/layers/MaximumLayer.cpp
 288     src/armnn/layers/MaximumLayer.hpp
 289     src/armnn/layers/MeanLayer.hpp
 290     src/armnn/layers/MeanLayer.cpp
 291     src/armnn/layers/MemCopyLayer.hpp
 292     src/armnn/layers/MemCopyLayer.cpp
 293     src/armnn/layers/MemImportLayer.hpp
 294     src/armnn/layers/MemImportLayer.cpp
 295     src/armnn/layers/MergeLayer.hpp
 296     src/armnn/layers/MergeLayer.cpp
 297     src/armnn/layers/MinimumLayer.cpp
 298     src/armnn/layers/MinimumLayer.hpp
 299     src/armnn/layers/MultiplicationLayer.hpp
 300     src/armnn/layers/MultiplicationLayer.cpp
 301     src/armnn/layers/NormalizationLayer.hpp
 302     src/armnn/layers/NormalizationLayer.cpp
 303     src/armnn/layers/OutputLayer.hpp
 304     src/armnn/layers/OutputLayer.cpp
 305     src/armnn/layers/PadLayer.hpp
 306     src/armnn/layers/PadLayer.cpp
 307     src/armnn/layers/PermuteLayer.hpp
 308     src/armnn/layers/PermuteLayer.cpp
 309     src/armnn/layers/Pooling2dLayer.hpp
 310     src/armnn/layers/Pooling2dLayer.cpp
 311     src/armnn/layers/QuantizeLayer.cpp
 312     src/armnn/layers/QuantizeLayer.hpp
 313     src/armnn/layers/QuantizedLstmLayer.hpp
 314     src/armnn/layers/QuantizedLstmLayer.cpp
 315     src/armnn/layers/DivisionLayer.cpp
 316     src/armnn/layers/DivisionLayer.hpp
 317     src/armnn/layers/PreCompiledLayer.hpp
 318     src/armnn/layers/PreCompiledLayer.cpp
 319     src/armnn/layers/PreluLayer.hpp
 320     src/armnn/layers/PreluLayer.cpp
 321     src/armnn/layers/ReshapeLayer.hpp
 322     src/armnn/layers/ReshapeLayer.cpp
 323     src/armnn/layers/ResizeLayer.hpp
 324     src/armnn/layers/ResizeLayer.cpp
 325     src/armnn/layers/RsqrtLayer.cpp
 326     src/armnn/layers/RsqrtLayer.hpp
 327     src/armnn/layers/SliceLayer.cpp
 328     src/armnn/layers/SliceLayer.hpp
 329     src/armnn/layers/SoftmaxLayer.hpp
 330     src/armnn/layers/SoftmaxLayer.cpp
 331     src/armnn/layers/SpaceToBatchNdLayer.hpp
 332     src/armnn/layers/SpaceToBatchNdLayer.cpp
 333     src/armnn/layers/SpaceToDepthLayer.hpp
 334     src/armnn/layers/SpaceToDepthLayer.cpp
 335     src/armnn/layers/SplitterLayer.hpp
 336     src/armnn/layers/SplitterLayer.cpp
 337     src/armnn/layers/StackLayer.hpp
 338     src/armnn/layers/StackLayer.cpp
 339     src/armnn/layers/StridedSliceLayer.cpp
 340     src/armnn/layers/StridedSliceLayer.hpp
 341     src/armnn/layers/SubtractionLayer.cpp
 342     src/armnn/layers/SubtractionLayer.hpp
 343     src/armnn/layers/SwitchLayer.cpp
 344     src/armnn/layers/SwitchLayer.hpp
 345     src/armnn/layers/TransposeConvolution2dLayer.cpp
 346     src/armnn/layers/TransposeConvolution2dLayer.hpp
 347     src/armnn/BackendSettings.hpp
 348     src/armnn/BackendHelper.cpp
 349     src/armnn/CompatibleTypes.hpp
 350     src/armnn/Descriptors.cpp
 351     src/armnn/DeviceSpec.hpp
 352     src/armnn/DynamicQuantizationVisitor.cpp
 353     src/armnn/DynamicQuantizationVisitor.hpp
 354     src/armnn/Exceptions.cpp
 355     src/armnn/ExecutionFrame.cpp
 356     src/armnn/ExecutionFrame.hpp
 357     src/armnn/Graph.cpp
 358     src/armnn/Graph.hpp
 359     src/armnn/IGraphObservable.hpp
 360     src/armnn/Instrument.hpp
 361     src/armnn/InternalTypes.cpp
 362     src/armnn/InternalTypes.hpp
 363     src/armnn/ISubgraphViewConverter.hpp
 364     src/armnn/JsonPrinter.cpp
 365     src/armnn/JsonPrinter.hpp
 366     src/armnn/Layer.cpp
 367     src/armnn/LayerFwd.hpp
 368     src/armnn/Layer.hpp
 369     src/armnn/LayersFwd.hpp
 370     src/armnn/LayerSupportCommon.hpp
 371     src/armnn/LayerSupport.cpp
 372     src/armnn/LoadedNetwork.cpp
 373     src/armnn/LoadedNetwork.hpp
 374     src/armnn/Network.cpp
 375     src/armnn/Network.hpp
 376     src/armnn/NetworkQuantizationScheme.hpp
 377     src/armnn/NetworkQuantizer.cpp
 378     src/armnn/NetworkQuantizer.hpp
 379     src/armnn/NetworkQuantizerUtils.cpp
 380     src/armnn/NetworkQuantizerUtils.hpp
 381     src/armnn/NetworkUtils.cpp
 382     src/armnn/NetworkUtils.hpp
 383     src/armnn/Observable.cpp
 384     src/armnn/Observable.hpp
 385     src/armnn/Optimizer.cpp
 386     src/armnn/Optimizer.hpp
 387     src/armnn/OverrideInputRangeVisitor.cpp
 388     src/armnn/OverrideInputRangeVisitor.hpp
 389     src/armnn/Profiling.cpp
 390     src/armnn/ProfilingEvent.cpp
 391     src/armnn/ProfilingEvent.hpp
 392     src/armnn/Profiling.hpp
 393     src/armnn/QuantizerVisitor.cpp
 394     src/armnn/QuantizerVisitor.hpp
 395     src/armnn/Runtime.cpp
 396     src/armnn/Runtime.hpp
 397     src/armnn/RangeTracker.cpp
 398     src/armnn/RangeTracker.hpp
 399     src/armnn/ResolveType.hpp
 400     src/armnn/SerializeLayerParameters.cpp
 401     src/armnn/SerializeLayerParameters.hpp
 402     src/armnn/StaticRangeVisitor.cpp
 403     src/armnn/StaticRangeVisitor.hpp
 404     src/armnn/SubgraphView.cpp
 405     src/armnn/SubgraphView.hpp
 406     src/armnn/SubgraphViewSelector.cpp
 407     src/armnn/SubgraphViewSelector.hpp
 408     src/armnn/Tensor.cpp
 409     src/armnn/TypesUtils.cpp
 410     src/armnn/Utils.cpp
 411     src/armnn/WallClockTimer.cpp
 412     src/armnn/WallClockTimer.hpp
 413     src/armnn/optimizations/AddDebug.hpp
 414     src/armnn/optimizations/All.hpp
 415     src/armnn/optimizations/ConvertConstants.hpp
 416     src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
 417     src/armnn/optimizations/FoldPadIntoConvolution2d.hpp
 418     src/armnn/optimizations/MovePermuteUp.hpp
 419     src/armnn/optimizations/Optimization.hpp
 420     src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
 421     src/armnn/optimizations/OptimizeInverseConversions.hpp
 422     src/armnn/optimizations/OptimizeInversePermutes.hpp
 423     src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp
 424     src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.cpp
 425     src/armnn/optimizations/PermuteAsReshape.hpp
 426     src/armnn/optimizations/SquashEqualSiblings.hpp
 427     src/profiling/CommandHandlerFunctor.cpp
 428     src/profiling/CommandHandlerFunctor.hpp
 429     src/profiling/CommandHandlerKey.cpp
 430     src/profiling/CommandHandlerKey.hpp
 431     src/profiling/CommandHandlerRegistry.cpp
 432     src/profiling/CommandHandlerRegistry.hpp
 433     src/profiling/CommandThread.cpp
 434     src/profiling/CommandThread.hpp
 435     src/profiling/ConnectionAcknowledgedCommandHandler.cpp
 436     src/profiling/ConnectionAcknowledgedCommandHandler.hpp
 437     src/profiling/CounterDirectory.cpp
 438     src/profiling/CounterDirectory.hpp
 439     src/profiling/CounterValues.hpp
 440     src/profiling/EncodeVersion.hpp
 441     src/profiling/Holder.cpp
 442     src/profiling/Holder.hpp
 443     src/profiling/IBufferManager.hpp
 444     src/profiling/ICounterDirectory.hpp
 445     src/profiling/ISendCounterPacket.hpp
 446     src/profiling/IPacketBuffer.hpp
 447     src/profiling/IPeriodicCounterCapture.hpp
 448     src/profiling/IProfilingConnection.hpp
 449     src/profiling/IReadCounterValue.hpp
 450     src/profiling/Packet.cpp
 451     src/profiling/Packet.hpp
 452     src/profiling/PacketVersionResolver.cpp
 453     src/profiling/PacketVersionResolver.hpp
 454     src/profiling/PeriodicCounterCapture.hpp
 455     src/profiling/PeriodicCounterCapture.cpp
 456     src/profiling/PeriodicCounterSelectionCommandHandler.cpp
 457     src/profiling/PeriodicCounterSelectionCommandHandler.hpp
 458     src/profiling/ProfilingConnectionDumpToFileDecorator.cpp
 459     src/profiling/ProfilingConnectionDumpToFileDecorator.hpp
 460     src/profiling/ProfilingConnectionFactory.cpp
 461     src/profiling/ProfilingConnectionFactory.hpp
 462     src/profiling/ProfilingService.cpp
 463     src/profiling/ProfilingService.hpp
 464     src/profiling/ProfilingStateMachine.cpp
 465     src/profiling/ProfilingStateMachine.hpp
 466     src/profiling/ProfilingUtils.cpp
 467     src/profiling/ProfilingUtils.hpp
 468     src/profiling/RequestCounterDirectoryCommandHandler.cpp
 469     src/profiling/RequestCounterDirectoryCommandHandler.hpp
 470     src/profiling/SendCounterPacket.cpp
 471     src/profiling/SendCounterPacket.hpp
 472     src/profiling/SocketProfilingConnection.cpp
 473     src/profiling/SocketProfilingConnection.hpp
 474     third-party/half/half.hpp
 475     )
 476
 477 # Files used for Streamline-based profiling backend
 478 if(PROFILING_BACKEND_STREAMLINE)
 479     list(APPEND armnn_sources
 480         ${GATOR_ROOT}/annotate/streamline_annotate.h
 481         ${GATOR_ROOT}/annotate/streamline_annotate.c)
 482 endif()
 483
 484 # the backends under src/backends extend the list of
 485 # object libs armnn to include in the build
 486 include(src/backends/backends.cmake)
 487 foreach(lib ${armnnLibraries})
 488     message("Adding object library dependency to armnn: ${lib}")
 489     list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
 490 endforeach()
 491
 492 add_library_ex(armnn SHARED ${armnn_sources})
 493
 494 target_include_directories(armnn PRIVATE src/armnn)
 495 target_include_directories(armnn PRIVATE src/armnnUtils)
 496 target_include_directories(armnn PRIVATE src/backends)
 497 target_include_directories(armnn PRIVATE src/profiling)
 498
 499 target_link_libraries(armnn armnnUtils)
 500
 501 target_link_libraries(armnn ${CMAKE_DL_LIBS})
 502
 503 install(TARGETS armnn
 504         LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
 505         RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
 506 if(BUILD_CAFFE_PARSER)
 507     install(TARGETS armnnCaffeParser
 508             LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
 509             RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
 510 endif()
 511 if(BUILD_ONNX_PARSER)
 512     install(TARGETS armnnOnnxParser
 513             LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
 514             RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
 515 endif()
 516 if(BUILD_TF_PARSER)
 517     install(TARGETS armnnTfParser
 518             LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
 519             RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
 520 endif()
 521
 522 install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
 523
 524 target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY}
 525                             ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
 526
 527 if(ARMCOMPUTENEON OR ARMCOMPUTECL)
 528     target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
 529 endif()
 530 if(ARMCOMPUTECL)
 531     target_link_libraries(armnn ${OPENCL_LIBRARIES})
 532 endif()
 533
 534 if(PROFILING_BACKEND_STREAMLINE)
 535     target_link_libraries(armnn pthread)
 536 endif()
 537
 538 if(BUILD_UNIT_TESTS)
 539     set(unittest_sources)
 540     list(APPEND unittest_sources
 541         src/armnn/test/ConstTensorLayerVisitor.hpp
 542         src/armnn/test/ConstTensorLayerVisitor.cpp
 543         src/armnn/test/CreateWorkload.hpp
 544         src/armnn/test/CsvReaderTest.cpp
 545         src/armnn/test/EndToEndTest.cpp
 546         src/armnn/test/ExecutionFrameTest.cpp
 547         src/armnn/test/FloatingPointConverterTest.cpp
 548         src/armnn/test/FlowControl.cpp
 549         src/armnn/test/GraphTests.cpp
 550         src/armnn/test/GraphUtils.cpp
 551         src/armnn/test/GraphUtils.hpp
 552         src/armnn/test/InstrumentTests.cpp
 553         src/armnn/test/InferOutputTests.cpp
 554         src/armnn/test/InferOutputTests.hpp
 555         src/armnn/test/ModelAccuracyCheckerTest.cpp
 556         src/armnn/test/NetworkTests.cpp
 557         src/armnn/test/ObservableTest.cpp
 558         src/armnn/test/OptimizerTests.cpp
 559         src/armnn/test/optimizations/ConvertConstantsFloatToHalfTests.cpp
 560         src/armnn/test/optimizations/ConvertConstantsHalfToFloatTests.cpp
 561         src/armnn/test/optimizations/Fp32NetworkToFp16ConverterTests.cpp
 562         src/armnn/test/optimizations/InsertDebugLayerTests.cpp
 563         src/armnn/test/optimizations/MovePermuteUpTests.cpp
 564         src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp
 565         src/armnn/test/optimizations/OptimizeInverseConversionsTests.cpp
 566         src/armnn/test/optimizations/OptimizeInversePermutesTests.cpp
 567         src/armnn/test/optimizations/PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp
 568         src/armnn/test/optimizations/PermuteAsReshapeTests.cpp
 569         src/armnn/test/optimizations/SquashEqualSiblingsTests.cpp
 570         src/armnn/test/OptionalTest.cpp
 571         src/armnn/test/ProfilerTests.cpp
 572         src/armnn/test/ProfilingEventTest.cpp
 573         src/armnn/test/SubgraphViewTests.cpp
 574         src/armnn/test/TensorHandleStrategyTest.cpp
 575         src/armnn/test/TensorHelpers.hpp
 576         src/armnn/test/TensorTest.cpp
 577         src/armnn/test/TestInputOutputLayerVisitor.cpp
 578         src/armnn/test/TestInputOutputLayerVisitor.hpp
 579         src/armnn/test/TestLayerVisitor.cpp
 580         src/armnn/test/TestLayerVisitor.hpp
 581         src/armnn/test/TestNameOnlyLayerVisitor.cpp
 582         src/armnn/test/TestNameOnlyLayerVisitor.hpp
 583         src/armnn/test/TestNameAndDescriptorLayerVisitor.hpp
 584         src/armnn/test/TestNameAndDescriptorLayerVisitor.cpp
 585         src/armnn/test/TestUtils.cpp
 586         src/armnn/test/TestUtils.hpp
 587         src/armnn/test/UnitTests.cpp
 588         src/armnn/test/UnitTests.hpp
 589         src/armnn/test/UtilsTests.cpp
 590         src/armnnUtils/test/PrototxtConversionsTest.cpp
 591         src/armnnUtils/test/ParserHelperTest.cpp
 592         src/armnnUtils/test/TensorUtilsTest.cpp
 593         src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp
 594         src/profiling/test/ProfilingTests.cpp
 595         src/profiling/test/SendCounterPacketTests.cpp
 596         )
 597
 598     if(ARMNNREF)
 599         list(APPEND unittest_sources
 600             src/armnn/test/DebugCallbackTest.cpp
 601             src/armnn/test/QuantizerTest.cpp
 602             src/armnn/test/RuntimeTests.cpp
 603             src/armnn/test/RuntimeTests.hpp
 604             )
 605     endif()
 606
 607     if(BUILD_TF_PARSER AND ARMNNREF)
 608         list(APPEND unittest_sources
 609             src/armnnTfParser/test/Activations.cpp
 610             src/armnnTfParser/test/Addition.cpp
 611             src/armnnTfParser/test/AddN.cpp
 612             src/armnnTfParser/test/Assert.cpp
 613             src/armnnTfParser/test/BiasAdd.cpp
 614             src/armnnTfParser/test/BroadcastForAdd.cpp
 615             src/armnnTfParser/test/Convolution2d.cpp
 616             src/armnnTfParser/test/Concat.cpp
 617             src/armnnTfParser/test/ConcatOfConcats.cpp
 618             src/armnnTfParser/test/DepthwiseConvolution2d.cpp
 619             src/armnnTfParser/test/Equal.cpp
 620             src/armnnTfParser/test/ExpandDims.cpp
 621             src/armnnTfParser/test/FusedBatchNorm.cpp
 622             src/armnnTfParser/test/Gather.cpp
 623             src/armnnTfParser/test/Greater.cpp
 624             src/armnnTfParser/test/Identity.cpp
 625             src/armnnTfParser/test/LocalResponseNormalization.cpp
 626             src/armnnTfParser/test/Maximum.cpp
 627             src/armnnTfParser/test/MaximumForLeakyRelu.cpp
 628             src/armnnTfParser/test/Mean.cpp
 629             src/armnnTfParser/test/Minimum.cpp
 630             src/armnnTfParser/test/Multiplication.cpp
 631             src/armnnTfParser/test/Pad.cpp
 632             src/armnnTfParser/test/MultiOutput.cpp
 633             src/armnnTfParser/test/PassThru.cpp
 634             src/armnnTfParser/test/Pooling.cpp
 635             src/armnnTfParser/test/RealDiv.cpp
 636             src/armnnTfParser/test/Reshape.cpp
 637             src/armnnTfParser/test/ResizeBilinear.cpp
 638             src/armnnTfParser/test/Rsqrt.cpp
 639             src/armnnTfParser/test/Shape.cpp
 640             src/armnnTfParser/test/Softmax.cpp
 641             src/armnnTfParser/test/TestDependencies.cpp
 642             src/armnnTfParser/test/FullyConnected.cpp
 643             src/armnnTfParser/test/Constant.cpp
 644             src/armnnTfParser/test/TestMultiInputsOutputs.cpp
 645             src/armnnTfParser/test/Split.cpp
 646             src/armnnTfParser/test/Squeeze.cpp
 647             src/armnnTfParser/test/Sub.cpp
 648             )
 649
 650     endif()
 651
 652     if(BUILD_TF_LITE_PARSER AND ARMNNREF)
 653         enable_language(ASM)
 654         list(APPEND unittest_sources
 655              src/armnnTfLiteParser/test/Activations.cpp
 656              src/armnnTfLiteParser/test/Addition.cpp
 657              src/armnnTfLiteParser/test/AvgPool2D.cpp
 658              src/armnnTfLiteParser/test/BatchToSpaceND.cpp
 659              src/armnnTfLiteParser/test/Concatenation.cpp
 660              src/armnnTfLiteParser/test/Constant.cpp
 661              src/armnnTfLiteParser/test/Conv2D.cpp
 662              src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
 663              src/armnnTfLiteParser/test/DetectionPostProcess.cpp
 664              src/armnnTfLiteParser/test/FullyConnected.cpp
 665              src/armnnTfLiteParser/test/L2Normalization.cpp
 666              src/armnnTfLiteParser/test/Maximum.cpp
 667              src/armnnTfLiteParser/test/MaxPool2D.cpp
 668              src/armnnTfLiteParser/test/Mean.cpp
 669              src/armnnTfLiteParser/test/Minimum.cpp
 670              src/armnnTfLiteParser/test/Multiplication.cpp
 671              src/armnnTfLiteParser/test/Pack.cpp
 672              src/armnnTfLiteParser/test/Pad.cpp
 673              src/armnnTfLiteParser/test/Reshape.cpp
 674              src/armnnTfLiteParser/test/ResizeBilinear.cpp
 675              src/armnnTfLiteParser/test/Softmax.cpp
 676              src/armnnTfLiteParser/test/SpaceToBatchND.cpp
 677              src/armnnTfLiteParser/test/Split.cpp
 678              src/armnnTfLiteParser/test/Squeeze.cpp
 679              src/armnnTfLiteParser/test/StridedSlice.cpp
 680              src/armnnTfLiteParser/test/Sub.cpp
 681              src/armnnTfLiteParser/test/TransposeConv.cpp
 682              src/armnnTfLiteParser/test/Transpose.cpp
 683              src/armnnTfLiteParser/test/Unpack.cpp
 684              src/armnnTfLiteParser/test/LoadModel.cpp
 685              src/armnnTfLiteParser/test/GetBuffer.cpp
 686              src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
 687              src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
 688              src/armnnTfLiteParser/test/GetTensorIds.cpp
 689              src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
 690              src/armnnTfLiteParser/test/GetInputsOutputs.cpp
 691              src/armnnTfLiteParser/test/Schema.s
 692              )
 693         set_source_files_properties(src/armnnTfLiteParser/test/Schema.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
 694
 695     endif()
 696
 697     if(BUILD_CAFFE_PARSER AND ARMNNREF)
 698         list(APPEND unittest_sources
 699             src/armnnCaffeParser/test/TestAdd.cpp
 700             src/armnnCaffeParser/test/TestConcat.cpp
 701             src/armnnCaffeParser/test/TestConvolution.cpp
 702             src/armnnCaffeParser/test/TestDropout.cpp
 703             src/armnnCaffeParser/test/TestInputs.cpp
 704             src/armnnCaffeParser/test/TestMul.cpp
 705             src/armnnCaffeParser/test/TestPooling2d.cpp
 706             src/armnnCaffeParser/test/TestInPlace.cpp
 707             src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
 708             src/armnnCaffeParser/test/TestSplit.cpp
 709             )
 710     endif()
 711
 712     if(BUILD_ONNX_PARSER AND ARMNNREF)
 713         list(APPEND unittest_sources
 714             src/armnnOnnxParser/test/Constructor.cpp
 715             src/armnnOnnxParser/test/CreateNetwork.cpp
 716             src/armnnOnnxParser/test/ProtoxtFixture.cpp
 717             src/armnnOnnxParser/test/Const.cpp
 718             src/armnnOnnxParser/test/Pooling.cpp
 719             src/armnnOnnxParser/test/Reshape.cpp
 720             src/armnnOnnxParser/test/Relu.cpp
 721             src/armnnOnnxParser/test/Conv2D.cpp
 722             src/armnnOnnxParser/test/Addition.cpp
 723             src/armnnOnnxParser/test/FullyConnected.cpp
 724             src/armnnOnnxParser/test/GetInputsOutputs.cpp
 725             src/armnnOnnxParser/test/BatchNorm.cpp
 726             src/armnnOnnxParser/test/DepthConv.cpp
 727             )
 728     endif()
 729
 730     if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
 731         list(APPEND unittest_sources
 732              src/armnnQuantizer/test/QuantizationDataSetTests.cpp
 733              )
 734     endif()
 735
 736     if(BUILD_ARMNN_SERIALIZER AND ARMNNREF)
 737         enable_language(ASM)
 738         list(APPEND unittest_sources
 739             src/armnnSerializer/test/ActivationSerializationTests.cpp
 740             src/armnnSerializer/test/SerializerTests.cpp
 741             src/armnnDeserializer/test/DeserializeAbs.cpp
 742             src/armnnDeserializer/test/DeserializeActivation.cpp
 743             src/armnnDeserializer/test/DeserializeAdd.cpp
 744             src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp
 745             src/armnnDeserializer/test/DeserializeBatchNormalization.cpp
 746             src/armnnDeserializer/test/DeserializeConstant.cpp
 747             src/armnnDeserializer/test/DeserializeConvolution2d.cpp
 748             src/armnnDeserializer/test/DeserializeDepthToSpace.cpp
 749             src/armnnDeserializer/test/DeserializeDivision.cpp
 750             src/armnnDeserializer/test/DeserializeEqual.cpp
 751             src/armnnDeserializer/test/DeserializeFloor.cpp
 752             src/armnnDeserializer/test/DeserializeFullyConnected.cpp
 753             src/armnnDeserializer/test/DeserializeGather.cpp
 754             src/armnnDeserializer/test/DeserializeGreater.cpp
 755             src/armnnDeserializer/test/DeserializeL2Normalization.cpp
 756             src/armnnDeserializer/test/DeserializeMean.cpp
 757             src/armnnDeserializer/test/DeserializeMultiplication.cpp
 758             src/armnnDeserializer/test/DeserializeNormalization.cpp
 759             src/armnnDeserializer/test/DeserializePad.cpp
 760             src/armnnDeserializer/test/DeserializePermute.cpp
 761             src/armnnDeserializer/test/DeserializePooling2d.cpp
 762             src/armnnDeserializer/test/DeserializeReshape.cpp
 763             src/armnnDeserializer/test/DeserializeResizeBilinear.cpp
 764             src/armnnDeserializer/test/DeserializeRsqrt.cpp
 765             src/armnnDeserializer/test/DeserializeSlice.cpp
 766             src/armnnDeserializer/test/DeserializeSpaceToBatchNd.cpp
 767             src/armnnDeserializer/test/DeserializeStridedSlice.cpp
 768             src/armnnDeserializer/test/DeserializeSubtraction.cpp
 769             src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp
 770             src/armnnDeserializer/test/SchemaSerialize.s
 771             )
 772         set_source_files_properties(src/armnnDeserializer/test/SchemaSerialize.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
 773     endif()
 774
 775     foreach(lib ${armnnUnitTestLibraries})
 776         message("Adding object library dependency to UnitTests: ${lib}")
 777         list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
 778     endforeach()
 779
 780     add_executable(UnitTests ${unittest_sources})
 781     target_include_directories(UnitTests PRIVATE src/armnn)
 782     target_include_directories(UnitTests PRIVATE src/armnnUtils)
 783     target_include_directories(UnitTests PRIVATE src/backends)
 784     target_include_directories(UnitTests PRIVATE src/profiling)
 785
 786     if(VALGRIND_FOUND)
 787         if(HEAP_PROFILING OR LEAK_CHECKING)
 788             message("Valgrind is disabled for heap profiling and leak checking builds.")
 789         else()
 790             # Valgrind works with gperftools version number <= 2.4
 791             target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
 792         endif()
 793     endif()
 794
 795     target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
 796     target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
 797
 798     target_link_libraries(UnitTests armnn)
 799     target_link_libraries(UnitTests armnnUtils)
 800
 801     if(BUILD_TF_PARSER)
 802         target_link_libraries(UnitTests armnnTfParser)
 803     endif()
 804
 805     if(BUILD_TF_LITE_PARSER)
 806         target_include_directories(UnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
 807         target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
 808         target_link_libraries(UnitTests armnnTfLiteParser)
 809     endif()
 810
 811     if(BUILD_CAFFE_PARSER)
 812         target_link_libraries(UnitTests armnnCaffeParser)
 813     endif()
 814
 815     if(BUILD_ARMNN_SERIALIZER AND ARMNNREF)
 816         target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer)
 817         target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
 818         target_link_libraries(UnitTests armnnSerializer)
 819     endif()
 820
 821     if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
 822         target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnQuantizer)
 823         target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
 824         target_link_libraries(UnitTests armnnQuantizer armnnSerializer)
 825     endif()
 826
 827     if(BUILD_ONNX_PARSER)
 828         target_link_libraries(UnitTests armnnOnnxParser)
 829     endif()
 830
 831     addDllCopyCommands(UnitTests)
 832 endif()
 833
 834 if (BUILD_ARMNN_SERIALIZER AND (BUILD_TF_PARSER OR BUILD_TF_LITE_PARSER OR BUILD_ONNX_PARSER OR BUILD_CAFFE_PARSER) AND ARMNNREF)
 835     set(ArmnnConverter_sources
 836         src/armnnConverter/ArmnnConverter.cpp)
 837
 838     add_executable_ex(ArmnnConverter ${ArmnnConverter_sources})
 839     target_include_directories(ArmnnConverter PRIVATE src/armnn)
 840     target_include_directories(ArmnnConverter PRIVATE src/armnnUtils)
 841
 842     if(BUILD_CAFFE_PARSER)
 843         target_link_libraries(ArmnnConverter armnnCaffeParser)
 844     endif()
 845
 846     if(BUILD_ONNX_PARSER)
 847         target_link_libraries(ArmnnConverter armnnOnnxParser)
 848     endif()
 849
 850     if(BUILD_TF_PARSER)
 851         target_link_libraries(ArmnnConverter armnnTfParser)
 852     endif()
 853
 854     if(BUILD_TF_LITE_PARSER)
 855         target_link_libraries(ArmnnConverter armnnTfLiteParser)
 856     endif()
 857
 858     target_link_libraries(ArmnnConverter armnnSerializer)
 859
 860     target_link_libraries(ArmnnConverter armnn)
 861     target_link_libraries(ArmnnConverter ${CMAKE_THREAD_LIBS_INIT})
 862     target_link_libraries(ArmnnConverter
 863         ${Boost_SYSTEM_LIBRARY}
 864         ${Boost_FILESYSTEM_LIBRARY}
 865         ${Boost_PROGRAM_OPTIONS_LIBRARY})
 866     addDllCopyCommands(ArmnnConverter)
 867 endif()
 868
 869 if(BUILD_GATORD_MOCK)
 870     set(gatord_mock_sources)
 871     list(APPEND gatord_mock_sources
 872         tests/profiling/GatordMockMain.cpp
 873         tests/profiling/CommandLineProcessor.hpp
 874         tests/profiling/CommandLineProcessor.cpp
 875         tests/profiling/GatordMockService.hpp
 876         tests/profiling/GatordMockService.cpp
 877         )
 878
 879     include_directories( ${Boost_INCLUDE_DIRS} )
 880
 881     add_executable_ex(GartordMock ${gatord_mock_sources})
 882
 883     target_link_libraries(GartordMock
 884         ${Boost_PROGRAM_OPTIONS_LIBRARY}
 885         )
 886
 887 endif()