IVGCVSW-3400 Fixed include directives for Gatord builds
[platform/upstream/armnn.git] / 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/InstanceNormalizationLayer.hpp
284     src/armnn/layers/InstanceNormalizationLayer.cpp
285     src/armnn/layers/L2NormalizationLayer.hpp
286     src/armnn/layers/L2NormalizationLayer.cpp
287     src/armnn/layers/LogSoftmaxLayer.hpp
288     src/armnn/layers/LogSoftmaxLayer.cpp
289     src/armnn/layers/LstmLayer.cpp
290     src/armnn/layers/LstmLayer.hpp
291     src/armnn/layers/MaximumLayer.cpp
292     src/armnn/layers/MaximumLayer.hpp
293     src/armnn/layers/MeanLayer.hpp
294     src/armnn/layers/MeanLayer.cpp
295     src/armnn/layers/MemCopyLayer.hpp
296     src/armnn/layers/MemCopyLayer.cpp
297     src/armnn/layers/MemImportLayer.hpp
298     src/armnn/layers/MemImportLayer.cpp
299     src/armnn/layers/MergeLayer.hpp
300     src/armnn/layers/MergeLayer.cpp
301     src/armnn/layers/MinimumLayer.cpp
302     src/armnn/layers/MinimumLayer.hpp
303     src/armnn/layers/MultiplicationLayer.hpp
304     src/armnn/layers/MultiplicationLayer.cpp
305     src/armnn/layers/NormalizationLayer.hpp
306     src/armnn/layers/NormalizationLayer.cpp
307     src/armnn/layers/OutputLayer.hpp
308     src/armnn/layers/OutputLayer.cpp
309     src/armnn/layers/PadLayer.hpp
310     src/armnn/layers/PadLayer.cpp
311     src/armnn/layers/PermuteLayer.hpp
312     src/armnn/layers/PermuteLayer.cpp
313     src/armnn/layers/Pooling2dLayer.hpp
314     src/armnn/layers/Pooling2dLayer.cpp
315     src/armnn/layers/QuantizeLayer.cpp
316     src/armnn/layers/QuantizeLayer.hpp
317     src/armnn/layers/QuantizedLstmLayer.hpp
318     src/armnn/layers/QuantizedLstmLayer.cpp
319     src/armnn/layers/DivisionLayer.cpp
320     src/armnn/layers/DivisionLayer.hpp
321     src/armnn/layers/PreCompiledLayer.hpp
322     src/armnn/layers/PreCompiledLayer.cpp
323     src/armnn/layers/PreluLayer.hpp
324     src/armnn/layers/PreluLayer.cpp
325     src/armnn/layers/ReshapeLayer.hpp
326     src/armnn/layers/ReshapeLayer.cpp
327     src/armnn/layers/ResizeLayer.hpp
328     src/armnn/layers/ResizeLayer.cpp
329     src/armnn/layers/RsqrtLayer.cpp
330     src/armnn/layers/RsqrtLayer.hpp
331     src/armnn/layers/SliceLayer.cpp
332     src/armnn/layers/SliceLayer.hpp
333     src/armnn/layers/SoftmaxLayer.hpp
334     src/armnn/layers/SoftmaxLayer.cpp
335     src/armnn/layers/SpaceToBatchNdLayer.hpp
336     src/armnn/layers/SpaceToBatchNdLayer.cpp
337     src/armnn/layers/SpaceToDepthLayer.hpp
338     src/armnn/layers/SpaceToDepthLayer.cpp
339     src/armnn/layers/SplitterLayer.hpp
340     src/armnn/layers/SplitterLayer.cpp
341     src/armnn/layers/StackLayer.hpp
342     src/armnn/layers/StackLayer.cpp
343     src/armnn/layers/StridedSliceLayer.cpp
344     src/armnn/layers/StridedSliceLayer.hpp
345     src/armnn/layers/SubtractionLayer.cpp
346     src/armnn/layers/SubtractionLayer.hpp
347     src/armnn/layers/SwitchLayer.cpp
348     src/armnn/layers/SwitchLayer.hpp
349     src/armnn/layers/TransposeConvolution2dLayer.cpp
350     src/armnn/layers/TransposeConvolution2dLayer.hpp
351     src/armnn/BackendSettings.hpp
352     src/armnn/BackendHelper.cpp
353     src/armnn/CompatibleTypes.hpp
354     src/armnn/Descriptors.cpp
355     src/armnn/DeviceSpec.hpp
356     src/armnn/DynamicQuantizationVisitor.cpp
357     src/armnn/DynamicQuantizationVisitor.hpp
358     src/armnn/Exceptions.cpp
359     src/armnn/ExecutionFrame.cpp
360     src/armnn/ExecutionFrame.hpp
361     src/armnn/Graph.cpp
362     src/armnn/Graph.hpp
363     src/armnn/IGraphObservable.hpp
364     src/armnn/Instrument.hpp
365     src/armnn/InternalTypes.cpp
366     src/armnn/InternalTypes.hpp
367     src/armnn/ISubgraphViewConverter.hpp
368     src/armnn/JsonPrinter.cpp
369     src/armnn/JsonPrinter.hpp
370     src/armnn/Layer.cpp
371     src/armnn/LayerFwd.hpp
372     src/armnn/Layer.hpp
373     src/armnn/LayersFwd.hpp
374     src/armnn/LayerSupportCommon.hpp
375     src/armnn/LayerSupport.cpp
376     src/armnn/LoadedNetwork.cpp
377     src/armnn/LoadedNetwork.hpp
378     src/armnn/Network.cpp
379     src/armnn/Network.hpp
380     src/armnn/NetworkQuantizationScheme.hpp
381     src/armnn/NetworkQuantizer.cpp
382     src/armnn/NetworkQuantizer.hpp
383     src/armnn/NetworkQuantizerUtils.cpp
384     src/armnn/NetworkQuantizerUtils.hpp
385     src/armnn/NetworkUtils.cpp
386     src/armnn/NetworkUtils.hpp
387     src/armnn/Observable.cpp
388     src/armnn/Observable.hpp
389     src/armnn/Optimizer.cpp
390     src/armnn/Optimizer.hpp
391     src/armnn/OverrideInputRangeVisitor.cpp
392     src/armnn/OverrideInputRangeVisitor.hpp
393     src/armnn/Profiling.cpp
394     src/armnn/ProfilingEvent.cpp
395     src/armnn/ProfilingEvent.hpp
396     src/armnn/Profiling.hpp
397     src/armnn/QuantizerVisitor.cpp
398     src/armnn/QuantizerVisitor.hpp
399     src/armnn/Runtime.cpp
400     src/armnn/Runtime.hpp
401     src/armnn/RangeTracker.cpp
402     src/armnn/RangeTracker.hpp
403     src/armnn/ResolveType.hpp
404     src/armnn/SerializeLayerParameters.cpp
405     src/armnn/SerializeLayerParameters.hpp
406     src/armnn/StaticRangeVisitor.cpp
407     src/armnn/StaticRangeVisitor.hpp
408     src/armnn/SubgraphView.cpp
409     src/armnn/SubgraphView.hpp
410     src/armnn/SubgraphViewSelector.cpp
411     src/armnn/SubgraphViewSelector.hpp
412     src/armnn/Tensor.cpp
413     src/armnn/TypesUtils.cpp
414     src/armnn/Utils.cpp
415     src/armnn/WallClockTimer.cpp
416     src/armnn/WallClockTimer.hpp
417     src/armnn/optimizations/AddDebug.hpp
418     src/armnn/optimizations/All.hpp
419     src/armnn/optimizations/ConvertConstants.hpp
420     src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
421     src/armnn/optimizations/FoldPadIntoConvolution2d.hpp
422     src/armnn/optimizations/MovePermuteUp.hpp
423     src/armnn/optimizations/Optimization.hpp
424     src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
425     src/armnn/optimizations/OptimizeInverseConversions.hpp
426     src/armnn/optimizations/OptimizeInversePermutes.hpp
427     src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp
428     src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.cpp
429     src/armnn/optimizations/PermuteAsReshape.hpp
430     src/armnn/optimizations/SquashEqualSiblings.hpp
431     src/profiling/BufferManager.cpp
432     src/profiling/BufferManager.hpp
433     src/profiling/CommandHandler.cpp
434     src/profiling/CommandHandler.hpp
435     src/profiling/CommandHandlerFunctor.cpp
436     src/profiling/CommandHandlerFunctor.hpp
437     src/profiling/CommandHandlerKey.cpp
438     src/profiling/CommandHandlerKey.hpp
439     src/profiling/CommandHandlerRegistry.cpp
440     src/profiling/CommandHandlerRegistry.hpp
441     src/profiling/ConnectionAcknowledgedCommandHandler.cpp
442     src/profiling/ConnectionAcknowledgedCommandHandler.hpp
443     src/profiling/CounterDirectory.cpp
444     src/profiling/CounterDirectory.hpp
445     src/profiling/EncodeVersion.hpp
446     src/profiling/Holder.cpp
447     src/profiling/Holder.hpp
448     src/profiling/IBufferManager.hpp
449     src/profiling/ICounterDirectory.hpp
450     src/profiling/ICounterValues.hpp
451     src/profiling/ISendCounterPacket.hpp
452     src/profiling/IPacketBuffer.hpp
453     src/profiling/IPeriodicCounterCapture.hpp
454     src/profiling/IProfilingConnection.hpp
455     src/profiling/IProfilingConnectionFactory.hpp
456     src/profiling/Packet.hpp
457     src/profiling/PacketBuffer.cpp
458     src/profiling/PacketBuffer.hpp
459     src/profiling/PacketVersionResolver.cpp
460     src/profiling/PacketVersionResolver.hpp
461     src/profiling/PeriodicCounterCapture.hpp
462     src/profiling/PeriodicCounterCapture.cpp
463     src/profiling/PeriodicCounterSelectionCommandHandler.cpp
464     src/profiling/PeriodicCounterSelectionCommandHandler.hpp
465     src/profiling/PerJobCounterSelectionCommandHandler.cpp
466     src/profiling/PerJobCounterSelectionCommandHandler.hpp
467     src/profiling/ProfilingConnectionDumpToFileDecorator.cpp
468     src/profiling/ProfilingConnectionDumpToFileDecorator.hpp
469     src/profiling/ProfilingConnectionFactory.cpp
470     src/profiling/ProfilingConnectionFactory.hpp
471     src/profiling/ProfilingService.cpp
472     src/profiling/ProfilingService.hpp
473     src/profiling/ProfilingStateMachine.cpp
474     src/profiling/ProfilingStateMachine.hpp
475     src/profiling/ProfilingUtils.cpp
476     src/profiling/ProfilingUtils.hpp
477     src/profiling/RequestCounterDirectoryCommandHandler.cpp
478     src/profiling/RequestCounterDirectoryCommandHandler.hpp
479     src/profiling/SendCounterPacket.cpp
480     src/profiling/SendCounterPacket.hpp
481     src/profiling/SocketProfilingConnection.cpp
482     src/profiling/SocketProfilingConnection.hpp
483     third-party/half/half.hpp
484     )
485
486 # Files used for Streamline-based profiling backend
487 if(PROFILING_BACKEND_STREAMLINE)
488     list(APPEND armnn_sources
489         ${GATOR_ROOT}/annotate/streamline_annotate.h
490         ${GATOR_ROOT}/annotate/streamline_annotate.c)
491 endif()
492
493 # the backends under src/backends extend the list of
494 # object libs armnn to include in the build
495 include(src/backends/backends.cmake)
496 foreach(lib ${armnnLibraries})
497     message("Adding object library dependency to armnn: ${lib}")
498     list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
499 endforeach()
500
501 add_library_ex(armnn SHARED ${armnn_sources})
502
503 target_include_directories(armnn PRIVATE src/armnn)
504 target_include_directories(armnn PRIVATE src/armnnUtils)
505 target_include_directories(armnn PRIVATE src/backends)
506 target_include_directories(armnn PRIVATE src/profiling)
507
508 target_link_libraries(armnn armnnUtils)
509
510 target_link_libraries(armnn ${CMAKE_DL_LIBS})
511
512 install(TARGETS armnn
513         LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
514         RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
515 if(BUILD_CAFFE_PARSER)
516     install(TARGETS armnnCaffeParser
517             LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
518             RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
519 endif()
520 if(BUILD_ONNX_PARSER)
521     install(TARGETS armnnOnnxParser
522             LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
523             RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
524 endif()
525 if(BUILD_TF_PARSER)
526     install(TARGETS armnnTfParser
527             LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
528             RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
529 endif()
530
531 install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
532
533 target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY}
534                             ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
535
536 if(ARMCOMPUTENEON OR ARMCOMPUTECL)
537     target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
538 endif()
539 if(ARMCOMPUTECL)
540     target_link_libraries(armnn ${OPENCL_LIBRARIES})
541 endif()
542
543 if(PROFILING_BACKEND_STREAMLINE)
544     target_link_libraries(armnn pthread)
545 endif()
546
547 if(BUILD_UNIT_TESTS)
548     set(unittest_sources)
549     list(APPEND unittest_sources
550         src/armnn/test/ConstTensorLayerVisitor.hpp
551         src/armnn/test/ConstTensorLayerVisitor.cpp
552         src/armnn/test/CreateWorkload.hpp
553         src/armnn/test/CsvReaderTest.cpp
554         src/armnn/test/EndToEndTest.cpp
555         src/armnn/test/ExecutionFrameTest.cpp
556         src/armnn/test/FloatingPointConverterTest.cpp
557         src/armnn/test/FlowControl.cpp
558         src/armnn/test/GraphTests.cpp
559         src/armnn/test/GraphUtils.cpp
560         src/armnn/test/GraphUtils.hpp
561         src/armnn/test/InstrumentTests.cpp
562         src/armnn/test/InferOutputTests.cpp
563         src/armnn/test/InferOutputTests.hpp
564         src/armnn/test/ModelAccuracyCheckerTest.cpp
565         src/armnn/test/NetworkTests.cpp
566         src/armnn/test/ObservableTest.cpp
567         src/armnn/test/OptimizerTests.cpp
568         src/armnn/test/optimizations/ConvertConstantsFloatToHalfTests.cpp
569         src/armnn/test/optimizations/ConvertConstantsHalfToFloatTests.cpp
570         src/armnn/test/optimizations/Fp32NetworkToFp16ConverterTests.cpp
571         src/armnn/test/optimizations/InsertDebugLayerTests.cpp
572         src/armnn/test/optimizations/MovePermuteUpTests.cpp
573         src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp
574         src/armnn/test/optimizations/OptimizeInverseConversionsTests.cpp
575         src/armnn/test/optimizations/OptimizeInversePermutesTests.cpp
576         src/armnn/test/optimizations/PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp
577         src/armnn/test/optimizations/PermuteAsReshapeTests.cpp
578         src/armnn/test/optimizations/SquashEqualSiblingsTests.cpp
579         src/armnn/test/OptionalTest.cpp
580         src/armnn/test/ProfilerTests.cpp
581         src/armnn/test/ProfilingEventTest.cpp
582         src/armnn/test/SubgraphViewTests.cpp
583         src/armnn/test/TensorHandleStrategyTest.cpp
584         src/armnn/test/TensorHelpers.hpp
585         src/armnn/test/TensorTest.cpp
586         src/armnn/test/TestInputOutputLayerVisitor.cpp
587         src/armnn/test/TestInputOutputLayerVisitor.hpp
588         src/armnn/test/TestLayerVisitor.cpp
589         src/armnn/test/TestLayerVisitor.hpp
590         src/armnn/test/TestNameOnlyLayerVisitor.cpp
591         src/armnn/test/TestNameOnlyLayerVisitor.hpp
592         src/armnn/test/TestNameAndDescriptorLayerVisitor.hpp
593         src/armnn/test/TestNameAndDescriptorLayerVisitor.cpp
594         src/armnn/test/TestUtils.cpp
595         src/armnn/test/TestUtils.hpp
596         src/armnn/test/UnitTests.cpp
597         src/armnn/test/UnitTests.hpp
598         src/armnn/test/UtilsTests.cpp
599         src/armnnUtils/test/PrototxtConversionsTest.cpp
600         src/armnnUtils/test/ParserHelperTest.cpp
601         src/armnnUtils/test/TensorUtilsTest.cpp
602         src/profiling/test/BufferTests.cpp
603         src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp
604         src/profiling/test/ProfilingTests.cpp
605         src/profiling/test/ProfilingTests.hpp
606         src/profiling/test/SendCounterPacketTests.cpp
607         src/profiling/test/SendCounterPacketTests.hpp
608         src/profiling/test/TimelinePacketTests.cpp
609         )
610
611     if(ARMNNREF)
612         list(APPEND unittest_sources
613             src/armnn/test/DebugCallbackTest.cpp
614             src/armnn/test/QuantizerTest.cpp
615             src/armnn/test/RuntimeTests.cpp
616             src/armnn/test/RuntimeTests.hpp
617             )
618     endif()
619
620     if(BUILD_TF_PARSER AND ARMNNREF)
621         list(APPEND unittest_sources
622             src/armnnTfParser/test/Activations.cpp
623             src/armnnTfParser/test/Addition.cpp
624             src/armnnTfParser/test/AddN.cpp
625             src/armnnTfParser/test/Assert.cpp
626             src/armnnTfParser/test/BiasAdd.cpp
627             src/armnnTfParser/test/BroadcastForAdd.cpp
628             src/armnnTfParser/test/Convolution2d.cpp
629             src/armnnTfParser/test/Concat.cpp
630             src/armnnTfParser/test/ConcatOfConcats.cpp
631             src/armnnTfParser/test/DepthwiseConvolution2d.cpp
632             src/armnnTfParser/test/Equal.cpp
633             src/armnnTfParser/test/ExpandDims.cpp
634             src/armnnTfParser/test/FusedBatchNorm.cpp
635             src/armnnTfParser/test/Gather.cpp
636             src/armnnTfParser/test/Greater.cpp
637             src/armnnTfParser/test/Identity.cpp
638             src/armnnTfParser/test/LocalResponseNormalization.cpp
639             src/armnnTfParser/test/Maximum.cpp
640             src/armnnTfParser/test/MaximumForLeakyRelu.cpp
641             src/armnnTfParser/test/Mean.cpp
642             src/armnnTfParser/test/Minimum.cpp
643             src/armnnTfParser/test/Multiplication.cpp
644             src/armnnTfParser/test/Pad.cpp
645             src/armnnTfParser/test/MultiOutput.cpp
646             src/armnnTfParser/test/PassThru.cpp
647             src/armnnTfParser/test/Pooling.cpp
648             src/armnnTfParser/test/RealDiv.cpp
649             src/armnnTfParser/test/Reshape.cpp
650             src/armnnTfParser/test/ResizeBilinear.cpp
651             src/armnnTfParser/test/Rsqrt.cpp
652             src/armnnTfParser/test/Shape.cpp
653             src/armnnTfParser/test/Softmax.cpp
654             src/armnnTfParser/test/TestDependencies.cpp
655             src/armnnTfParser/test/FullyConnected.cpp
656             src/armnnTfParser/test/Constant.cpp
657             src/armnnTfParser/test/TestMultiInputsOutputs.cpp
658             src/armnnTfParser/test/Split.cpp
659             src/armnnTfParser/test/Squeeze.cpp
660             src/armnnTfParser/test/Sub.cpp
661             )
662
663     endif()
664
665     if(BUILD_TF_LITE_PARSER AND ARMNNREF)
666         enable_language(ASM)
667         list(APPEND unittest_sources
668              src/armnnTfLiteParser/test/Activations.cpp
669              src/armnnTfLiteParser/test/Addition.cpp
670              src/armnnTfLiteParser/test/AvgPool2D.cpp
671              src/armnnTfLiteParser/test/BatchToSpaceND.cpp
672              src/armnnTfLiteParser/test/Concatenation.cpp
673              src/armnnTfLiteParser/test/Constant.cpp
674              src/armnnTfLiteParser/test/Conv2D.cpp
675              src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
676              src/armnnTfLiteParser/test/DetectionPostProcess.cpp
677              src/armnnTfLiteParser/test/FullyConnected.cpp
678              src/armnnTfLiteParser/test/L2Normalization.cpp
679              src/armnnTfLiteParser/test/Maximum.cpp
680              src/armnnTfLiteParser/test/MaxPool2D.cpp
681              src/armnnTfLiteParser/test/Mean.cpp
682              src/armnnTfLiteParser/test/Minimum.cpp
683              src/armnnTfLiteParser/test/Multiplication.cpp
684              src/armnnTfLiteParser/test/Pack.cpp
685              src/armnnTfLiteParser/test/Pad.cpp
686              src/armnnTfLiteParser/test/Reshape.cpp
687              src/armnnTfLiteParser/test/ResizeBilinear.cpp
688              src/armnnTfLiteParser/test/Softmax.cpp
689              src/armnnTfLiteParser/test/SpaceToBatchND.cpp
690              src/armnnTfLiteParser/test/Split.cpp
691              src/armnnTfLiteParser/test/Squeeze.cpp
692              src/armnnTfLiteParser/test/StridedSlice.cpp
693              src/armnnTfLiteParser/test/Sub.cpp
694              src/armnnTfLiteParser/test/TransposeConv.cpp
695              src/armnnTfLiteParser/test/Transpose.cpp
696              src/armnnTfLiteParser/test/Unpack.cpp
697              src/armnnTfLiteParser/test/LoadModel.cpp
698              src/armnnTfLiteParser/test/GetBuffer.cpp
699              src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
700              src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
701              src/armnnTfLiteParser/test/GetTensorIds.cpp
702              src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
703              src/armnnTfLiteParser/test/GetInputsOutputs.cpp
704              src/armnnTfLiteParser/test/Schema.s
705              )
706         set_source_files_properties(src/armnnTfLiteParser/test/Schema.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
707
708     endif()
709
710     if(BUILD_CAFFE_PARSER AND ARMNNREF)
711         list(APPEND unittest_sources
712             src/armnnCaffeParser/test/TestAdd.cpp
713             src/armnnCaffeParser/test/TestConcat.cpp
714             src/armnnCaffeParser/test/TestConvolution.cpp
715             src/armnnCaffeParser/test/TestDropout.cpp
716             src/armnnCaffeParser/test/TestInputs.cpp
717             src/armnnCaffeParser/test/TestMul.cpp
718             src/armnnCaffeParser/test/TestPooling2d.cpp
719             src/armnnCaffeParser/test/TestInPlace.cpp
720             src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
721             src/armnnCaffeParser/test/TestSplit.cpp
722             )
723     endif()
724
725     if(BUILD_ONNX_PARSER AND ARMNNREF)
726         list(APPEND unittest_sources
727             src/armnnOnnxParser/test/Constructor.cpp
728             src/armnnOnnxParser/test/CreateNetwork.cpp
729             src/armnnOnnxParser/test/ProtoxtFixture.cpp
730             src/armnnOnnxParser/test/Const.cpp
731             src/armnnOnnxParser/test/Pooling.cpp
732             src/armnnOnnxParser/test/Reshape.cpp
733             src/armnnOnnxParser/test/Relu.cpp
734             src/armnnOnnxParser/test/Conv2D.cpp
735             src/armnnOnnxParser/test/Addition.cpp
736             src/armnnOnnxParser/test/FullyConnected.cpp
737             src/armnnOnnxParser/test/GetInputsOutputs.cpp
738             src/armnnOnnxParser/test/BatchNorm.cpp
739             src/armnnOnnxParser/test/DepthConv.cpp
740             )
741     endif()
742
743     if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
744         list(APPEND unittest_sources
745              src/armnnQuantizer/test/QuantizationDataSetTests.cpp
746              )
747     endif()
748
749     if(BUILD_ARMNN_SERIALIZER AND ARMNNREF)
750         enable_language(ASM)
751         list(APPEND unittest_sources
752             src/armnnSerializer/test/ActivationSerializationTests.cpp
753             src/armnnSerializer/test/SerializerTests.cpp
754             src/armnnDeserializer/test/DeserializeAbs.cpp
755             src/armnnDeserializer/test/DeserializeActivation.cpp
756             src/armnnDeserializer/test/DeserializeAdd.cpp
757             src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp
758             src/armnnDeserializer/test/DeserializeBatchNormalization.cpp
759             src/armnnDeserializer/test/DeserializeConstant.cpp
760             src/armnnDeserializer/test/DeserializeConvolution2d.cpp
761             src/armnnDeserializer/test/DeserializeDepthToSpace.cpp
762             src/armnnDeserializer/test/DeserializeDivision.cpp
763             src/armnnDeserializer/test/DeserializeEqual.cpp
764             src/armnnDeserializer/test/DeserializeFloor.cpp
765             src/armnnDeserializer/test/DeserializeFullyConnected.cpp
766             src/armnnDeserializer/test/DeserializeGather.cpp
767             src/armnnDeserializer/test/DeserializeGreater.cpp
768             src/armnnDeserializer/test/DeserializeInstanceNormalization.cpp
769             src/armnnDeserializer/test/DeserializeL2Normalization.cpp
770             src/armnnDeserializer/test/DeserializeLogSoftmax.cpp
771             src/armnnDeserializer/test/DeserializeMean.cpp
772             src/armnnDeserializer/test/DeserializeMultiplication.cpp
773             src/armnnDeserializer/test/DeserializeNormalization.cpp
774             src/armnnDeserializer/test/DeserializePad.cpp
775             src/armnnDeserializer/test/DeserializePermute.cpp
776             src/armnnDeserializer/test/DeserializePooling2d.cpp
777             src/armnnDeserializer/test/DeserializeReshape.cpp
778             src/armnnDeserializer/test/DeserializeResizeBilinear.cpp
779             src/armnnDeserializer/test/DeserializeRsqrt.cpp
780             src/armnnDeserializer/test/DeserializeSlice.cpp
781             src/armnnDeserializer/test/DeserializeSpaceToBatchNd.cpp
782             src/armnnDeserializer/test/DeserializeStridedSlice.cpp
783             src/armnnDeserializer/test/DeserializeSubtraction.cpp
784             src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp
785             src/armnnDeserializer/test/SchemaSerialize.s
786             )
787         set_source_files_properties(src/armnnDeserializer/test/SchemaSerialize.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
788     endif()
789
790     if(BUILD_GATORD_MOCK)
791         list(APPEND unittest_sources
792             tests/profiling/gatordmock/tests/GatordMockTests.cpp
793             )
794     endif()
795
796     foreach(lib ${armnnUnitTestLibraries})
797         message("Adding object library dependency to UnitTests: ${lib}")
798         list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
799     endforeach()
800
801     add_executable(UnitTests ${unittest_sources})
802     target_include_directories(UnitTests PRIVATE src/armnn)
803     target_include_directories(UnitTests PRIVATE src/armnnUtils)
804     target_include_directories(UnitTests PRIVATE src/backends)
805     target_include_directories(UnitTests PRIVATE src/profiling)
806
807     if(VALGRIND_FOUND)
808         if(HEAP_PROFILING OR LEAK_CHECKING)
809             message("Valgrind is disabled for heap profiling and leak checking builds.")
810         else()
811             # Valgrind works with gperftools version number <= 2.4
812             target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
813         endif()
814     endif()
815
816     target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
817     target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
818
819     target_link_libraries(UnitTests armnn)
820     target_link_libraries(UnitTests armnnUtils)
821
822     if(BUILD_TF_PARSER)
823         target_link_libraries(UnitTests armnnTfParser)
824     endif()
825
826     if(BUILD_GATORD_MOCK)
827         target_link_libraries(UnitTests gatordMockService)
828     endif()
829
830     if(BUILD_TF_LITE_PARSER)
831         target_include_directories(UnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
832         target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
833         target_link_libraries(UnitTests armnnTfLiteParser)
834     endif()
835
836     if(BUILD_CAFFE_PARSER)
837         target_link_libraries(UnitTests armnnCaffeParser)
838     endif()
839
840     if(BUILD_ARMNN_SERIALIZER AND ARMNNREF)
841         target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer)
842         target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
843         target_link_libraries(UnitTests armnnSerializer)
844     endif()
845
846     if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
847         target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnQuantizer)
848         target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
849         target_link_libraries(UnitTests armnnQuantizer armnnSerializer)
850     endif()
851
852     if(BUILD_ONNX_PARSER)
853         target_link_libraries(UnitTests armnnOnnxParser)
854     endif()
855
856     addDllCopyCommands(UnitTests)
857 endif()
858
859 if (BUILD_ARMNN_SERIALIZER AND (BUILD_TF_PARSER OR BUILD_TF_LITE_PARSER OR BUILD_ONNX_PARSER OR BUILD_CAFFE_PARSER) AND ARMNNREF)
860     set(ArmnnConverter_sources
861         src/armnnConverter/ArmnnConverter.cpp)
862
863     add_executable_ex(ArmnnConverter ${ArmnnConverter_sources})
864     target_include_directories(ArmnnConverter PRIVATE src/armnn)
865     target_include_directories(ArmnnConverter PRIVATE src/armnnUtils)
866
867     if(BUILD_CAFFE_PARSER)
868         target_link_libraries(ArmnnConverter armnnCaffeParser)
869     endif()
870
871     if(BUILD_ONNX_PARSER)
872         target_link_libraries(ArmnnConverter armnnOnnxParser)
873     endif()
874
875     if(BUILD_TF_PARSER)
876         target_link_libraries(ArmnnConverter armnnTfParser)
877     endif()
878
879     if(BUILD_TF_LITE_PARSER)
880         target_link_libraries(ArmnnConverter armnnTfLiteParser)
881     endif()
882
883     target_link_libraries(ArmnnConverter armnnSerializer)
884
885     target_link_libraries(ArmnnConverter armnn)
886     target_link_libraries(ArmnnConverter ${CMAKE_THREAD_LIBS_INIT})
887     target_link_libraries(ArmnnConverter
888         ${Boost_SYSTEM_LIBRARY}
889         ${Boost_FILESYSTEM_LIBRARY}
890         ${Boost_PROGRAM_OPTIONS_LIBRARY})
891     addDllCopyCommands(ArmnnConverter)
892 endif()
893
894 if(BUILD_GATORD_MOCK)
895     set(gatord_mock_sources)
896     list(APPEND gatord_mock_sources
897         tests/profiling/gatordmock/CommandFileParser.hpp
898         tests/profiling/gatordmock/CommandFileParser.cpp
899         tests/profiling/gatordmock/CommandLineProcessor.hpp
900         tests/profiling/gatordmock/CommandLineProcessor.cpp
901         tests/profiling/gatordmock/GatordMockService.hpp
902         tests/profiling/gatordmock/GatordMockService.cpp
903         tests/profiling/gatordmock/MockUtils.hpp
904         tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.cpp
905         tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp
906         )
907
908     include_directories(${Boost_INCLUDE_DIRS} src/profiling)
909
910     add_library_ex(gatordMockService STATIC ${gatord_mock_sources})
911
912     add_executable_ex(GartordMock tests/profiling/gatordmock/GatordMockMain.cpp)
913
914     if(Threads_FOUND AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android)))
915         target_link_libraries(GartordMock pthread)
916     endif()
917
918     target_link_libraries(GartordMock
919         armnn
920         gatordMockService
921         ${Boost_PROGRAM_OPTIONS_LIBRARY}
922         ${Boost_SYSTEM_LIBRARY})
923
924 endif()