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
11 cmake/GlobalConfig.cmake
12 cmake/AddDllCopyCommands.cmake)
14 foreach(cmake_file ${additional_cmake_files})
15 include(${cmake_file})
18 include(GNUInstallDirs)
20 add_subdirectory(samples)
21 add_subdirectory(src/armnnTfLiteParser)
22 add_subdirectory(src/armnnSerializer)
25 add_subdirectory(tests)
28 # Include the additional cmake files in their own target so that they will appear nicely in IDEs
29 add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files})
31 set(armnnUtils_sources)
32 list(APPEND armnnUtils_sources
33 src/armnnUtils/GraphTopologicalSort.hpp
34 src/armnnUtils/Half.hpp
35 src/armnnUtils/Logging.hpp
36 src/armnnUtils/Logging.cpp
37 src/armnnUtils/Permute.hpp
38 src/armnnUtils/Permute.cpp
39 src/armnnUtils/DataLayoutIndexed.cpp
40 src/armnnUtils/DataLayoutIndexed.hpp
41 src/armnnUtils/DotSerializer.cpp
42 src/armnnUtils/DotSerializer.hpp
43 src/armnnUtils/HeapProfiling.cpp
44 src/armnnUtils/HeapProfiling.hpp
45 src/armnnUtils/LeakChecking.cpp
46 src/armnnUtils/LeakChecking.hpp
47 src/armnnUtils/ModelAccuracyChecker.cpp
48 src/armnnUtils/ModelAccuracyChecker.hpp
49 src/armnnUtils/CsvReader.cpp
50 src/armnnUtils/CsvReader.hpp
51 src/armnnUtils/FloatingPointConverter.cpp
52 src/armnnUtils/FloatingPointConverter.hpp
53 src/armnnUtils/VerificationHelpers.hpp
54 src/armnnUtils/VerificationHelpers.cpp
55 src/armnnUtils/ParserHelper.hpp
56 src/armnnUtils/ParserHelper.cpp
57 src/armnnUtils/ParserPrototxtFixture.hpp
58 src/armnnUtils/PrototxtConversions.hpp
59 src/armnnUtils/PrototxtConversions.cpp
60 src/armnnUtils/TensorIOUtils.hpp
61 src/armnnUtils/TensorUtils.hpp
62 src/armnnUtils/TensorUtils.cpp
65 add_library_ex(armnnUtils STATIC ${armnnUtils_sources})
66 target_include_directories(armnnUtils PRIVATE src/backends)
68 if(BUILD_CAFFE_PARSER)
69 # ArmNN Parser source files required for all build options
70 set(armnn_caffe_parser_sources)
71 list(APPEND armnn_caffe_parser_sources
72 include/armnnCaffeParser/ICaffeParser.hpp
73 src/armnnCaffeParser/RecordByRecordCaffeParser.hpp
74 src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
75 src/armnnCaffeParser/CaffeParser.hpp
76 src/armnnCaffeParser/CaffeParser.cpp
77 ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc
79 # The generated Caffe protobuf .cc file is not warning clean and we can't fix them.
80 if(COMPILER_IS_GNU_LIKE)
81 set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
84 add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})
85 set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")
87 target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)
89 target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
91 target_link_libraries(armnnCaffeParser armnn)
92 target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})
97 set(armnn_onnx_parser_sources)
98 list(APPEND armnn_onnx_parser_sources
99 include/armnnOnnxParser/IOnnxParser.hpp
100 src/armnnOnnxParser/OnnxParser.hpp
101 src/armnnOnnxParser/OnnxParser.cpp
102 ${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc
104 # The generated onnx protobuf .cc files are not warning clean and we can't fix them.
105 if(COMPILER_IS_GNU_LIKE)
106 set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
109 add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})
111 target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)
113 target_link_libraries(armnnOnnxParser armnn)
116 target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})
120 set(armnn_tf_parser_sources)
121 list(APPEND armnn_tf_parser_sources
122 include/armnnTfParser/ITfParser.hpp
123 src/armnnTfParser/TfParser.hpp
124 src/armnnTfParser/TfParser.cpp
127 # The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.
128 if(COMPILER_IS_GNU_LIKE)
129 set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
132 add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})
134 target_include_directories(armnnTfParser PRIVATE src/armnnUtils)
136 target_link_libraries(armnnTfParser armnn)
138 # Protobuf (use the specific version tensorflow wants)
139 target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})
142 if(BUILD_ARMNN_QUANTIZER)
143 if(NOT BUILD_ARMNN_SERIALIZER)
144 message(ERROR, "In order to build the ArmNN Quantization Tool you must set BUILD_ARMNN_SERIALIZER = YES")
147 set(armnn_quantizer_sources)
148 list(APPEND armnn_quantizer_sources
149 src/armnnQuantizer/QuantizationDataSet.hpp
150 src/armnnQuantizer/QuantizationDataSet.cpp
151 src/armnnQuantizer/QuantizationInput.hpp
152 src/armnnQuantizer/QuantizationInput.cpp
153 src/armnnQuantizer/CommandLineProcessor.hpp
154 src/armnnQuantizer/CommandLineProcessor.cpp
157 add_library_ex(armnnQuantizer SHARED ${armnn_quantizer_sources})
159 target_include_directories(armnnQuantizer PRIVATE include/armnnDeserializer)
160 target_include_directories(armnnQuantizer PRIVATE src/armnnUtils)
161 target_include_directories(armnnQuantizer PRIVATE src/armnn)
163 include_directories(SYSTEM "${FLATBUFFERS_INCLUDE_PATH}")
164 set_target_properties(armnnQuantizer PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR})
166 target_link_libraries(armnnQuantizer
168 ${Boost_SYSTEM_LIBRARY}
169 ${Boost_PROGRAM_OPTIONS_LIBRARY}
170 ${Boost_FILESYSTEM_LIBRARY}
172 ${Boost_THREAD_LIBRARY} )
174 add_executable_ex(ArmnnQuantizer
175 src/armnnQuantizer/ArmNNQuantizerMain.cpp)
177 target_include_directories(ArmnnQuantizer PRIVATE include/armnnDeserializer)
178 target_include_directories(ArmnnQuantizer PRIVATE src/armnn)
180 target_link_libraries(ArmnnQuantizer
181 ${Boost_SYSTEM_LIBRARY}
182 ${Boost_PROGRAM_OPTIONS_LIBRARY}
183 ${Boost_FILESYSTEM_LIBRARY}
185 ${Boost_THREAD_LIBRARY} )
187 target_link_libraries(ArmnnQuantizer
192 ${FLATBUFFERS_LIBRARY})
194 if(Threads_FOUND AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android)))
195 target_link_libraries(ArmnnQuantizer pthread)
201 list(APPEND armnn_sources
202 include/armnn/ArmNN.hpp
203 include/armnn/BackendId.hpp
204 include/armnn/Descriptors.hpp
205 include/armnn/DescriptorsFwd.hpp
206 include/armnn/Exceptions.hpp
207 include/armnn/ILayerSupport.hpp
208 include/armnn/ILayerVisitor.hpp
209 include/armnn/INetwork.hpp
210 include/armnn/IProfiler.hpp
211 include/armnn/IRuntime.hpp
212 include/armnn/LayerSupport.hpp
213 include/armnn/LayerVisitorBase.hpp
214 include/armnn/LstmParams.hpp
215 include/armnn/NetworkFwd.hpp
216 include/armnn/Optional.hpp
217 include/armnn/QuantizedLstmParams.hpp
218 include/armnn/Tensor.hpp
219 include/armnn/TensorFwd.hpp
220 include/armnn/Types.hpp
221 include/armnn/TypesUtils.hpp
222 include/armnn/Utils.hpp
223 include/armnn/Version.hpp
224 src/armnn/layers/LayerCloneBase.hpp
225 src/armnn/layers/LayerWithParameters.hpp
226 src/armnn/layers/ActivationLayer.hpp
227 src/armnn/layers/ActivationLayer.cpp
228 src/armnn/layers/AdditionLayer.hpp
229 src/armnn/layers/AdditionLayer.cpp
230 src/armnn/layers/BatchNormalizationLayer.hpp
231 src/armnn/layers/BatchNormalizationLayer.cpp
232 src/armnn/layers/BatchToSpaceNdLayer.hpp
233 src/armnn/layers/BatchToSpaceNdLayer.cpp
234 src/armnn/layers/ConcatLayer.hpp
235 src/armnn/layers/ConcatLayer.cpp
236 src/armnn/layers/ConstantLayer.hpp
237 src/armnn/layers/ConstantLayer.cpp
238 src/armnn/layers/Convolution2dLayer.hpp
239 src/armnn/layers/Convolution2dLayer.cpp
240 src/armnn/layers/ConvertFp16ToFp32Layer.hpp
241 src/armnn/layers/ConvertFp16ToFp32Layer.cpp
242 src/armnn/layers/ConvertFp32ToFp16Layer.hpp
243 src/armnn/layers/ConvertFp32ToFp16Layer.cpp
244 src/armnn/layers/DebugLayer.hpp
245 src/armnn/layers/DebugLayer.cpp
246 src/armnn/layers/DepthwiseConvolution2dLayer.hpp
247 src/armnn/layers/DepthwiseConvolution2dLayer.cpp
248 src/armnn/layers/DequantizeLayer.hpp
249 src/armnn/layers/DequantizeLayer.cpp
250 src/armnn/layers/DetectionPostProcessLayer.hpp
251 src/armnn/layers/DetectionPostProcessLayer.cpp
252 src/armnn/layers/ElementwiseBaseLayer.hpp
253 src/armnn/layers/ElementwiseBaseLayer.cpp
254 src/armnn/layers/EqualLayer.hpp
255 src/armnn/layers/EqualLayer.cpp
256 src/armnn/layers/FakeQuantizationLayer.hpp
257 src/armnn/layers/FakeQuantizationLayer.cpp
258 src/armnn/layers/FloorLayer.hpp
259 src/armnn/layers/FloorLayer.cpp
260 src/armnn/layers/FullyConnectedLayer.hpp
261 src/armnn/layers/FullyConnectedLayer.cpp
262 src/armnn/layers/GatherLayer.cpp
263 src/armnn/layers/GatherLayer.hpp
264 src/armnn/layers/GreaterLayer.cpp
265 src/armnn/layers/GreaterLayer.hpp
266 src/armnn/layers/InputLayer.hpp
267 src/armnn/layers/InputLayer.cpp
268 src/armnn/layers/L2NormalizationLayer.hpp
269 src/armnn/layers/L2NormalizationLayer.cpp
270 src/armnn/layers/LstmLayer.cpp
271 src/armnn/layers/LstmLayer.hpp
272 src/armnn/layers/MaximumLayer.cpp
273 src/armnn/layers/MaximumLayer.hpp
274 src/armnn/layers/MeanLayer.hpp
275 src/armnn/layers/MeanLayer.cpp
276 src/armnn/layers/MemCopyLayer.hpp
277 src/armnn/layers/MemCopyLayer.cpp
278 src/armnn/layers/MergeLayer.hpp
279 src/armnn/layers/MergeLayer.cpp
280 src/armnn/layers/MinimumLayer.cpp
281 src/armnn/layers/MinimumLayer.hpp
282 src/armnn/layers/MultiplicationLayer.hpp
283 src/armnn/layers/MultiplicationLayer.cpp
284 src/armnn/layers/NormalizationLayer.hpp
285 src/armnn/layers/NormalizationLayer.cpp
286 src/armnn/layers/OutputLayer.hpp
287 src/armnn/layers/OutputLayer.cpp
288 src/armnn/layers/PadLayer.hpp
289 src/armnn/layers/PadLayer.cpp
290 src/armnn/layers/PermuteLayer.hpp
291 src/armnn/layers/PermuteLayer.cpp
292 src/armnn/layers/Pooling2dLayer.hpp
293 src/armnn/layers/Pooling2dLayer.cpp
294 src/armnn/layers/QuantizeLayer.cpp
295 src/armnn/layers/QuantizeLayer.hpp
296 src/armnn/layers/QuantizedLstmLayer.hpp
297 src/armnn/layers/QuantizedLstmLayer.cpp
298 src/armnn/layers/DivisionLayer.cpp
299 src/armnn/layers/DivisionLayer.hpp
300 src/armnn/layers/PreCompiledLayer.hpp
301 src/armnn/layers/PreCompiledLayer.cpp
302 src/armnn/layers/PreluLayer.hpp
303 src/armnn/layers/PreluLayer.cpp
304 src/armnn/layers/ReshapeLayer.hpp
305 src/armnn/layers/ReshapeLayer.cpp
306 src/armnn/layers/ResizeLayer.hpp
307 src/armnn/layers/ResizeLayer.cpp
308 src/armnn/layers/RsqrtLayer.cpp
309 src/armnn/layers/RsqrtLayer.hpp
310 src/armnn/layers/SoftmaxLayer.hpp
311 src/armnn/layers/SoftmaxLayer.cpp
312 src/armnn/layers/SpaceToBatchNdLayer.hpp
313 src/armnn/layers/SpaceToBatchNdLayer.cpp
314 src/armnn/layers/SpaceToDepthLayer.hpp
315 src/armnn/layers/SpaceToDepthLayer.cpp
316 src/armnn/layers/SplitterLayer.hpp
317 src/armnn/layers/SplitterLayer.cpp
318 src/armnn/layers/StackLayer.hpp
319 src/armnn/layers/StackLayer.cpp
320 src/armnn/layers/StridedSliceLayer.cpp
321 src/armnn/layers/StridedSliceLayer.hpp
322 src/armnn/layers/SubtractionLayer.cpp
323 src/armnn/layers/SubtractionLayer.hpp
324 src/armnn/layers/SwitchLayer.cpp
325 src/armnn/layers/SwitchLayer.hpp
326 src/armnn/layers/TransposeConvolution2dLayer.cpp
327 src/armnn/layers/TransposeConvolution2dLayer.hpp
328 src/armnn/BackendSettings.hpp
329 src/armnn/BackendHelper.cpp
330 src/armnn/CompatibleTypes.hpp
331 src/armnn/Descriptors.cpp
332 src/armnn/DeviceSpec.hpp
333 src/armnn/DynamicQuantizationVisitor.cpp
334 src/armnn/DynamicQuantizationVisitor.hpp
335 src/armnn/Exceptions.cpp
336 src/armnn/ExecutionFrame.cpp
337 src/armnn/ExecutionFrame.hpp
340 src/armnn/IGraphObservable.hpp
341 src/armnn/Instrument.hpp
342 src/armnn/InternalTypes.cpp
343 src/armnn/InternalTypes.hpp
344 src/armnn/ISubgraphViewConverter.hpp
345 src/armnn/JsonPrinter.cpp
346 src/armnn/JsonPrinter.hpp
348 src/armnn/LayerFwd.hpp
350 src/armnn/LayersFwd.hpp
351 src/armnn/LayerSupportCommon.hpp
352 src/armnn/LayerSupport.cpp
353 src/armnn/LoadedNetwork.cpp
354 src/armnn/LoadedNetwork.hpp
355 src/armnn/Network.cpp
356 src/armnn/Network.hpp
357 src/armnn/NetworkQuantizationScheme.hpp
358 src/armnn/NetworkQuantizer.cpp
359 src/armnn/NetworkQuantizer.hpp
360 src/armnn/NetworkQuantizerUtils.cpp
361 src/armnn/NetworkQuantizerUtils.hpp
362 src/armnn/NetworkUtils.cpp
363 src/armnn/NetworkUtils.hpp
364 src/armnn/Observable.cpp
365 src/armnn/Observable.hpp
366 src/armnn/Optimizer.cpp
367 src/armnn/Optimizer.hpp
368 src/armnn/OverrideInputRangeVisitor.cpp
369 src/armnn/OverrideInputRangeVisitor.hpp
370 src/armnn/Profiling.cpp
371 src/armnn/ProfilingEvent.cpp
372 src/armnn/ProfilingEvent.hpp
373 src/armnn/Profiling.hpp
374 src/armnn/QuantizerVisitor.cpp
375 src/armnn/QuantizerVisitor.hpp
376 src/armnn/Runtime.cpp
377 src/armnn/Runtime.hpp
378 src/armnn/RangeTracker.cpp
379 src/armnn/RangeTracker.hpp
380 src/armnn/ResolveType.hpp
381 src/armnn/SerializeLayerParameters.cpp
382 src/armnn/SerializeLayerParameters.hpp
383 src/armnn/StaticRangeVisitor.cpp
384 src/armnn/StaticRangeVisitor.hpp
385 src/armnn/SubgraphView.cpp
386 src/armnn/SubgraphView.hpp
387 src/armnn/SubgraphViewSelector.cpp
388 src/armnn/SubgraphViewSelector.hpp
390 src/armnn/TypesUtils.cpp
392 src/armnn/WallClockTimer.cpp
393 src/armnn/WallClockTimer.hpp
394 src/armnn/optimizations/AddDebug.hpp
395 src/armnn/optimizations/All.hpp
396 src/armnn/optimizations/ConvertConstants.hpp
397 src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
398 src/armnn/optimizations/FoldPadIntoConvolution2d.hpp
399 src/armnn/optimizations/MovePermuteUp.hpp
400 src/armnn/optimizations/Optimization.hpp
401 src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
402 src/armnn/optimizations/OptimizeInverseConversions.hpp
403 src/armnn/optimizations/OptimizeInversePermutes.hpp
404 src/armnn/optimizations/PermuteAsReshape.hpp
405 src/armnn/optimizations/SquashEqualSiblings.hpp
406 third-party/half/half.hpp
409 # Files used for Streamline-based profiling backend
410 if(PROFILING_BACKEND_STREAMLINE)
411 list(APPEND armnn_sources
412 ${GATOR_ROOT}/annotate/streamline_annotate.h
413 ${GATOR_ROOT}/annotate/streamline_annotate.c)
416 # the backends under src/backends extend the list of
417 # object libs armnn to include in the build
418 include(src/backends/backends.cmake)
419 foreach(lib ${armnnLibraries})
420 message("Adding object library dependency to armnn: ${lib}")
421 list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
424 add_library_ex(armnn SHARED ${armnn_sources})
426 target_include_directories(armnn PRIVATE src/armnn)
427 target_include_directories(armnn PRIVATE src/armnnUtils)
428 target_include_directories(armnn PRIVATE src/backends)
430 target_link_libraries(armnn armnnUtils)
432 target_link_libraries(armnn ${CMAKE_DL_LIBS})
434 install(TARGETS armnn
435 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
436 if(BUILD_CAFFE_PARSER)
437 install(TARGETS armnnCaffeParser
438 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
440 if(BUILD_ONNX_PARSER)
441 install(TARGETS armnnOnnxParser
442 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
445 install(TARGETS armnnTfParser
446 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
449 install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
451 target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
453 if(ARMCOMPUTENEON OR ARMCOMPUTECL)
454 target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
457 target_link_libraries(armnn ${OPENCL_LIBRARIES})
460 if(PROFILING_BACKEND_STREAMLINE)
461 target_link_libraries(armnn pthread)
465 set(unittest_sources)
466 list(APPEND unittest_sources
467 src/armnn/test/ConstTensorLayerVisitor.hpp
468 src/armnn/test/ConstTensorLayerVisitor.cpp
469 src/armnn/test/CreateWorkload.hpp
470 src/armnn/test/CsvReaderTest.cpp
471 src/armnn/test/DebugCallbackTest.cpp
472 src/armnn/test/EndToEndTest.cpp
473 src/armnn/test/ExecutionFrameTest.cpp
474 src/armnn/test/FloatingPointConverterTest.cpp
475 src/armnn/test/FlowControl.cpp
476 src/armnn/test/GraphTests.cpp
477 src/armnn/test/GraphUtils.cpp
478 src/armnn/test/GraphUtils.hpp
479 src/armnn/test/InstrumentTests.cpp
480 src/armnn/test/InferOutputTests.cpp
481 src/armnn/test/InferOutputTests.hpp
482 src/armnn/test/ModelAccuracyCheckerTest.cpp
483 src/armnn/test/NetworkTests.cpp
484 src/armnn/test/ObservableTest.cpp
485 src/armnn/test/OptimizerTests.cpp
486 src/armnn/test/OptionalTest.cpp
487 src/armnn/test/ProfilerTests.cpp
488 src/armnn/test/ProfilingEventTest.cpp
489 src/armnn/test/QuantizerTest.cpp
490 src/armnn/test/RuntimeTests.cpp
491 src/armnn/test/RuntimeTests.hpp
492 src/armnn/test/SubgraphViewTests.cpp
493 src/armnn/test/TensorHandleStrategyTest.cpp
494 src/armnn/test/TensorHelpers.hpp
495 src/armnn/test/TensorTest.cpp
496 src/armnn/test/TestInputOutputLayerVisitor.cpp
497 src/armnn/test/TestInputOutputLayerVisitor.hpp
498 src/armnn/test/TestLayerVisitor.cpp
499 src/armnn/test/TestLayerVisitor.hpp
500 src/armnn/test/TestNameOnlyLayerVisitor.cpp
501 src/armnn/test/TestNameOnlyLayerVisitor.hpp
502 src/armnn/test/TestNameAndDescriptorLayerVisitor.hpp
503 src/armnn/test/TestNameAndDescriptorLayerVisitor.cpp
504 src/armnn/test/TestUtils.cpp
505 src/armnn/test/TestUtils.hpp
506 src/armnn/test/UnitTests.cpp
507 src/armnn/test/UnitTests.hpp
508 src/armnn/test/UtilsTests.cpp
509 src/armnnUtils/test/PrototxtConversionsTest.cpp
510 src/armnnUtils/test/ParserHelperTest.cpp
514 list(APPEND unittest_sources
515 src/armnnTfParser/test/Activations.cpp
516 src/armnnTfParser/test/Addition.cpp
517 src/armnnTfParser/test/AddN.cpp
518 src/armnnTfParser/test/Assert.cpp
519 src/armnnTfParser/test/BiasAdd.cpp
520 src/armnnTfParser/test/BroadcastForAdd.cpp
521 src/armnnTfParser/test/Convolution2d.cpp
522 src/armnnTfParser/test/Concat.cpp
523 src/armnnTfParser/test/ConcatOfConcats.cpp
524 src/armnnTfParser/test/DepthwiseConvolution2d.cpp
525 src/armnnTfParser/test/Equal.cpp
526 src/armnnTfParser/test/ExpandDims.cpp
527 src/armnnTfParser/test/FusedBatchNorm.cpp
528 src/armnnTfParser/test/Gather.cpp
529 src/armnnTfParser/test/Greater.cpp
530 src/armnnTfParser/test/Identity.cpp
531 src/armnnTfParser/test/LocalResponseNormalization.cpp
532 src/armnnTfParser/test/Maximum.cpp
533 src/armnnTfParser/test/MaximumForLeakyRelu.cpp
534 src/armnnTfParser/test/Mean.cpp
535 src/armnnTfParser/test/Minimum.cpp
536 src/armnnTfParser/test/Multiplication.cpp
537 src/armnnTfParser/test/Pad.cpp
538 src/armnnTfParser/test/MultiOutput.cpp
539 src/armnnTfParser/test/PassThru.cpp
540 src/armnnTfParser/test/Pooling.cpp
541 src/armnnTfParser/test/RealDiv.cpp
542 src/armnnTfParser/test/Reshape.cpp
543 src/armnnTfParser/test/ResizeBilinear.cpp
544 src/armnnTfParser/test/Rsqrt.cpp
545 src/armnnTfParser/test/Shape.cpp
546 src/armnnTfParser/test/Softmax.cpp
547 src/armnnTfParser/test/TestDependencies.cpp
548 src/armnnTfParser/test/FullyConnected.cpp
549 src/armnnTfParser/test/Constant.cpp
550 src/armnnTfParser/test/TestMultiInputsOutputs.cpp
551 src/armnnTfParser/test/Split.cpp
552 src/armnnTfParser/test/Squeeze.cpp
553 src/armnnTfParser/test/Sub.cpp
557 if(BUILD_TF_LITE_PARSER)
559 list(APPEND unittest_sources
560 src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp
561 src/armnnTfLiteParser/test/Activations.cpp
562 src/armnnTfLiteParser/test/Addition.cpp
563 src/armnnTfLiteParser/test/AvgPool2D.cpp
564 src/armnnTfLiteParser/test/BatchToSpaceND.cpp
565 src/armnnTfLiteParser/test/Concatenation.cpp
566 src/armnnTfLiteParser/test/Constant.cpp
567 src/armnnTfLiteParser/test/Conv2D.cpp
568 src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
569 src/armnnTfLiteParser/test/DetectionPostProcess.cpp
570 src/armnnTfLiteParser/test/FullyConnected.cpp
571 src/armnnTfLiteParser/test/L2Normalization.cpp
572 src/armnnTfLiteParser/test/Maximum.cpp
573 src/armnnTfLiteParser/test/MaxPool2D.cpp
574 src/armnnTfLiteParser/test/Mean.cpp
575 src/armnnTfLiteParser/test/Minimum.cpp
576 src/armnnTfLiteParser/test/Multiplication.cpp
577 src/armnnTfLiteParser/test/Pack.cpp
578 src/armnnTfLiteParser/test/Pad.cpp
579 src/armnnTfLiteParser/test/Reshape.cpp
580 src/armnnTfLiteParser/test/ResizeBilinear.cpp
581 src/armnnTfLiteParser/test/Softmax.cpp
582 src/armnnTfLiteParser/test/SpaceToBatchND.cpp
583 src/armnnTfLiteParser/test/Split.cpp
584 src/armnnTfLiteParser/test/Squeeze.cpp
585 src/armnnTfLiteParser/test/StridedSlice.cpp
586 src/armnnTfLiteParser/test/Sub.cpp
587 src/armnnTfLiteParser/test/Unpack.cpp
588 src/armnnTfLiteParser/test/LoadModel.cpp
589 src/armnnTfLiteParser/test/GetBuffer.cpp
590 src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
591 src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
592 src/armnnTfLiteParser/test/GetTensorIds.cpp
593 src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
594 src/armnnTfLiteParser/test/GetInputsOutputs.cpp
595 src/armnnTfLiteParser/test/Schema.s
597 set_source_files_properties(src/armnnTfLiteParser/test/Schema.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
600 if(BUILD_CAFFE_PARSER)
601 list(APPEND unittest_sources
602 src/armnnCaffeParser/test/TestAdd.cpp
603 src/armnnCaffeParser/test/TestConcat.cpp
604 src/armnnCaffeParser/test/TestConvolution.cpp
605 src/armnnCaffeParser/test/TestDropout.cpp
606 src/armnnCaffeParser/test/TestInputs.cpp
607 src/armnnCaffeParser/test/TestMul.cpp
608 src/armnnCaffeParser/test/TestPooling2d.cpp
609 src/armnnCaffeParser/test/TestInPlace.cpp
610 src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
611 src/armnnCaffeParser/test/TestSplit.cpp
615 if(BUILD_ARMNN_QUANTIZER)
616 list(APPEND unittest_sources
617 src/armnnQuantizer/test/QuantizationDataSetTests.cpp
621 if(BUILD_ARMNN_SERIALIZER)
623 list(APPEND unittest_sources
624 src/armnnSerializer/test/ActivationSerializationTests.cpp
625 src/armnnSerializer/test/SerializerTests.cpp
626 src/armnnDeserializer/test/DeserializeActivation.cpp
627 src/armnnDeserializer/test/DeserializeAdd.cpp
628 src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp
629 src/armnnDeserializer/test/DeserializeBatchNormalization.cpp
630 src/armnnDeserializer/test/DeserializeConstant.cpp
631 src/armnnDeserializer/test/DeserializeConvolution2d.cpp
632 src/armnnDeserializer/test/DeserializeDivision.cpp
633 src/armnnDeserializer/test/DeserializeEqual.cpp
634 src/armnnDeserializer/test/DeserializeFloor.cpp
635 src/armnnDeserializer/test/DeserializeFullyConnected.cpp
636 src/armnnDeserializer/test/DeserializeGather.cpp
637 src/armnnDeserializer/test/DeserializeGreater.cpp
638 src/armnnDeserializer/test/DeserializeL2Normalization.cpp
639 src/armnnDeserializer/test/DeserializeMean.cpp
640 src/armnnDeserializer/test/DeserializeMultiplication.cpp
641 src/armnnDeserializer/test/DeserializeNormalization.cpp
642 src/armnnDeserializer/test/DeserializePad.cpp
643 src/armnnDeserializer/test/DeserializePermute.cpp
644 src/armnnDeserializer/test/DeserializePooling2d.cpp
645 src/armnnDeserializer/test/DeserializeReshape.cpp
646 src/armnnDeserializer/test/DeserializeResizeBilinear.cpp
647 src/armnnDeserializer/test/DeserializeRsqrt.cpp
648 src/armnnDeserializer/test/DeserializeSpaceToBatchNd.cpp
649 src/armnnDeserializer/test/DeserializeStridedSlice.cpp
650 src/armnnDeserializer/test/DeserializeSubtraction.cpp
651 src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp
652 src/armnnDeserializer/test/SchemaSerialize.s
654 set_source_files_properties(src/armnnDeserializer/test/SchemaSerialize.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
657 if(BUILD_ONNX_PARSER)
658 list(APPEND unittest_sources
659 src/armnnOnnxParser/test/Constructor.cpp
660 src/armnnOnnxParser/test/CreateNetwork.cpp
661 src/armnnOnnxParser/test/ProtoxtFixture.cpp
662 src/armnnOnnxParser/test/Const.cpp
663 src/armnnOnnxParser/test/Pooling.cpp
664 src/armnnOnnxParser/test/Reshape.cpp
665 src/armnnOnnxParser/test/Relu.cpp
666 src/armnnOnnxParser/test/Conv2D.cpp
667 src/armnnOnnxParser/test/Addition.cpp
668 src/armnnOnnxParser/test/FullyConnected.cpp
669 src/armnnOnnxParser/test/GetInputsOutputs.cpp
670 src/armnnOnnxParser/test/BatchNorm.cpp
671 src/armnnOnnxParser/test/DepthConv.cpp
675 foreach(lib ${armnnUnitTestLibraries})
676 message("Adding object library dependency to UnitTests: ${lib}")
677 list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
680 add_executable(UnitTests ${unittest_sources})
681 target_include_directories(UnitTests PRIVATE src/armnn)
682 target_include_directories(UnitTests PRIVATE src/armnnUtils)
683 target_include_directories(UnitTests PRIVATE src/backends)
686 if(HEAP_PROFILING OR LEAK_CHECKING)
687 message("Valgrind is disabled for heap profiling and leak checking builds.")
689 # Valgrind works with gperftools version number <= 2.4
690 target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
694 target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
695 target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
697 target_link_libraries(UnitTests armnn)
698 target_link_libraries(UnitTests armnnUtils)
701 target_link_libraries(UnitTests armnnTfParser)
704 if(BUILD_TF_LITE_PARSER)
705 target_include_directories(UnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
706 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
707 target_link_libraries(UnitTests armnnTfLiteParser)
710 if(BUILD_CAFFE_PARSER)
711 target_link_libraries(UnitTests armnnCaffeParser)
714 if(BUILD_ARMNN_SERIALIZER)
715 target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer)
716 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
717 target_link_libraries(UnitTests armnnSerializer)
720 if(BUILD_ARMNN_QUANTIZER)
721 target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnQuantizer)
722 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
723 target_link_libraries(UnitTests armnnQuantizer armnnSerializer)
726 if(BUILD_ONNX_PARSER)
727 target_link_libraries(UnitTests armnnOnnxParser)
730 addDllCopyCommands(UnitTests)
733 if (BUILD_ARMNN_SERIALIZER AND (BUILD_TF_PARSER OR BUILD_TF_LITE_PARSER OR BUILD_ONNX_PARSER OR BUILD_CAFFE_PARSER))
734 set(ArmnnConverter_sources
735 src/armnnConverter/ArmnnConverter.cpp)
737 add_executable_ex(ArmnnConverter ${ArmnnConverter_sources})
738 target_include_directories(ArmnnConverter PRIVATE src/armnn)
739 target_include_directories(ArmnnConverter PRIVATE src/armnnUtils)
741 if(BUILD_CAFFE_PARSER)
742 target_link_libraries(ArmnnConverter armnnCaffeParser)
745 if(BUILD_ONNX_PARSER)
746 target_link_libraries(ArmnnConverter armnnOnnxParser)
750 target_link_libraries(ArmnnConverter armnnTfParser)
753 if(BUILD_TF_LITE_PARSER)
754 target_link_libraries(ArmnnConverter armnnTfLiteParser)
757 target_link_libraries(ArmnnConverter armnnSerializer)
759 target_link_libraries(ArmnnConverter armnn)
760 target_link_libraries(ArmnnConverter ${CMAKE_THREAD_LIBS_INIT})
761 target_link_libraries(ArmnnConverter
762 ${Boost_SYSTEM_LIBRARY}
763 ${Boost_FILESYSTEM_LIBRARY}
764 ${Boost_PROGRAM_OPTIONS_LIBRARY})
765 addDllCopyCommands(ArmnnConverter)