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/Tensor.hpp
218 include/armnn/TensorFwd.hpp
219 include/armnn/Types.hpp
220 include/armnn/TypesUtils.hpp
221 include/armnn/Utils.hpp
222 include/armnn/Version.hpp
223 src/armnn/layers/LayerCloneBase.hpp
224 src/armnn/layers/LayerWithParameters.hpp
225 src/armnn/layers/ActivationLayer.hpp
226 src/armnn/layers/ActivationLayer.cpp
227 src/armnn/layers/AdditionLayer.hpp
228 src/armnn/layers/AdditionLayer.cpp
229 src/armnn/layers/BatchNormalizationLayer.hpp
230 src/armnn/layers/BatchNormalizationLayer.cpp
231 src/armnn/layers/BatchToSpaceNdLayer.hpp
232 src/armnn/layers/BatchToSpaceNdLayer.cpp
233 src/armnn/layers/ConcatLayer.hpp
234 src/armnn/layers/ConcatLayer.cpp
235 src/armnn/layers/ConstantLayer.hpp
236 src/armnn/layers/ConstantLayer.cpp
237 src/armnn/layers/Convolution2dLayer.hpp
238 src/armnn/layers/Convolution2dLayer.cpp
239 src/armnn/layers/ConvertFp16ToFp32Layer.hpp
240 src/armnn/layers/ConvertFp16ToFp32Layer.cpp
241 src/armnn/layers/ConvertFp32ToFp16Layer.hpp
242 src/armnn/layers/ConvertFp32ToFp16Layer.cpp
243 src/armnn/layers/DebugLayer.hpp
244 src/armnn/layers/DebugLayer.cpp
245 src/armnn/layers/DepthwiseConvolution2dLayer.hpp
246 src/armnn/layers/DepthwiseConvolution2dLayer.cpp
247 src/armnn/layers/DequantizeLayer.hpp
248 src/armnn/layers/DequantizeLayer.cpp
249 src/armnn/layers/DetectionPostProcessLayer.hpp
250 src/armnn/layers/DetectionPostProcessLayer.cpp
251 src/armnn/layers/ElementwiseBaseLayer.hpp
252 src/armnn/layers/ElementwiseBaseLayer.cpp
253 src/armnn/layers/EqualLayer.hpp
254 src/armnn/layers/EqualLayer.cpp
255 src/armnn/layers/FakeQuantizationLayer.hpp
256 src/armnn/layers/FakeQuantizationLayer.cpp
257 src/armnn/layers/FloorLayer.hpp
258 src/armnn/layers/FloorLayer.cpp
259 src/armnn/layers/FullyConnectedLayer.hpp
260 src/armnn/layers/FullyConnectedLayer.cpp
261 src/armnn/layers/GatherLayer.cpp
262 src/armnn/layers/GatherLayer.hpp
263 src/armnn/layers/GreaterLayer.cpp
264 src/armnn/layers/GreaterLayer.hpp
265 src/armnn/layers/InputLayer.hpp
266 src/armnn/layers/InputLayer.cpp
267 src/armnn/layers/L2NormalizationLayer.hpp
268 src/armnn/layers/L2NormalizationLayer.cpp
269 src/armnn/layers/LstmLayer.cpp
270 src/armnn/layers/LstmLayer.hpp
271 src/armnn/layers/MaximumLayer.cpp
272 src/armnn/layers/MaximumLayer.hpp
273 src/armnn/layers/MeanLayer.hpp
274 src/armnn/layers/MeanLayer.cpp
275 src/armnn/layers/MemCopyLayer.hpp
276 src/armnn/layers/MemCopyLayer.cpp
277 src/armnn/layers/MergeLayer.hpp
278 src/armnn/layers/MergeLayer.cpp
279 src/armnn/layers/MinimumLayer.cpp
280 src/armnn/layers/MinimumLayer.hpp
281 src/armnn/layers/MultiplicationLayer.hpp
282 src/armnn/layers/MultiplicationLayer.cpp
283 src/armnn/layers/NormalizationLayer.hpp
284 src/armnn/layers/NormalizationLayer.cpp
285 src/armnn/layers/OutputLayer.hpp
286 src/armnn/layers/OutputLayer.cpp
287 src/armnn/layers/PadLayer.hpp
288 src/armnn/layers/PadLayer.cpp
289 src/armnn/layers/PermuteLayer.hpp
290 src/armnn/layers/PermuteLayer.cpp
291 src/armnn/layers/Pooling2dLayer.hpp
292 src/armnn/layers/Pooling2dLayer.cpp
293 src/armnn/layers/QuantizeLayer.cpp
294 src/armnn/layers/QuantizeLayer.hpp
295 src/armnn/layers/DivisionLayer.cpp
296 src/armnn/layers/DivisionLayer.hpp
297 src/armnn/layers/PreCompiledLayer.hpp
298 src/armnn/layers/PreCompiledLayer.cpp
299 src/armnn/layers/PreluLayer.hpp
300 src/armnn/layers/PreluLayer.cpp
301 src/armnn/layers/ReshapeLayer.hpp
302 src/armnn/layers/ReshapeLayer.cpp
303 src/armnn/layers/ResizeLayer.hpp
304 src/armnn/layers/ResizeLayer.cpp
305 src/armnn/layers/RsqrtLayer.cpp
306 src/armnn/layers/RsqrtLayer.hpp
307 src/armnn/layers/SoftmaxLayer.hpp
308 src/armnn/layers/SoftmaxLayer.cpp
309 src/armnn/layers/SpaceToBatchNdLayer.hpp
310 src/armnn/layers/SpaceToBatchNdLayer.cpp
311 src/armnn/layers/SpaceToDepthLayer.hpp
312 src/armnn/layers/SpaceToDepthLayer.cpp
313 src/armnn/layers/SplitterLayer.hpp
314 src/armnn/layers/SplitterLayer.cpp
315 src/armnn/layers/StackLayer.hpp
316 src/armnn/layers/StackLayer.cpp
317 src/armnn/layers/StridedSliceLayer.cpp
318 src/armnn/layers/StridedSliceLayer.hpp
319 src/armnn/layers/SubtractionLayer.cpp
320 src/armnn/layers/SubtractionLayer.hpp
321 src/armnn/layers/SwitchLayer.cpp
322 src/armnn/layers/SwitchLayer.hpp
323 src/armnn/layers/TransposeConvolution2dLayer.cpp
324 src/armnn/layers/TransposeConvolution2dLayer.hpp
325 src/armnn/BackendSettings.hpp
326 src/armnn/CompatibleTypes.hpp
327 src/armnn/Descriptors.cpp
328 src/armnn/DeviceSpec.hpp
329 src/armnn/DynamicQuantizationVisitor.cpp
330 src/armnn/DynamicQuantizationVisitor.hpp
331 src/armnn/Exceptions.cpp
332 src/armnn/ExecutionFrame.cpp
333 src/armnn/ExecutionFrame.hpp
336 src/armnn/IGraphObservable.hpp
337 src/armnn/Instrument.hpp
338 src/armnn/InternalTypes.cpp
339 src/armnn/InternalTypes.hpp
340 src/armnn/ISubgraphViewConverter.hpp
341 src/armnn/JsonPrinter.cpp
342 src/armnn/JsonPrinter.hpp
344 src/armnn/LayerFwd.hpp
346 src/armnn/LayersFwd.hpp
347 src/armnn/LayerSupportCommon.hpp
348 src/armnn/LayerSupport.cpp
349 src/armnn/LoadedNetwork.cpp
350 src/armnn/LoadedNetwork.hpp
351 src/armnn/Network.cpp
352 src/armnn/Network.hpp
353 src/armnn/NetworkQuantizationScheme.hpp
354 src/armnn/NetworkQuantizer.cpp
355 src/armnn/NetworkQuantizer.hpp
356 src/armnn/NetworkQuantizerUtils.cpp
357 src/armnn/NetworkQuantizerUtils.hpp
358 src/armnn/NetworkUtils.cpp
359 src/armnn/NetworkUtils.hpp
360 src/armnn/Observable.cpp
361 src/armnn/Observable.hpp
362 src/armnn/Optimizer.cpp
363 src/armnn/Optimizer.hpp
364 src/armnn/OverrideInputRangeVisitor.cpp
365 src/armnn/OverrideInputRangeVisitor.hpp
366 src/armnn/Profiling.cpp
367 src/armnn/ProfilingEvent.cpp
368 src/armnn/ProfilingEvent.hpp
369 src/armnn/Profiling.hpp
370 src/armnn/QuantizerVisitor.cpp
371 src/armnn/QuantizerVisitor.hpp
372 src/armnn/Runtime.cpp
373 src/armnn/Runtime.hpp
374 src/armnn/RangeTracker.cpp
375 src/armnn/RangeTracker.hpp
376 src/armnn/ResolveType.hpp
377 src/armnn/SerializeLayerParameters.cpp
378 src/armnn/SerializeLayerParameters.hpp
379 src/armnn/StaticRangeVisitor.cpp
380 src/armnn/StaticRangeVisitor.hpp
381 src/armnn/SubgraphView.cpp
382 src/armnn/SubgraphView.hpp
383 src/armnn/SubgraphViewSelector.cpp
384 src/armnn/SubgraphViewSelector.hpp
386 src/armnn/TypesUtils.cpp
388 src/armnn/WallClockTimer.cpp
389 src/armnn/WallClockTimer.hpp
390 src/armnn/optimizations/AddDebug.hpp
391 src/armnn/optimizations/All.hpp
392 src/armnn/optimizations/ConvertConstants.hpp
393 src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
394 src/armnn/optimizations/FoldPadIntoConvolution2d.hpp
395 src/armnn/optimizations/MovePermuteUp.hpp
396 src/armnn/optimizations/Optimization.hpp
397 src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
398 src/armnn/optimizations/OptimizeInverseConversions.hpp
399 src/armnn/optimizations/OptimizeInversePermutes.hpp
400 src/armnn/optimizations/PermuteAsReshape.hpp
401 src/armnn/optimizations/SquashEqualSiblings.hpp
402 third-party/half/half.hpp
405 # Files used for Streamline-based profiling backend
406 if(PROFILING_BACKEND_STREAMLINE)
407 list(APPEND armnn_sources
408 ${GATOR_ROOT}/annotate/streamline_annotate.h
409 ${GATOR_ROOT}/annotate/streamline_annotate.c)
412 # the backends under src/backends extend the list of
413 # object libs armnn to include in the build
414 include(src/backends/backends.cmake)
415 foreach(lib ${armnnLibraries})
416 message("Adding object library dependency to armnn: ${lib}")
417 list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
420 add_library_ex(armnn SHARED ${armnn_sources})
422 target_include_directories(armnn PRIVATE src/armnn)
423 target_include_directories(armnn PRIVATE src/armnnUtils)
424 target_include_directories(armnn PRIVATE src/backends)
426 target_link_libraries(armnn armnnUtils)
428 target_link_libraries(armnn ${CMAKE_DL_LIBS})
430 install(TARGETS armnn
431 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
432 if(BUILD_CAFFE_PARSER)
433 install(TARGETS armnnCaffeParser
434 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
436 if(BUILD_ONNX_PARSER)
437 install(TARGETS armnnOnnxParser
438 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
441 install(TARGETS armnnTfParser
442 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
445 install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
447 target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
449 if(ARMCOMPUTENEON OR ARMCOMPUTECL)
450 target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
453 target_link_libraries(armnn ${OPENCL_LIBRARIES})
456 if(PROFILING_BACKEND_STREAMLINE)
457 target_link_libraries(armnn pthread)
461 set(unittest_sources)
462 list(APPEND unittest_sources
463 src/armnn/test/ConstTensorLayerVisitor.hpp
464 src/armnn/test/ConstTensorLayerVisitor.cpp
465 src/armnn/test/CreateWorkload.hpp
466 src/armnn/test/CsvReaderTest.cpp
467 src/armnn/test/DebugCallbackTest.cpp
468 src/armnn/test/EndToEndTest.cpp
469 src/armnn/test/ExecutionFrameTest.cpp
470 src/armnn/test/FloatingPointConverterTest.cpp
471 src/armnn/test/FlowControl.cpp
472 src/armnn/test/GraphTests.cpp
473 src/armnn/test/GraphUtils.cpp
474 src/armnn/test/GraphUtils.hpp
475 src/armnn/test/InstrumentTests.cpp
476 src/armnn/test/InferOutputTests.cpp
477 src/armnn/test/InferOutputTests.hpp
478 src/armnn/test/ModelAccuracyCheckerTest.cpp
479 src/armnn/test/NetworkTests.cpp
480 src/armnn/test/ObservableTest.cpp
481 src/armnn/test/OptimizerTests.cpp
482 src/armnn/test/OptionalTest.cpp
483 src/armnn/test/ProfilerTests.cpp
484 src/armnn/test/ProfilingEventTest.cpp
485 src/armnn/test/QuantizerTest.cpp
486 src/armnn/test/RuntimeTests.cpp
487 src/armnn/test/RuntimeTests.hpp
488 src/armnn/test/SubgraphViewTests.cpp
489 src/armnn/test/TensorHandleStrategyTest.cpp
490 src/armnn/test/TensorHelpers.hpp
491 src/armnn/test/TensorTest.cpp
492 src/armnn/test/TestInputOutputLayerVisitor.cpp
493 src/armnn/test/TestInputOutputLayerVisitor.hpp
494 src/armnn/test/TestLayerVisitor.cpp
495 src/armnn/test/TestLayerVisitor.hpp
496 src/armnn/test/TestNameOnlyLayerVisitor.cpp
497 src/armnn/test/TestNameOnlyLayerVisitor.hpp
498 src/armnn/test/TestNameAndDescriptorLayerVisitor.hpp
499 src/armnn/test/TestNameAndDescriptorLayerVisitor.cpp
500 src/armnn/test/TestUtils.cpp
501 src/armnn/test/TestUtils.hpp
502 src/armnn/test/UnitTests.cpp
503 src/armnn/test/UnitTests.hpp
504 src/armnn/test/UtilsTests.cpp
505 src/armnnUtils/test/PrototxtConversionsTest.cpp
506 src/armnnUtils/test/ParserHelperTest.cpp
510 list(APPEND unittest_sources
511 src/armnnTfParser/test/Activations.cpp
512 src/armnnTfParser/test/Addition.cpp
513 src/armnnTfParser/test/AddN.cpp
514 src/armnnTfParser/test/Assert.cpp
515 src/armnnTfParser/test/BiasAdd.cpp
516 src/armnnTfParser/test/BroadcastForAdd.cpp
517 src/armnnTfParser/test/Convolution2d.cpp
518 src/armnnTfParser/test/Concat.cpp
519 src/armnnTfParser/test/ConcatOfConcats.cpp
520 src/armnnTfParser/test/DepthwiseConvolution2d.cpp
521 src/armnnTfParser/test/Equal.cpp
522 src/armnnTfParser/test/ExpandDims.cpp
523 src/armnnTfParser/test/FusedBatchNorm.cpp
524 src/armnnTfParser/test/Gather.cpp
525 src/armnnTfParser/test/Greater.cpp
526 src/armnnTfParser/test/Identity.cpp
527 src/armnnTfParser/test/LocalResponseNormalization.cpp
528 src/armnnTfParser/test/Maximum.cpp
529 src/armnnTfParser/test/MaximumForLeakyRelu.cpp
530 src/armnnTfParser/test/Mean.cpp
531 src/armnnTfParser/test/Minimum.cpp
532 src/armnnTfParser/test/Multiplication.cpp
533 src/armnnTfParser/test/Pad.cpp
534 src/armnnTfParser/test/MultiOutput.cpp
535 src/armnnTfParser/test/PassThru.cpp
536 src/armnnTfParser/test/Pooling.cpp
537 src/armnnTfParser/test/RealDiv.cpp
538 src/armnnTfParser/test/Reshape.cpp
539 src/armnnTfParser/test/ResizeBilinear.cpp
540 src/armnnTfParser/test/Rsqrt.cpp
541 src/armnnTfParser/test/Shape.cpp
542 src/armnnTfParser/test/Softmax.cpp
543 src/armnnTfParser/test/TestDependencies.cpp
544 src/armnnTfParser/test/FullyConnected.cpp
545 src/armnnTfParser/test/Constant.cpp
546 src/armnnTfParser/test/TestMultiInputsOutputs.cpp
547 src/armnnTfParser/test/Split.cpp
548 src/armnnTfParser/test/Squeeze.cpp
549 src/armnnTfParser/test/Sub.cpp
553 if(BUILD_TF_LITE_PARSER)
555 list(APPEND unittest_sources
556 src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp
557 src/armnnTfLiteParser/test/Activations.cpp
558 src/armnnTfLiteParser/test/Addition.cpp
559 src/armnnTfLiteParser/test/AvgPool2D.cpp
560 src/armnnTfLiteParser/test/BatchToSpaceND.cpp
561 src/armnnTfLiteParser/test/Concatenation.cpp
562 src/armnnTfLiteParser/test/Constant.cpp
563 src/armnnTfLiteParser/test/Conv2D.cpp
564 src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
565 src/armnnTfLiteParser/test/DetectionPostProcess.cpp
566 src/armnnTfLiteParser/test/FullyConnected.cpp
567 src/armnnTfLiteParser/test/Maximum.cpp
568 src/armnnTfLiteParser/test/MaxPool2D.cpp
569 src/armnnTfLiteParser/test/Mean.cpp
570 src/armnnTfLiteParser/test/Minimum.cpp
571 src/armnnTfLiteParser/test/Multiplication.cpp
572 src/armnnTfLiteParser/test/Pad.cpp
573 src/armnnTfLiteParser/test/Reshape.cpp
574 src/armnnTfLiteParser/test/ResizeBilinear.cpp
575 src/armnnTfLiteParser/test/Softmax.cpp
576 src/armnnTfLiteParser/test/SpaceToBatchND.cpp
577 src/armnnTfLiteParser/test/Split.cpp
578 src/armnnTfLiteParser/test/Squeeze.cpp
579 src/armnnTfLiteParser/test/StridedSlice.cpp
580 src/armnnTfLiteParser/test/Sub.cpp
581 src/armnnTfLiteParser/test/Unpack.cpp
582 src/armnnTfLiteParser/test/LoadModel.cpp
583 src/armnnTfLiteParser/test/GetBuffer.cpp
584 src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
585 src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
586 src/armnnTfLiteParser/test/GetTensorIds.cpp
587 src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
588 src/armnnTfLiteParser/test/GetInputsOutputs.cpp
589 src/armnnTfLiteParser/test/Schema.s
591 set_source_files_properties(src/armnnTfLiteParser/test/Schema.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
594 if(BUILD_CAFFE_PARSER)
595 list(APPEND unittest_sources
596 src/armnnCaffeParser/test/TestAdd.cpp
597 src/armnnCaffeParser/test/TestConcat.cpp
598 src/armnnCaffeParser/test/TestConvolution.cpp
599 src/armnnCaffeParser/test/TestDropout.cpp
600 src/armnnCaffeParser/test/TestInputs.cpp
601 src/armnnCaffeParser/test/TestMul.cpp
602 src/armnnCaffeParser/test/TestPooling2d.cpp
603 src/armnnCaffeParser/test/TestInPlace.cpp
604 src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
605 src/armnnCaffeParser/test/TestSplit.cpp
609 if(BUILD_ARMNN_QUANTIZER)
610 list(APPEND unittest_sources
611 src/armnnQuantizer/test/QuantizationDataSetTests.cpp
615 if(BUILD_ARMNN_SERIALIZER)
617 list(APPEND unittest_sources
618 src/armnnSerializer/test/ActivationSerializationTests.cpp
619 src/armnnSerializer/test/SerializerTests.cpp
620 src/armnnDeserializer/test/DeserializeActivation.cpp
621 src/armnnDeserializer/test/DeserializeAdd.cpp
622 src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp
623 src/armnnDeserializer/test/DeserializeBatchNormalization.cpp
624 src/armnnDeserializer/test/DeserializeConstant.cpp
625 src/armnnDeserializer/test/DeserializeConvolution2d.cpp
626 src/armnnDeserializer/test/DeserializeDivision.cpp
627 src/armnnDeserializer/test/DeserializeEqual.cpp
628 src/armnnDeserializer/test/DeserializeFloor.cpp
629 src/armnnDeserializer/test/DeserializeFullyConnected.cpp
630 src/armnnDeserializer/test/DeserializeGather.cpp
631 src/armnnDeserializer/test/DeserializeGreater.cpp
632 src/armnnDeserializer/test/DeserializeL2Normalization.cpp
633 src/armnnDeserializer/test/DeserializeMean.cpp
634 src/armnnDeserializer/test/DeserializeMultiplication.cpp
635 src/armnnDeserializer/test/DeserializeNormalization.cpp
636 src/armnnDeserializer/test/DeserializePad.cpp
637 src/armnnDeserializer/test/DeserializePermute.cpp
638 src/armnnDeserializer/test/DeserializePooling2d.cpp
639 src/armnnDeserializer/test/DeserializeReshape.cpp
640 src/armnnDeserializer/test/DeserializeResizeBilinear.cpp
641 src/armnnDeserializer/test/DeserializeRsqrt.cpp
642 src/armnnDeserializer/test/DeserializeSpaceToBatchNd.cpp
643 src/armnnDeserializer/test/DeserializeStridedSlice.cpp
644 src/armnnDeserializer/test/DeserializeSubtraction.cpp
645 src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp
646 src/armnnDeserializer/test/SchemaSerialize.s
648 set_source_files_properties(src/armnnDeserializer/test/SchemaSerialize.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
651 if(BUILD_ONNX_PARSER)
652 list(APPEND unittest_sources
653 src/armnnOnnxParser/test/Constructor.cpp
654 src/armnnOnnxParser/test/CreateNetwork.cpp
655 src/armnnOnnxParser/test/ProtoxtFixture.cpp
656 src/armnnOnnxParser/test/Const.cpp
657 src/armnnOnnxParser/test/Pooling.cpp
658 src/armnnOnnxParser/test/Reshape.cpp
659 src/armnnOnnxParser/test/Relu.cpp
660 src/armnnOnnxParser/test/Conv2D.cpp
661 src/armnnOnnxParser/test/Addition.cpp
662 src/armnnOnnxParser/test/FullyConnected.cpp
663 src/armnnOnnxParser/test/GetInputsOutputs.cpp
664 src/armnnOnnxParser/test/BatchNorm.cpp
665 src/armnnOnnxParser/test/DepthConv.cpp
669 foreach(lib ${armnnUnitTestLibraries})
670 message("Adding object library dependency to UnitTests: ${lib}")
671 list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
674 add_executable(UnitTests ${unittest_sources})
675 target_include_directories(UnitTests PRIVATE src/armnn)
676 target_include_directories(UnitTests PRIVATE src/armnnUtils)
677 target_include_directories(UnitTests PRIVATE src/backends)
680 if(HEAP_PROFILING OR LEAK_CHECKING)
681 message("Valgrind is disabled for heap profiling and leak checking builds.")
683 # Valgrind works with gperftools version number <= 2.4
684 target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
688 target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
689 target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
691 target_link_libraries(UnitTests armnn)
692 target_link_libraries(UnitTests armnnUtils)
695 target_link_libraries(UnitTests armnnTfParser)
698 if(BUILD_TF_LITE_PARSER)
699 target_include_directories(UnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
700 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
701 target_link_libraries(UnitTests armnnTfLiteParser)
704 if(BUILD_CAFFE_PARSER)
705 target_link_libraries(UnitTests armnnCaffeParser)
708 if(BUILD_ARMNN_SERIALIZER)
709 target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer)
710 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
711 target_link_libraries(UnitTests armnnSerializer)
714 if(BUILD_ARMNN_QUANTIZER)
715 target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnQuantizer)
716 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
717 target_link_libraries(UnitTests armnnQuantizer armnnSerializer)
720 if(BUILD_ONNX_PARSER)
721 target_link_libraries(UnitTests armnnOnnxParser)
724 addDllCopyCommands(UnitTests)
727 if (BUILD_ARMNN_SERIALIZER AND (BUILD_TF_PARSER OR BUILD_TF_LITE_PARSER OR BUILD_ONNX_PARSER OR BUILD_CAFFE_PARSER))
728 set(ArmnnConverter_sources
729 src/armnnConverter/ArmnnConverter.cpp)
731 add_executable_ex(ArmnnConverter ${ArmnnConverter_sources})
732 target_include_directories(ArmnnConverter PRIVATE src/armnn)
733 target_include_directories(ArmnnConverter PRIVATE src/armnnUtils)
735 if(BUILD_CAFFE_PARSER)
736 target_link_libraries(ArmnnConverter armnnCaffeParser)
739 if(BUILD_ONNX_PARSER)
740 target_link_libraries(ArmnnConverter armnnOnnxParser)
744 target_link_libraries(ArmnnConverter armnnTfParser)
747 if(BUILD_TF_LITE_PARSER)
748 target_link_libraries(ArmnnConverter armnnTfLiteParser)
751 target_link_libraries(ArmnnConverter armnnSerializer)
753 target_link_libraries(ArmnnConverter armnn)
754 target_link_libraries(ArmnnConverter ${CMAKE_THREAD_LIBS_INIT})
755 target_link_libraries(ArmnnConverter
756 ${Boost_SYSTEM_LIBRARY}
757 ${Boost_FILESYSTEM_LIBRARY}
758 ${Boost_PROGRAM_OPTIONS_LIBRARY})
759 addDllCopyCommands(ArmnnConverter)