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 (BUILD_TESTS)
  19     add_subdirectory(tests)
  20 endif()
  21
  22 add_subdirectory(samples)
  23
  24 # Include the additional cmake files in their own target so that they will appear nicely in IDEs
  25 add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files})
  26
  27 set(armnnUtils_sources)
  28 list(APPEND armnnUtils_sources
  29     src/armnnUtils/GraphTopologicalSort.hpp
  30     src/armnnUtils/Half.hpp
  31     src/armnnUtils/Logging.hpp
  32     src/armnnUtils/Logging.cpp
  33     src/armnnUtils/Permute.hpp
  34     src/armnnUtils/Permute.cpp
  35     src/armnnUtils/DataLayoutIndexed.cpp
  36     src/armnnUtils/DataLayoutIndexed.hpp
  37     src/armnnUtils/DotSerializer.cpp
  38     src/armnnUtils/DotSerializer.hpp
  39     src/armnnUtils/HeapProfiling.cpp
  40     src/armnnUtils/HeapProfiling.hpp
  41     src/armnnUtils/LeakChecking.cpp
  42     src/armnnUtils/LeakChecking.hpp
  43     src/armnnUtils/CsvReader.cpp
  44     src/armnnUtils/CsvReader.hpp
  45     src/armnnUtils/FloatingPointConverter.cpp
  46     src/armnnUtils/FloatingPointConverter.hpp
  47     src/armnnUtils/VerificationHelpers.hpp
  48     src/armnnUtils/VerificationHelpers.cpp
  49     src/armnnUtils/ParserHelper.hpp
  50     src/armnnUtils/ParserHelper.cpp
  51     src/armnnUtils/ParserPrototxtFixture.hpp
  52     src/armnnUtils/PrototxtConversions.hpp
  53     src/armnnUtils/PrototxtConversions.cpp
  54     src/armnnUtils/TensorUtils.hpp
  55     src/armnnUtils/TensorUtils.cpp
  56     )
  57
  58 add_library_ex(armnnUtils STATIC ${armnnUtils_sources})
  59
  60 if(BUILD_CAFFE_PARSER)
  61     # ArmNN Parser source files required for all build options
  62     set(armnn_caffe_parser_sources)
  63     list(APPEND armnn_caffe_parser_sources
  64         include/armnnCaffeParser/ICaffeParser.hpp
  65         src/armnnCaffeParser/RecordByRecordCaffeParser.hpp
  66         src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
  67         src/armnnCaffeParser/CaffeParser.hpp
  68         src/armnnCaffeParser/CaffeParser.cpp
  69         ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc
  70         )
  71     # The generated Caffe protobuf .cc file is not warning clean and we can't fix them.
  72     if(COMPILER_IS_GNU_LIKE)
  73         set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
  74     endif()
  75
  76     add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})
  77     set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")
  78
  79     target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)
  80
  81     target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
  82
  83     target_link_libraries(armnnCaffeParser armnn)
  84     target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})
  85
  86 endif()
  87
  88 if(BUILD_ONNX_PARSER)
  89     set(armnn_onnx_parser_sources)
  90     list(APPEND armnn_onnx_parser_sources
  91         include/armnnOnnxParser/IOnnxParser.hpp
  92         src/armnnOnnxParser/OnnxParser.hpp
  93         src/armnnOnnxParser/OnnxParser.cpp
  94         ${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc
  95         )
  96     # The generated onnx protobuf .cc files are not warning clean and we can't fix them.
  97     if(COMPILER_IS_GNU_LIKE)
  98         set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
  99     endif()
 100
 101     add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})
 102
 103     target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)
 104
 105     target_link_libraries(armnnOnnxParser armnn)
 106
 107     # Protobuf
 108     target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})
 109 endif()
 110
 111 if(BUILD_TF_PARSER)
 112     set(armnn_tf_parser_sources)
 113     list(APPEND armnn_tf_parser_sources
 114         include/armnnTfParser/ITfParser.hpp
 115         src/armnnTfParser/TfParser.hpp
 116         src/armnnTfParser/TfParser.cpp
 117         ${TF_PROTOBUFS}
 118         )
 119     # The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.
 120     if(COMPILER_IS_GNU_LIKE)
 121         set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
 122     endif()
 123
 124     add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})
 125
 126     target_include_directories(armnnTfParser PRIVATE src/armnnUtils)
 127
 128     target_link_libraries(armnnTfParser armnn)
 129
 130     # Protobuf (use the specific version tensorflow wants)
 131     target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})
 132 endif()
 133
 134 if(BUILD_TF_LITE_PARSER)
 135     set(armnn_tf_lite_parser_sources)
 136     list(APPEND armnn_tf_lite_parser_sources
 137         include/armnnTfLiteParser/ITfLiteParser.hpp
 138         src/armnnTfLiteParser/TfLiteParser.hpp
 139         src/armnnTfLiteParser/TfLiteParser.cpp
 140         )
 141
 142     add_library_ex(armnnTfLiteParser SHARED ${armnn_tf_lite_parser_sources})
 143
 144     target_include_directories(armnnTfLiteParser PRIVATE src/armnn)
 145     target_include_directories(armnnTfLiteParser PRIVATE src/armnnUtils)
 146     target_include_directories(armnnTfLiteParser SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
 147
 148     target_link_libraries(armnnTfLiteParser ${Boost_FILESYSTEM_LIBRARY} ${Boost_THREAD_LIBRARY})
 149     target_link_libraries(armnnTfLiteParser armnn ${FLATBUFFERS_LIBRARY})
 150 endif()
 151
 152 if(BUILD_ARMNN_SERIALIZER)
 153     add_custom_command (
 154         OUTPUT  src/armnnSerializer/Schema_generated.h
 155         COMMAND ${FLATC_DIR}/flatc -o ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer --cpp
 156                 ${CMAKE_CURRENT_SOURCE_DIR}/src/armnnSerializer/Schema.fbs
 157     )
 158
 159     set(armnn_serializer_sources)
 160     list(APPEND armnn_serializer_sources
 161         include/armnnSerializer/ISerializer.hpp
 162         include/armnnDeserializer/IDeserializer.hpp
 163         src/armnnSerializer/Schema_generated.h
 164         src/armnnSerializer/Serializer.hpp
 165         src/armnnSerializer/Serializer.cpp
 166         src/armnnSerializer/SerializerUtils.hpp
 167         src/armnnSerializer/SerializerUtils.cpp
 168         src/armnnDeserializer/Deserializer.hpp
 169         src/armnnDeserializer/Deserializer.cpp
 170         )
 171
 172     add_library_ex(armnnSerializer SHARED ${armnn_serializer_sources})
 173
 174     target_include_directories(armnnSerializer PRIVATE src/armnn)
 175     target_include_directories(armnnSerializer PRIVATE src/armnnUtils)
 176
 177     # System include to suppress warnings for flatbuffers generated files
 178     target_include_directories(armnnSerializer SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer)
 179
 180     target_link_libraries(armnnSerializer armnn ${FLATBUFFERS_LIBRARY})
 181 endif()
 182
 183 if(BUILD_ARMNN_QUANTIZER)
 184
 185     if(NOT BUILD_ARMNN_SERIALIZER)
 186         message(ERROR, "In order to build the ArmNN Quantization Tool you must set BUILD_ARMNN_SERIALZER = YES")
 187     endif()
 188
 189     add_executable_ex(ArmnnQuantizer
 190         src/armnnQuantizer/CommandLineProcessor.hpp
 191         src/armnnQuantizer/CommandLineProcessor.cpp
 192         src/armnnQuantizer/ArmNNQuantizerMain.cpp)
 193
 194     target_include_directories(ArmnnQuantizer PRIVATE include/armnnDeserializeParser)
 195
 196     target_link_libraries(ArmnnQuantizer
 197             ${Boost_SYSTEM_LIBRARY}
 198             ${Boost_PROGRAM_OPTIONS_LIBRARY}
 199             ${Boost_FILESYSTEM_LIBRARY}
 200             ${Boost_LOG_LIBRARY}
 201             ${Boost_THREAD_LIBRARY} )
 202
 203     target_link_libraries(ArmnnQuantizer
 204             armnnSerializer
 205             armnn
 206             ${FLATBUFFERS_LIBRARY})
 207
 208     if(Threads_FOUND AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android)))
 209         target_link_libraries(ArmnnQuantizer pthread)
 210     endif()
 211
 212 endif()
 213
 214
 215 list(APPEND armnn_sources
 216     include/armnn/ArmNN.hpp
 217     include/armnn/BackendId.hpp
 218     include/armnn/Descriptors.hpp
 219     include/armnn/DescriptorsFwd.hpp
 220     include/armnn/Exceptions.hpp
 221     include/armnn/ILayerSupport.hpp
 222     include/armnn/ILayerVisitor.hpp
 223     include/armnn/INetwork.hpp
 224     include/armnn/INetworkQuantizer.hpp
 225     include/armnn/IProfiler.hpp
 226     include/armnn/IRuntime.hpp
 227     include/armnn/LayerSupport.hpp
 228     include/armnn/LayerVisitorBase.hpp
 229     include/armnn/LstmParams.hpp
 230     include/armnn/NetworkFwd.hpp
 231     include/armnn/Optional.hpp
 232     include/armnn/Tensor.hpp
 233     include/armnn/TensorFwd.hpp
 234     include/armnn/Types.hpp
 235     include/armnn/TypesUtils.hpp
 236     include/armnn/Utils.hpp
 237     include/armnn/Version.hpp
 238     src/armnn/layers/LayerCloneBase.hpp
 239     src/armnn/layers/LayerWithParameters.hpp
 240     src/armnn/layers/ActivationLayer.hpp
 241     src/armnn/layers/ActivationLayer.cpp
 242     src/armnn/layers/AdditionLayer.hpp
 243     src/armnn/layers/AdditionLayer.cpp
 244     src/armnn/layers/BatchNormalizationLayer.hpp
 245     src/armnn/layers/BatchNormalizationLayer.cpp
 246     src/armnn/layers/BatchToSpaceNdLayer.hpp
 247     src/armnn/layers/BatchToSpaceNdLayer.cpp
 248     src/armnn/layers/ConstantLayer.hpp
 249     src/armnn/layers/ConstantLayer.cpp
 250     src/armnn/layers/Convolution2dLayer.hpp
 251     src/armnn/layers/Convolution2dLayer.cpp
 252     src/armnn/layers/ConvertFp16ToFp32Layer.hpp
 253     src/armnn/layers/ConvertFp16ToFp32Layer.cpp
 254     src/armnn/layers/ConvertFp32ToFp16Layer.hpp
 255     src/armnn/layers/ConvertFp32ToFp16Layer.cpp
 256     src/armnn/layers/DebugLayer.hpp
 257     src/armnn/layers/DebugLayer.cpp
 258     src/armnn/layers/DepthwiseConvolution2dLayer.hpp
 259     src/armnn/layers/DepthwiseConvolution2dLayer.cpp
 260     src/armnn/layers/DetectionPostProcessLayer.hpp
 261     src/armnn/layers/DetectionPostProcessLayer.cpp
 262     src/armnn/layers/ElementwiseBaseLayer.hpp
 263     src/armnn/layers/ElementwiseBaseLayer.cpp
 264     src/armnn/layers/EqualLayer.hpp
 265     src/armnn/layers/EqualLayer.cpp
 266     src/armnn/layers/FakeQuantizationLayer.hpp
 267     src/armnn/layers/FakeQuantizationLayer.cpp
 268     src/armnn/layers/FloorLayer.hpp
 269     src/armnn/layers/FloorLayer.cpp
 270     src/armnn/layers/FullyConnectedLayer.hpp
 271     src/armnn/layers/FullyConnectedLayer.cpp
 272     src/armnn/layers/GatherLayer.cpp
 273     src/armnn/layers/GatherLayer.hpp
 274     src/armnn/layers/GreaterLayer.cpp
 275     src/armnn/layers/GreaterLayer.hpp
 276     src/armnn/layers/InputLayer.hpp
 277     src/armnn/layers/InputLayer.cpp
 278     src/armnn/layers/L2NormalizationLayer.hpp
 279     src/armnn/layers/L2NormalizationLayer.cpp
 280     src/armnn/layers/LstmLayer.cpp
 281     src/armnn/layers/LstmLayer.hpp
 282     src/armnn/layers/MaximumLayer.cpp
 283     src/armnn/layers/MaximumLayer.hpp
 284     src/armnn/layers/MeanLayer.hpp
 285     src/armnn/layers/MeanLayer.cpp
 286     src/armnn/layers/MemCopyLayer.hpp
 287     src/armnn/layers/MemCopyLayer.cpp
 288     src/armnn/layers/MergerLayer.hpp
 289     src/armnn/layers/MergerLayer.cpp
 290     src/armnn/layers/MinimumLayer.cpp
 291     src/armnn/layers/MinimumLayer.hpp
 292     src/armnn/layers/MultiplicationLayer.hpp
 293     src/armnn/layers/MultiplicationLayer.cpp
 294     src/armnn/layers/NormalizationLayer.hpp
 295     src/armnn/layers/NormalizationLayer.cpp
 296     src/armnn/layers/OutputLayer.hpp
 297     src/armnn/layers/OutputLayer.cpp
 298     src/armnn/layers/PadLayer.hpp
 299     src/armnn/layers/PadLayer.cpp
 300     src/armnn/layers/PermuteLayer.hpp
 301     src/armnn/layers/PermuteLayer.cpp
 302     src/armnn/layers/Pooling2dLayer.hpp
 303     src/armnn/layers/Pooling2dLayer.cpp
 304     src/armnn/layers/DivisionLayer.cpp
 305     src/armnn/layers/DivisionLayer.hpp
 306     src/armnn/layers/PreCompiledLayer.hpp
 307     src/armnn/layers/PreCompiledLayer.cpp
 308     src/armnn/layers/ReshapeLayer.hpp
 309     src/armnn/layers/ReshapeLayer.cpp
 310     src/armnn/layers/SpaceToBatchNdLayer.hpp
 311     src/armnn/layers/SpaceToBatchNdLayer.cpp
 312     src/armnn/layers/ResizeBilinearLayer.hpp
 313     src/armnn/layers/ResizeBilinearLayer.cpp
 314     src/armnn/layers/RsqrtLayer.cpp
 315     src/armnn/layers/RsqrtLayer.hpp
 316     src/armnn/layers/SoftmaxLayer.hpp
 317     src/armnn/layers/SoftmaxLayer.cpp
 318     src/armnn/layers/SplitterLayer.hpp
 319     src/armnn/layers/SplitterLayer.cpp
 320     src/armnn/layers/StridedSliceLayer.cpp
 321     src/armnn/layers/StridedSliceLayer.hpp
 322     src/armnn/layers/SubtractionLayer.cpp
 323     src/armnn/layers/SubtractionLayer.hpp
 324     src/armnn/BackendSettings.hpp
 325     src/armnn/CompatibleTypes.hpp
 326     src/armnn/Descriptors.cpp
 327     src/armnn/DeviceSpec.hpp
 328     src/armnn/Exceptions.cpp
 329     src/armnn/Graph.cpp
 330     src/armnn/Graph.hpp
 331     src/armnn/IGraphObservable.hpp
 332     src/armnn/Instrument.hpp
 333     src/armnn/InternalTypes.cpp
 334     src/armnn/InternalTypes.hpp
 335     src/armnn/ISubGraphConverter.hpp
 336     src/armnn/JsonPrinter.cpp
 337     src/armnn/JsonPrinter.hpp
 338     src/armnn/Layer.cpp
 339     src/armnn/LayerFwd.hpp
 340     src/armnn/Layer.hpp
 341     src/armnn/LayersFwd.hpp
 342     src/armnn/LayerSupportCommon.hpp
 343     src/armnn/LayerSupport.cpp
 344     src/armnn/LoadedNetwork.cpp
 345     src/armnn/LoadedNetwork.hpp
 346     src/armnn/Network.cpp
 347     src/armnn/Network.hpp
 348     src/armnn/NetworkQuantizer.cpp
 349     src/armnn/NetworkQuantizer.hpp
 350     src/armnn/NetworkQuantizerUtils.cpp
 351     src/armnn/NetworkQuantizerUtils.hpp
 352     src/armnn/NetworkUtils.cpp
 353     src/armnn/NetworkUtils.hpp
 354     src/armnn/Observable.cpp
 355     src/armnn/Observable.hpp
 356     src/armnn/Optimizer.cpp
 357     src/armnn/Optimizer.hpp
 358     src/armnn/OverrideInputRangeVisitor.cpp
 359     src/armnn/OverrideInputRangeVisitor.hpp
 360     src/armnn/Profiling.cpp
 361     src/armnn/ProfilingEvent.cpp
 362     src/armnn/ProfilingEvent.hpp
 363     src/armnn/Profiling.hpp
 364     src/armnn/QuantizerVisitor.cpp
 365     src/armnn/QuantizerVisitor.hpp
 366     src/armnn/Runtime.cpp
 367     src/armnn/Runtime.hpp
 368     src/armnn/RangeTracker.cpp
 369     src/armnn/RangeTracker.hpp
 370     src/armnn/SerializeLayerParameters.cpp
 371     src/armnn/SerializeLayerParameters.hpp
 372     src/armnn/StaticRangeVisitor.cpp
 373     src/armnn/StaticRangeVisitor.hpp
 374     src/armnn/SubGraph.cpp
 375     src/armnn/SubGraph.hpp
 376     src/armnn/SubGraphSelector.cpp
 377     src/armnn/SubGraphSelector.hpp
 378     src/armnn/Tensor.cpp
 379     src/armnn/TypeUtils.hpp
 380     src/armnn/Utils.cpp
 381     src/armnn/WallClockTimer.cpp
 382     src/armnn/WallClockTimer.hpp
 383     src/armnn/optimizations/All.hpp
 384     src/armnn/optimizations/ConvertConstants.hpp
 385     src/armnn/optimizations/MovePermuteUp.hpp
 386     src/armnn/optimizations/Optimization.hpp
 387     src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
 388     src/armnn/optimizations/OptimizeInversePermutes.hpp
 389     src/armnn/optimizations/PermuteAsReshape.hpp
 390     src/armnn/optimizations/SquashEqualSiblings.hpp
 391     src/armnn/optimizations/OptimizeInverseConversions.hpp
 392     src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
 393     src/armnn/optimizations/AddDebug.hpp
 394     third-party/half/half.hpp
 395     )
 396
 397 # Files used for Streamline-based profiling backend
 398 if(PROFILING_BACKEND_STREAMLINE)
 399     list(APPEND armnn_sources
 400         ${GATOR_ROOT}/annotate/streamline_annotate.h
 401         ${GATOR_ROOT}/annotate/streamline_annotate.c)
 402 endif()
 403
 404 # the backends under src/backends extend the list of
 405 # object libs armnn to include in the build
 406 include(src/backends/backends.cmake)
 407 foreach(lib ${armnnLibraries})
 408     message("Adding object library dependency to armnn: ${lib}")
 409     list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
 410 endforeach()
 411
 412 add_library_ex(armnn SHARED ${armnn_sources})
 413
 414 target_include_directories(armnn PRIVATE src/armnn)
 415 target_include_directories(armnn PRIVATE src/armnnUtils)
 416 target_include_directories(armnn PRIVATE src/backends)
 417
 418 target_link_libraries(armnn armnnUtils)
 419
 420 target_link_libraries(armnn ${CMAKE_DL_LIBS})
 421
 422 install(TARGETS armnn DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 423 if(BUILD_CAFFE_PARSER)
 424     install(TARGETS armnnCaffeParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 425 endif()
 426 if(BUILD_ONNX_PARSER)
 427     install(TARGETS armnnOnnxParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 428 endif()
 429 if(BUILD_TF_PARSER)
 430     install(TARGETS armnnTfParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 431 endif()
 432 if(BUILD_TF_LITE_PARSER)
 433     install(TARGETS armnnTfLiteParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 434 endif()
 435 if(BUILD_ARMNN_SERIALIZER)
 436     install(TARGETS armnnSerializer DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 437 endif()
 438 install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_PREFIX}/include)
 439
 440 target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
 441
 442 if(ARMCOMPUTENEON OR ARMCOMPUTECL)
 443     target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
 444 endif()
 445 if(ARMCOMPUTECL)
 446     target_link_libraries(armnn ${OPENCL_LIBRARIES})
 447 endif()
 448
 449 if(PROFILING_BACKEND_STREAMLINE)
 450     target_link_libraries(armnn pthread)
 451 endif()
 452
 453 if(BUILD_UNIT_TESTS)
 454     set(unittest_sources)
 455     list(APPEND unittest_sources
 456         src/armnn/test/ConstTensorLayerVisitor.hpp
 457         src/armnn/test/ConstTensorLayerVisitor.cpp
 458         src/armnn/test/CreateWorkload.hpp
 459         src/armnn/test/CsvReaderTest.cpp
 460         src/armnn/test/EndToEndTest.cpp
 461         src/armnn/test/FloatingPointConverterTest.cpp
 462         src/armnn/test/GraphTests.cpp
 463         src/armnn/test/GraphUtils.cpp
 464         src/armnn/test/GraphUtils.hpp
 465         src/armnn/test/InstrumentTests.cpp
 466         src/armnn/test/LayerValidateOutputTest.cpp
 467         src/armnn/test/NetworkTests.cpp
 468         src/armnn/test/ObservableTest.cpp
 469         src/armnn/test/OptimizerTests.cpp
 470         src/armnn/test/OptionalTest.cpp
 471         src/armnn/test/ProfilerTests.cpp
 472         src/armnn/test/ProfilingEventTest.cpp
 473         src/armnn/test/QuantizerTest.cpp
 474         src/armnn/test/RuntimeTests.cpp
 475         src/armnn/test/RuntimeTests.hpp
 476         src/armnn/test/SubGraphTests.cpp
 477         src/armnn/test/TensorHelpers.hpp
 478         src/armnn/test/TensorTest.cpp
 479         src/armnn/test/TestInputOutputLayerVisitor.cpp
 480         src/armnn/test/TestInputOutputLayerVisitor.hpp
 481         src/armnn/test/TestLayerVisitor.cpp
 482         src/armnn/test/TestLayerVisitor.hpp
 483         src/armnn/test/TestNameOnlyLayerVisitor.cpp
 484         src/armnn/test/TestNameOnlyLayerVisitor.hpp
 485         src/armnn/test/TestNameAndDescriptorLayerVisitor.hpp
 486         src/armnn/test/TestNameAndDescriptorLayerVisitor.cpp
 487         src/armnn/test/UnitTests.cpp
 488         src/armnn/test/UnitTests.hpp
 489         src/armnn/test/UtilsTests.cpp
 490         src/armnnUtils/test/PrototxtConversionsTest.cpp
 491         src/armnnUtils/test/ParserHelperTest.cpp
 492         )
 493
 494     if(BUILD_TF_PARSER)
 495         list(APPEND unittest_sources
 496             src/armnnTfParser/test/Activations.cpp
 497             src/armnnTfParser/test/Addition.cpp
 498             src/armnnTfParser/test/AddN.cpp
 499             src/armnnTfParser/test/Assert.cpp
 500             src/armnnTfParser/test/BiasAdd.cpp
 501             src/armnnTfParser/test/BroadcastForAdd.cpp
 502             src/armnnTfParser/test/Convolution2d.cpp
 503             src/armnnTfParser/test/Concat.cpp
 504             src/armnnTfParser/test/ConcatOfConcats.cpp
 505             src/armnnTfParser/test/DepthwiseConvolution2d.cpp
 506             src/armnnTfParser/test/Equal.cpp
 507             src/armnnTfParser/test/ExpandDims.cpp
 508             src/armnnTfParser/test/FusedBatchNorm.cpp
 509             src/armnnTfParser/test/Gather.cpp
 510             src/armnnTfParser/test/Greater.cpp
 511             src/armnnTfParser/test/Identity.cpp
 512             src/armnnTfParser/test/LocalResponseNormalization.cpp
 513             src/armnnTfParser/test/Maximum.cpp
 514             src/armnnTfParser/test/MaximumForLeakyRelu.cpp
 515             src/armnnTfParser/test/Mean.cpp
 516             src/armnnTfParser/test/Minimum.cpp
 517             src/armnnTfParser/test/Multiplication.cpp
 518             src/armnnTfParser/test/Pad.cpp
 519             src/armnnTfParser/test/MultiOutput.cpp
 520             src/armnnTfParser/test/PassThru.cpp
 521             src/armnnTfParser/test/Pooling.cpp
 522             src/armnnTfParser/test/RealDiv.cpp
 523             src/armnnTfParser/test/Reshape.cpp
 524             src/armnnTfParser/test/ResizeBilinear.cpp
 525             src/armnnTfParser/test/Rsqrt.cpp
 526             src/armnnTfParser/test/Shape.cpp
 527             src/armnnTfParser/test/Softmax.cpp
 528             src/armnnTfParser/test/TestDependencies.cpp
 529             src/armnnTfParser/test/FullyConnected.cpp
 530             src/armnnTfParser/test/Constant.cpp
 531             src/armnnTfParser/test/TestMultiInputsOutputs.cpp
 532             src/armnnTfParser/test/Split.cpp
 533             src/armnnTfParser/test/Squeeze.cpp
 534             src/armnnTfParser/test/Sub.cpp
 535             )
 536     endif()
 537
 538     if(BUILD_TF_LITE_PARSER)
 539         enable_language(ASM)
 540         list(APPEND unittest_sources
 541              src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp
 542              src/armnnTfLiteParser/test/Activations.cpp
 543              src/armnnTfLiteParser/test/Addition.cpp
 544              src/armnnTfLiteParser/test/AvgPool2D.cpp
 545              src/armnnTfLiteParser/test/Concatenation.cpp
 546              src/armnnTfLiteParser/test/Constant.cpp
 547              src/armnnTfLiteParser/test/Conv2D.cpp
 548              src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
 549              src/armnnTfLiteParser/test/FullyConnected.cpp
 550              src/armnnTfLiteParser/test/MaxPool2D.cpp
 551              src/armnnTfLiteParser/test/Mean.cpp
 552              src/armnnTfLiteParser/test/Multiplication.cpp
 553              src/armnnTfLiteParser/test/Pad.cpp
 554              src/armnnTfLiteParser/test/Reshape.cpp
 555              src/armnnTfLiteParser/test/Softmax.cpp
 556              src/armnnTfLiteParser/test/Squeeze.cpp
 557              src/armnnTfLiteParser/test/LoadModel.cpp
 558              src/armnnTfLiteParser/test/GetBuffer.cpp
 559              src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
 560              src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
 561              src/armnnTfLiteParser/test/GetTensorIds.cpp
 562              src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
 563              src/armnnTfLiteParser/test/GetInputsOutputs.cpp
 564              src/armnnTfLiteParser/test/Schema.s
 565              )
 566         set_source_files_properties(src/armnnTfLiteParser/test/Schema.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
 567     endif()
 568
 569     if(BUILD_CAFFE_PARSER)
 570         list(APPEND unittest_sources
 571             src/armnnCaffeParser/test/TestAdd.cpp
 572             src/armnnCaffeParser/test/TestConcat.cpp
 573             src/armnnCaffeParser/test/TestConvolution.cpp
 574             src/armnnCaffeParser/test/TestDropout.cpp
 575             src/armnnCaffeParser/test/TestInputs.cpp
 576             src/armnnCaffeParser/test/TestMul.cpp
 577             src/armnnCaffeParser/test/TestPooling2d.cpp
 578             src/armnnCaffeParser/test/TestInPlace.cpp
 579             src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
 580             src/armnnCaffeParser/test/TestSplit.cpp
 581             )
 582     endif()
 583
 584     if(BUILD_ARMNN_SERIALIZER)
 585         enable_language(ASM)
 586         list(APPEND unittest_sources
 587                 src/armnnSerializer/Schema_generated.h
 588                 src/armnnSerializer/test/ActivationSerializationTests.cpp
 589                 src/armnnSerializer/test/SerializerTests.cpp
 590                 src/armnnDeserializer/test/DeserializeActivation.cpp
 591                 src/armnnDeserializer/test/DeserializeAdd.cpp
 592                 src/armnnDeserializer/test/DeserializeConvolution2d.cpp
 593                 src/armnnDeserializer/test/DeserializeFullyConnected.cpp
 594                 src/armnnDeserializer/test/DeserializeMultiplication.cpp
 595                 src/armnnDeserializer/test/DeserializePermute.cpp
 596                 src/armnnDeserializer/test/DeserializePooling2d.cpp
 597                 src/armnnDeserializer/test/DeserializeReshape.cpp
 598                 src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp
 599                 src/armnnDeserializer/test/SchemaSerialize.s
 600                 )
 601         set_source_files_properties(src/armnnDeserializer/test/SchemaSerialize.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
 602     endif()
 603
 604     if(BUILD_ONNX_PARSER)
 605         list(APPEND unittest_sources
 606             src/armnnOnnxParser/test/Constructor.cpp
 607             src/armnnOnnxParser/test/CreateNetwork.cpp
 608             src/armnnOnnxParser/test/ProtoxtFixture.cpp
 609             src/armnnOnnxParser/test/Const.cpp
 610             src/armnnOnnxParser/test/Pooling.cpp
 611             src/armnnOnnxParser/test/Reshape.cpp
 612             src/armnnOnnxParser/test/Relu.cpp
 613             src/armnnOnnxParser/test/Conv2D.cpp
 614             src/armnnOnnxParser/test/Addition.cpp
 615             src/armnnOnnxParser/test/FullyConnected.cpp
 616             src/armnnOnnxParser/test/GetInputsOutputs.cpp
 617             src/armnnOnnxParser/test/BatchNorm.cpp
 618             src/armnnOnnxParser/test/DepthConv.cpp
 619             )
 620     endif()
 621
 622     foreach(lib ${armnnUnitTestLibraries})
 623         message("Adding object library dependency to UnitTests: ${lib}")
 624         list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
 625     endforeach()
 626
 627     add_executable(UnitTests ${unittest_sources})
 628     target_include_directories(UnitTests PRIVATE src/armnn)
 629     target_include_directories(UnitTests PRIVATE src/armnnUtils)
 630     target_include_directories(UnitTests PRIVATE src/backends)
 631
 632     if(BUILD_TF_LITE_PARSER)
 633         target_include_directories(UnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
 634     endif()
 635
 636     if(VALGRIND_FOUND)
 637         if(HEAP_PROFILING OR LEAK_CHECKING)
 638             message("Valgrind is disabled for heap profiling and leak checking builds.")
 639         else()
 640             # Valgrind works with gperftools version number <= 2.4
 641             target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
 642         endif()
 643     endif()
 644
 645     target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
 646     target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
 647
 648     target_link_libraries(UnitTests armnn)
 649     target_link_libraries(UnitTests armnnUtils)
 650
 651     if(BUILD_TF_PARSER)
 652         target_link_libraries(UnitTests armnnTfParser)
 653     endif()
 654
 655     if(BUILD_CAFFE_PARSER)
 656         target_link_libraries(UnitTests armnnCaffeParser)
 657     endif()
 658
 659     if(BUILD_ARMNN_SERIALIZER)
 660         target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer)
 661         target_link_libraries(UnitTests armnnSerializer)
 662     endif()
 663
 664     if(BUILD_TF_LITE_PARSER)
 665         target_link_libraries(UnitTests armnnTfLiteParser)
 666     endif()
 667
 668     if(BUILD_ONNX_PARSER)
 669         target_link_libraries(UnitTests armnnOnnxParser)
 670     endif()
 671
 672     addDllCopyCommands(UnitTests)
 673 endif()
 674
 675 if (BUILD_ARMNN_SERIALIZER AND BUILD_TF_PARSER)
 676     set(ArmnnConverter_sources
 677         src/armnnConverter/ArmnnConverter.cpp)
 678
 679     add_executable_ex(ArmnnConverter ${ArmnnConverter_sources})
 680     target_include_directories(ArmnnConverter PRIVATE src/armnn)
 681     target_include_directories(ArmnnConverter PRIVATE src/armnnUtils)
 682
 683     target_link_libraries(ArmnnConverter armnnTfParser)
 684     target_link_libraries(ArmnnConverter armnnSerializer)
 685
 686     target_link_libraries(ArmnnConverter armnn)
 687     target_link_libraries(ArmnnConverter ${CMAKE_THREAD_LIBS_INIT})
 688     target_link_libraries(ArmnnConverter
 689         ${Boost_SYSTEM_LIBRARY}
 690         ${Boost_FILESYSTEM_LIBRARY}
 691         ${Boost_PROGRAM_OPTIONS_LIBRARY})
 692     addDllCopyCommands(ArmnnConverter)
 693 endif()