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/TensorUtils.hpp
  52     src/armnnUtils/TensorUtils.cpp
  53     )
  54 if(BUILD_TF_PARSER OR BUILD_CAFFE_PARSER)
  55     list(APPEND armnnUtils_sources
  56     src/armnnUtils/ParserPrototxtFixture.hpp
  57     )
  58 endif()
  59
  60 add_library_ex(armnnUtils STATIC ${armnnUtils_sources})
  61
  62 if(BUILD_CAFFE_PARSER)
  63     # ArmNN Parser source files required for all build options
  64     set(armnn_caffe_parser_sources)
  65     list(APPEND armnn_caffe_parser_sources
  66         include/armnnCaffeParser/ICaffeParser.hpp
  67         src/armnnCaffeParser/RecordByRecordCaffeParser.hpp
  68         src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
  69         src/armnnCaffeParser/CaffeParser.hpp
  70         src/armnnCaffeParser/CaffeParser.cpp
  71         ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc
  72         )
  73     # The generated Caffe protobuf .cc file is not warning clean and we can't fix them.
  74     if(COMPILER_IS_GNU_LIKE)
  75         set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
  76     endif()
  77
  78     add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})
  79     set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")
  80
  81     target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)
  82
  83     target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
  84
  85     target_link_libraries(armnnCaffeParser armnn)
  86     target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})
  87
  88 endif()
  89
  90 if(BUILD_ONNX_PARSER)
  91     set(armnn_onnx_parser_sources)
  92     list(APPEND armnn_onnx_parser_sources
  93         include/armnnOnnxParser/IOnnxParser.hpp
  94         src/armnnOnnxParser/OnnxParser.hpp
  95         src/armnnOnnxParser/OnnxParser.cpp
  96         ${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc
  97         )
  98     # The generated onnx protobuf .cc files are not warning clean and we can't fix them.
  99     if(COMPILER_IS_GNU_LIKE)
 100         set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
 101     endif()
 102
 103     add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})
 104
 105     target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)
 106
 107     target_link_libraries(armnnOnnxParser armnn)
 108
 109     # Protobuf
 110     target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})
 111 endif()
 112
 113 if(BUILD_TF_PARSER)
 114     set(armnn_tf_parser_sources)
 115     list(APPEND armnn_tf_parser_sources
 116         include/armnnTfParser/ITfParser.hpp
 117         src/armnnTfParser/TfParser.hpp
 118         src/armnnTfParser/TfParser.cpp
 119         ${TF_PROTOBUFS}
 120         )
 121     # The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.
 122     if(COMPILER_IS_GNU_LIKE)
 123         set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
 124     endif()
 125
 126     add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})
 127
 128     target_include_directories(armnnTfParser PRIVATE src/armnnUtils)
 129
 130     target_link_libraries(armnnTfParser armnn)
 131
 132     # Protobuf (use the specific version tensorflow wants)
 133     target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})
 134 endif()
 135
 136 if(BUILD_TF_LITE_PARSER)
 137     set(armnn_tf_lite_parser_sources)
 138     list(APPEND armnn_tf_lite_parser_sources
 139         include/armnnTfLiteParser/ITfLiteParser.hpp
 140         src/armnnTfLiteParser/TfLiteParser.hpp
 141         src/armnnTfLiteParser/TfLiteParser.cpp
 142         )
 143
 144     add_library_ex(armnnTfLiteParser SHARED ${armnn_tf_lite_parser_sources})
 145
 146     target_include_directories(armnnTfLiteParser PRIVATE src/armnn)
 147     target_include_directories(armnnTfLiteParser PRIVATE src/armnnUtils)
 148
 149     target_link_libraries(armnnTfLiteParser ${Boost_FILESYSTEM_LIBRARY} ${Boost_THREAD_LIBRARY})
 150     target_link_libraries(armnnTfLiteParser armnn ${FLATBUFFERS_LIBRARY})
 151 endif()
 152
 153 list(APPEND armnn_sources
 154     include/armnn/ArmNN.hpp
 155     include/armnn/BackendId.hpp
 156     include/armnn/Descriptors.hpp
 157     include/armnn/DescriptorsFwd.hpp
 158     include/armnn/Exceptions.hpp
 159     include/armnn/ILayerSupport.hpp
 160     include/armnn/INetwork.hpp
 161     include/armnn/IProfiler.hpp
 162     include/armnn/IRuntime.hpp
 163     include/armnn/LayerSupport.hpp
 164     include/armnn/LstmParams.hpp
 165     include/armnn/NetworkFwd.hpp
 166     include/armnn/Optional.hpp
 167     include/armnn/Tensor.hpp
 168     include/armnn/TensorFwd.hpp
 169     include/armnn/Types.hpp
 170     include/armnn/TypesUtils.hpp
 171     include/armnn/Utils.hpp
 172     include/armnn/Version.hpp
 173     src/armnn/layers/LayerCloneBase.hpp
 174     src/armnn/layers/LayerWithParameters.hpp
 175     src/armnn/layers/ActivationLayer.hpp
 176     src/armnn/layers/ActivationLayer.cpp
 177     src/armnn/layers/AdditionLayer.hpp
 178     src/armnn/layers/AdditionLayer.cpp
 179     src/armnn/layers/ArithmeticBaseLayer.hpp
 180     src/armnn/layers/ArithmeticBaseLayer.cpp
 181     src/armnn/layers/BatchNormalizationLayer.hpp
 182     src/armnn/layers/BatchNormalizationLayer.cpp
 183     src/armnn/layers/BatchToSpaceNdLayer.hpp
 184     src/armnn/layers/BatchToSpaceNdLayer.cpp
 185     src/armnn/layers/ConstantLayer.hpp
 186     src/armnn/layers/ConstantLayer.cpp
 187     src/armnn/layers/Convolution2dLayer.hpp
 188     src/armnn/layers/Convolution2dLayer.cpp
 189     src/armnn/layers/ConvertFp16ToFp32Layer.hpp
 190     src/armnn/layers/ConvertFp16ToFp32Layer.cpp
 191     src/armnn/layers/ConvertFp32ToFp16Layer.hpp
 192     src/armnn/layers/ConvertFp32ToFp16Layer.cpp
 193     src/armnn/layers/DebugLayer.hpp
 194     src/armnn/layers/DebugLayer.cpp
 195     src/armnn/layers/DepthwiseConvolution2dLayer.hpp
 196     src/armnn/layers/DepthwiseConvolution2dLayer.cpp
 197     src/armnn/layers/FakeQuantizationLayer.hpp
 198     src/armnn/layers/FakeQuantizationLayer.cpp
 199     src/armnn/layers/FloorLayer.hpp
 200     src/armnn/layers/FloorLayer.cpp
 201     src/armnn/layers/FullyConnectedLayer.hpp
 202     src/armnn/layers/FullyConnectedLayer.cpp
 203     src/armnn/layers/InputLayer.hpp
 204     src/armnn/layers/InputLayer.cpp
 205     src/armnn/layers/L2NormalizationLayer.hpp
 206     src/armnn/layers/L2NormalizationLayer.cpp
 207     src/armnn/layers/LstmLayer.cpp
 208     src/armnn/layers/LstmLayer.hpp
 209     src/armnn/layers/MaximumLayer.cpp
 210     src/armnn/layers/MaximumLayer.hpp
 211     src/armnn/layers/MeanLayer.hpp
 212     src/armnn/layers/MeanLayer.cpp
 213     src/armnn/layers/MemCopyLayer.hpp
 214     src/armnn/layers/MemCopyLayer.cpp
 215     src/armnn/layers/MergerLayer.hpp
 216     src/armnn/layers/MergerLayer.cpp
 217     src/armnn/layers/MinimumLayer.cpp
 218     src/armnn/layers/MinimumLayer.hpp
 219     src/armnn/layers/MultiplicationLayer.hpp
 220     src/armnn/layers/MultiplicationLayer.cpp
 221     src/armnn/layers/NormalizationLayer.hpp
 222     src/armnn/layers/NormalizationLayer.cpp
 223     src/armnn/layers/OutputLayer.hpp
 224     src/armnn/layers/OutputLayer.cpp
 225     src/armnn/layers/PadLayer.hpp
 226     src/armnn/layers/PadLayer.cpp
 227     src/armnn/layers/PermuteLayer.hpp
 228     src/armnn/layers/PermuteLayer.cpp
 229     src/armnn/layers/Pooling2dLayer.hpp
 230     src/armnn/layers/Pooling2dLayer.cpp
 231     src/armnn/layers/DivisionLayer.cpp
 232     src/armnn/layers/DivisionLayer.hpp
 233     src/armnn/layers/ReshapeLayer.hpp
 234     src/armnn/layers/ReshapeLayer.cpp
 235     src/armnn/layers/SpaceToBatchNdLayer.hpp
 236     src/armnn/layers/SpaceToBatchNdLayer.cpp
 237     src/armnn/layers/ResizeBilinearLayer.hpp
 238     src/armnn/layers/ResizeBilinearLayer.cpp
 239     src/armnn/layers/SoftmaxLayer.hpp
 240     src/armnn/layers/SoftmaxLayer.cpp
 241     src/armnn/layers/SplitterLayer.hpp
 242     src/armnn/layers/SplitterLayer.cpp
 243     src/armnn/layers/StridedSliceLayer.cpp
 244     src/armnn/layers/StridedSliceLayer.hpp
 245     src/armnn/layers/SubtractionLayer.cpp
 246     src/armnn/layers/SubtractionLayer.hpp
 247     src/armnn/Descriptors.cpp
 248     src/armnn/DeviceSpec.hpp
 249     src/armnn/Exceptions.cpp
 250     src/armnn/Graph.cpp
 251     src/armnn/Graph.hpp
 252     src/armnn/IGraphObservable.hpp
 253     src/armnn/Instrument.hpp
 254     src/armnn/InternalTypes.cpp
 255     src/armnn/InternalTypes.hpp
 256     src/armnn/JsonPrinter.cpp
 257     src/armnn/JsonPrinter.hpp
 258     src/armnn/Layer.cpp
 259     src/armnn/LayerFwd.hpp
 260     src/armnn/Layer.hpp
 261     src/armnn/LayersFwd.hpp
 262     src/armnn/LayerSupportCommon.hpp
 263     src/armnn/LayerSupport.cpp
 264     src/armnn/LoadedNetwork.cpp
 265     src/armnn/LoadedNetwork.hpp
 266     src/armnn/Network.cpp
 267     src/armnn/Network.hpp
 268     src/armnn/NetworkUtils.cpp
 269     src/armnn/NetworkUtils.hpp
 270     src/armnn/Observable.cpp
 271     src/armnn/Observable.hpp
 272     src/armnn/Optimizer.cpp
 273     src/armnn/Optimizer.hpp
 274     src/armnn/Profiling.cpp
 275     src/armnn/ProfilingEvent.cpp
 276     src/armnn/ProfilingEvent.hpp
 277     src/armnn/Profiling.hpp
 278     src/armnn/Runtime.cpp
 279     src/armnn/Runtime.hpp
 280     src/armnn/SerializeLayerParameters.cpp
 281     src/armnn/SerializeLayerParameters.hpp
 282     src/armnn/SubGraph.cpp
 283     src/armnn/SubGraph.hpp
 284     src/armnn/SubGraphSelector.cpp
 285     src/armnn/SubGraphSelector.hpp
 286     src/armnn/Tensor.cpp
 287     src/armnn/TypeUtils.hpp
 288     src/armnn/Utils.cpp
 289     src/armnn/WallClockTimer.cpp
 290     src/armnn/WallClockTimer.hpp
 291     src/armnn/optimizations/All.hpp
 292     src/armnn/optimizations/ConvertConstants.hpp
 293     src/armnn/optimizations/MovePermuteUp.hpp
 294     src/armnn/optimizations/Optimization.hpp
 295     src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
 296     src/armnn/optimizations/OptimizeInversePermutes.hpp
 297     src/armnn/optimizations/PermuteAsReshape.hpp
 298     src/armnn/optimizations/SquashEqualSiblings.hpp
 299     src/armnn/optimizations/OptimizeInverseConversions.hpp
 300     src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
 301     src/armnn/optimizations/AddDebug.hpp
 302     third-party/half/half.hpp
 303     )
 304
 305 # Files used for Streamline-based profiling backend
 306 if(PROFILING_BACKEND_STREAMLINE)
 307     list(APPEND armnn_sources
 308         ${GATOR_ROOT}/annotate/streamline_annotate.h
 309         ${GATOR_ROOT}/annotate/streamline_annotate.c)
 310 endif()
 311
 312 # the backends under src/backends extend the list of
 313 # object libs armnn to include in the build
 314 include(src/backends/backends.cmake)
 315 foreach(lib ${armnnLibraries})
 316     message("Adding object library dependency to armnn: ${lib}")
 317     list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
 318 endforeach()
 319
 320 add_library_ex(armnn SHARED ${armnn_sources})
 321
 322 target_include_directories(armnn PRIVATE src/armnn)
 323 target_include_directories(armnn PRIVATE src/armnnUtils)
 324 target_include_directories(armnn PRIVATE src/backends)
 325
 326 target_link_libraries(armnn armnnUtils)
 327
 328 target_link_libraries(armnn ${CMAKE_DL_LIBS})
 329
 330 install(TARGETS armnn DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 331 if(BUILD_CAFFE_PARSER)
 332     install(TARGETS armnnCaffeParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 333 endif()
 334 if(BUILD_ONNX_PARSER)
 335     install(TARGETS armnnOnnxParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 336 endif()
 337 if(BUILD_TF_PARSER)
 338     install(TARGETS armnnTfParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 339 endif()
 340 if(BUILD_TF_LITE_PARSER)
 341     install(TARGETS armnnTfLiteParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
 342 endif()
 343 install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_PREFIX}/include)
 344
 345 target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
 346
 347 if(ARMCOMPUTENEON OR ARMCOMPUTECL)
 348     target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
 349 endif()
 350 if(ARMCOMPUTECL)
 351     target_link_libraries(armnn ${OPENCL_LIBRARIES})
 352 endif()
 353
 354 if(PROFILING_BACKEND_STREAMLINE)
 355     target_link_libraries(armnn pthread)
 356 endif()
 357
 358 if(BUILD_UNIT_TESTS)
 359     set(unittest_sources)
 360     list(APPEND unittest_sources
 361         src/armnn/test/CreateWorkload.hpp
 362         src/armnn/test/CsvReaderTest.cpp
 363         src/armnn/test/EndToEndTest.cpp
 364         src/armnn/test/FloatingPointConverterTest.cpp
 365         src/armnn/test/GraphTests.cpp
 366         src/armnn/test/GraphUtils.hpp
 367         src/armnn/test/InstrumentTests.cpp
 368         src/armnn/test/LayerValidateOutputTest.cpp
 369         src/armnn/test/NetworkTests.cpp
 370         src/armnn/test/ObservableTest.cpp
 371         src/armnn/test/OptimizerTests.cpp
 372         src/armnn/test/OptionalTest.cpp
 373         src/armnn/test/ProfilerTests.cpp
 374         src/armnn/test/ProfilingEventTest.cpp
 375         src/armnn/test/RuntimeTests.cpp
 376         src/armnn/test/RuntimeTests.hpp
 377         src/armnn/test/SubGraphTests.cpp
 378         src/armnn/test/TensorHelpers.hpp
 379         src/armnn/test/TensorTest.cpp
 380         src/armnn/test/UnitTests.cpp
 381         src/armnn/test/UnitTests.hpp
 382         src/armnn/test/UtilsTests.cpp)
 383
 384     if(BUILD_TF_PARSER)
 385         list(APPEND unittest_sources
 386             src/armnnTfParser/test/Activations.cpp
 387             src/armnnTfParser/test/Addition.cpp
 388             src/armnnTfParser/test/BiasAdd.cpp
 389             src/armnnTfParser/test/BroadcastForAdd.cpp
 390             src/armnnTfParser/test/Convolution2d.cpp
 391             src/armnnTfParser/test/Concat.cpp
 392             src/armnnTfParser/test/ConcatOfConcats.cpp
 393             src/armnnTfParser/test/DepthwiseConvolution2d.cpp
 394             src/armnnTfParser/test/ExpandDims.cpp
 395             src/armnnTfParser/test/FusedBatchNorm.cpp
 396             src/armnnTfParser/test/Identity.cpp
 397             src/armnnTfParser/test/LocalResponseNormalization.cpp
 398             src/armnnTfParser/test/Maximum.cpp
 399             src/armnnTfParser/test/MaximumForLeakyRelu.cpp
 400             src/armnnTfParser/test/Minimum.cpp
 401             src/armnnTfParser/test/Multiplication.cpp
 402             src/armnnTfParser/test/Pad.cpp
 403             src/armnnTfParser/test/MultiOutput.cpp
 404             src/armnnTfParser/test/PassThru.cpp
 405             src/armnnTfParser/test/Pooling.cpp
 406             src/armnnTfParser/test/RealDiv.cpp
 407             src/armnnTfParser/test/Reshape.cpp
 408             src/armnnTfParser/test/ResizeBilinear.cpp
 409             src/armnnTfParser/test/Shape.cpp
 410             src/armnnTfParser/test/Softmax.cpp
 411             src/armnnTfParser/test/TestDependencies.cpp
 412             src/armnnTfParser/test/FullyConnected.cpp
 413             src/armnnTfParser/test/Constant.cpp
 414             src/armnnTfParser/test/TestMultiInputsOutputs.cpp
 415             src/armnnTfParser/test/Squeeze.cpp
 416             src/armnnTfParser/test/Sub.cpp)
 417     endif()
 418
 419     if(BUILD_TF_LITE_PARSER)
 420         list(APPEND unittest_sources
 421              src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp
 422              src/armnnTfLiteParser/test/Activations.cpp
 423              src/armnnTfLiteParser/test/AvgPool2D.cpp
 424              src/armnnTfLiteParser/test/Concatenation.cpp
 425              src/armnnTfLiteParser/test/Conv2D.cpp
 426              src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
 427              src/armnnTfLiteParser/test/FullyConnected.cpp
 428              src/armnnTfLiteParser/test/MaxPool2D.cpp
 429              src/armnnTfLiteParser/test/Reshape.cpp
 430              src/armnnTfLiteParser/test/Softmax.cpp
 431              src/armnnTfLiteParser/test/Squeeze.cpp
 432              src/armnnTfLiteParser/test/LoadModel.cpp
 433              src/armnnTfLiteParser/test/GetBuffer.cpp
 434              src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
 435              src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
 436              src/armnnTfLiteParser/test/GetTensorIds.cpp
 437              src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
 438              src/armnnTfLiteParser/test/GetInputsOutputs.cpp
 439              )
 440     endif()
 441
 442     if(BUILD_CAFFE_PARSER)
 443         list(APPEND unittest_sources
 444             src/armnnCaffeParser/test/TestAdd.cpp
 445             src/armnnCaffeParser/test/TestConcat.cpp
 446             src/armnnCaffeParser/test/TestConvolution.cpp
 447             src/armnnCaffeParser/test/TestDropout.cpp
 448             src/armnnCaffeParser/test/TestInputs.cpp
 449             src/armnnCaffeParser/test/TestMul.cpp
 450             src/armnnCaffeParser/test/TestPooling2d.cpp
 451             src/armnnCaffeParser/test/TestInPlace.cpp
 452             src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
 453             src/armnnCaffeParser/test/TestSplit.cpp
 454             )
 455     endif()
 456
 457     if(BUILD_ONNX_PARSER)
 458         list(APPEND unittest_sources
 459             src/armnnOnnxParser/test/Constructor.cpp
 460             src/armnnOnnxParser/test/CreateNetwork.cpp
 461             src/armnnOnnxParser/test/ProtoxtFixture.cpp
 462             src/armnnOnnxParser/test/Const.cpp
 463             src/armnnOnnxParser/test/Pooling.cpp
 464             src/armnnOnnxParser/test/Reshape.cpp
 465             src/armnnOnnxParser/test/Relu.cpp
 466             src/armnnOnnxParser/test/Conv2D.cpp
 467             src/armnnOnnxParser/test/Addition.cpp
 468             src/armnnOnnxParser/test/FullyConnected.cpp
 469             src/armnnOnnxParser/test/GetInputsOutputs.cpp
 470             src/armnnOnnxParser/test/BatchNorm.cpp
 471             src/armnnOnnxParser/test/DepthConv.cpp
 472             )
 473     endif()
 474
 475     foreach(lib ${armnnUnitTestLibraries})
 476         message("Adding object library dependency to UnitTests: ${lib}")
 477         list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
 478     endforeach()
 479
 480     add_executable(UnitTests ${unittest_sources})
 481     target_include_directories(UnitTests PRIVATE src/armnn)
 482     target_include_directories(UnitTests PRIVATE src/armnnUtils)
 483     target_include_directories(UnitTests PRIVATE src/backends)
 484
 485     if(VALGRIND_FOUND)
 486         if(HEAP_PROFILING OR LEAK_CHECKING)
 487             message("Valgrind is disabled for heap profiling and leak checking builds.")
 488         else()
 489             # Valgrind works with gperftools version number <= 2.4
 490             target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
 491         endif()
 492     endif()
 493
 494     target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
 495     target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
 496
 497     target_link_libraries(UnitTests armnn)
 498     target_link_libraries(UnitTests armnnUtils)
 499
 500     if(BUILD_TF_PARSER)
 501         target_link_libraries(UnitTests armnnTfParser)
 502     endif()
 503
 504     if(BUILD_CAFFE_PARSER)
 505         target_link_libraries(UnitTests armnnCaffeParser)
 506     endif()
 507
 508     if(BUILD_TF_LITE_PARSER)
 509         target_link_libraries(UnitTests armnnTfLiteParser)
 510     endif()
 511
 512     if(BUILD_ONNX_PARSER)
 513         target_link_libraries(UnitTests armnnOnnxParser)
 514     endif()
 515
 516     addDllCopyCommands(UnitTests)
 517 endif()