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