blob: a00a44a6cee361bf14a396be78e9847624e4f958 [file] [log] [blame]
David Beck10b4dfd2018-09-19 12:03:20 +01001#
2# Copyright © 2017 Arm Ltd. All rights reserved.
3# SPDX-License-Identifier: MIT
4#
telsoa014fcda012018-03-09 14:13:49 +00005cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake.
6project(armnn)
7
8set(additional_cmake_files)
9list(APPEND additional_cmake_files
10 cmake/Utils.cmake
11 cmake/GlobalConfig.cmake
12 cmake/AddDllCopyCommands.cmake)
13
14foreach(cmake_file ${additional_cmake_files})
15 include(${cmake_file})
16endforeach()
17
18if (BUILD_TESTS)
19 add_subdirectory(tests)
20endif()
21
telsoa01c577f2c2018-08-31 09:22:23 +010022add_subdirectory(samples)
23
telsoa014fcda012018-03-09 14:13:49 +000024# Include the additional cmake files in their own target so that they will appear nicely in IDEs
25add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files})
26
27set(armnnUtils_sources)
28list(APPEND armnnUtils_sources
29 src/armnnUtils/GraphTopologicalSort.hpp
arovir01616e7752018-10-01 17:08:59 +010030 src/armnnUtils/Half.hpp
telsoa014fcda012018-03-09 14:13:49 +000031 src/armnnUtils/Logging.hpp
telsoa014fcda012018-03-09 14:13:49 +000032 src/armnnUtils/Logging.cpp
Matteo Martincigh21350152018-11-28 16:22:22 +000033 src/armnnUtils/Permute.hpp
telsoa014fcda012018-03-09 14:13:49 +000034 src/armnnUtils/Permute.cpp
Matteo Martincigh21350152018-11-28 16:22:22 +000035 src/armnnUtils/DataLayoutIndexed.cpp
36 src/armnnUtils/DataLayoutIndexed.hpp
surmeh01bceff2f2018-03-29 16:29:27 +010037 src/armnnUtils/DotSerializer.cpp
38 src/armnnUtils/DotSerializer.hpp
surmeh013537c2c2018-05-18 16:31:43 +010039 src/armnnUtils/HeapProfiling.cpp
40 src/armnnUtils/HeapProfiling.hpp
41 src/armnnUtils/LeakChecking.cpp
42 src/armnnUtils/LeakChecking.hpp
telsoa01c577f2c2018-08-31 09:22:23 +010043 src/armnnUtils/CsvReader.cpp
44 src/armnnUtils/CsvReader.hpp
45 src/armnnUtils/FloatingPointConverter.cpp
46 src/armnnUtils/FloatingPointConverter.hpp
47 src/armnnUtils/VerificationHelpers.hpp
48 src/armnnUtils/VerificationHelpers.cpp
Sadik Armagan479045b2018-10-01 11:51:37 +010049 src/armnnUtils/ParserHelper.hpp
50 src/armnnUtils/ParserHelper.cpp
Nina Drozdd41b2592018-11-19 13:03:36 +000051 src/armnnUtils/TensorUtils.hpp
52 src/armnnUtils/TensorUtils.cpp
telsoa014fcda012018-03-09 14:13:49 +000053 )
surmeh01bceff2f2018-03-29 16:29:27 +010054if(BUILD_TF_PARSER OR BUILD_CAFFE_PARSER)
telsoa014fcda012018-03-09 14:13:49 +000055 list(APPEND armnnUtils_sources
56 src/armnnUtils/ParserPrototxtFixture.hpp
57 )
58endif()
59
60add_library_ex(armnnUtils STATIC ${armnnUtils_sources})
61
62if(BUILD_CAFFE_PARSER)
63 # ArmNN Parser source files required for all build options
64 set(armnn_caffe_parser_sources)
65 list(APPEND armnn_caffe_parser_sources
66 include/armnnCaffeParser/ICaffeParser.hpp
telsoa01c577f2c2018-08-31 09:22:23 +010067 src/armnnCaffeParser/RecordByRecordCaffeParser.hpp
68 src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
telsoa014fcda012018-03-09 14:13:49 +000069 src/armnnCaffeParser/CaffeParser.hpp
70 src/armnnCaffeParser/CaffeParser.cpp
71 ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc
72 )
73 # The generated Caffe protobuf .cc file is not warning clean and we can't fix them.
74 if(COMPILER_IS_GNU_LIKE)
75 set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
76 endif()
77
78 add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})
79 set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")
80
81 target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)
82
83 target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
84
85 target_link_libraries(armnnCaffeParser armnn)
86 target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})
telsoa01c577f2c2018-08-31 09:22:23 +010087
88endif()
89
90if(BUILD_ONNX_PARSER)
91 set(armnn_onnx_parser_sources)
92 list(APPEND armnn_onnx_parser_sources
93 include/armnnOnnxParser/IOnnxParser.hpp
94 src/armnnOnnxParser/OnnxParser.hpp
95 src/armnnOnnxParser/OnnxParser.cpp
96 ${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc
97 )
98 # The generated onnx protobuf .cc files are not warning clean and we can't fix them.
99 if(COMPILER_IS_GNU_LIKE)
100 set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
101 endif()
102
103 add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})
104
105 target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)
106
107 target_link_libraries(armnnOnnxParser armnn)
108
109 # Protobuf
110 target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})
telsoa014fcda012018-03-09 14:13:49 +0000111endif()
112
surmeh01bceff2f2018-03-29 16:29:27 +0100113if(BUILD_TF_PARSER)
114 set(armnn_tf_parser_sources)
115 list(APPEND armnn_tf_parser_sources
116 include/armnnTfParser/ITfParser.hpp
117 src/armnnTfParser/TfParser.hpp
118 src/armnnTfParser/TfParser.cpp
119 ${TF_PROTOBUFS}
120 )
121 # The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.
122 if(COMPILER_IS_GNU_LIKE)
123 set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
124 endif()
125
126 add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})
127
128 target_include_directories(armnnTfParser PRIVATE src/armnnUtils)
129
130 target_link_libraries(armnnTfParser armnn)
131
132 # Protobuf (use the specific version tensorflow wants)
133 target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})
134endif()
135
telsoa01c577f2c2018-08-31 09:22:23 +0100136if(BUILD_TF_LITE_PARSER)
137 set(armnn_tf_lite_parser_sources)
138 list(APPEND armnn_tf_lite_parser_sources
139 include/armnnTfLiteParser/ITfLiteParser.hpp
140 src/armnnTfLiteParser/TfLiteParser.hpp
141 src/armnnTfLiteParser/TfLiteParser.cpp
Matteo Martincigh21350152018-11-28 16:22:22 +0000142 )
telsoa01c577f2c2018-08-31 09:22:23 +0100143
Matteo Martincigh21350152018-11-28 16:22:22 +0000144 add_library_ex(armnnTfLiteParser SHARED ${armnn_tf_lite_parser_sources})
telsoa01c577f2c2018-08-31 09:22:23 +0100145
Matteo Martincigh21350152018-11-28 16:22:22 +0000146 target_include_directories(armnnTfLiteParser PRIVATE src/armnn)
147 target_include_directories(armnnTfLiteParser PRIVATE src/armnnUtils)
telsoa01c577f2c2018-08-31 09:22:23 +0100148
Matteo Martincigh21350152018-11-28 16:22:22 +0000149 target_link_libraries(armnnTfLiteParser ${Boost_FILESYSTEM_LIBRARY} ${Boost_THREAD_LIBRARY})
150 target_link_libraries(armnnTfLiteParser armnn ${FLATBUFFERS_LIBRARY})
telsoa01c577f2c2018-08-31 09:22:23 +0100151endif()
152
telsoa014fcda012018-03-09 14:13:49 +0000153list(APPEND armnn_sources
154 include/armnn/ArmNN.hpp
David Beck9df2d952018-10-10 15:11:44 +0100155 include/armnn/BackendId.hpp
telsoa014fcda012018-03-09 14:13:49 +0000156 include/armnn/Descriptors.hpp
157 include/armnn/DescriptorsFwd.hpp
David Beck9df2d952018-10-10 15:11:44 +0100158 include/armnn/Exceptions.hpp
David Becke97c6e02018-10-03 13:09:28 +0100159 include/armnn/ILayerSupport.hpp
telsoa014fcda012018-03-09 14:13:49 +0000160 include/armnn/INetwork.hpp
David Beck9df2d952018-10-10 15:11:44 +0100161 include/armnn/IProfiler.hpp
162 include/armnn/IRuntime.hpp
163 include/armnn/LayerSupport.hpp
164 include/armnn/LstmParams.hpp
165 include/armnn/NetworkFwd.hpp
David Beck5eec11d2018-10-04 15:43:17 +0100166 include/armnn/Optional.hpp
telsoa014fcda012018-03-09 14:13:49 +0000167 include/armnn/Tensor.hpp
168 include/armnn/TensorFwd.hpp
169 include/armnn/Types.hpp
170 include/armnn/TypesUtils.hpp
171 include/armnn/Utils.hpp
telsoa014fcda012018-03-09 14:13:49 +0000172 include/armnn/Version.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100173 src/armnn/layers/LayerCloneBase.hpp
174 src/armnn/layers/LayerWithParameters.hpp
175 src/armnn/layers/ActivationLayer.hpp
176 src/armnn/layers/ActivationLayer.cpp
177 src/armnn/layers/AdditionLayer.hpp
178 src/armnn/layers/AdditionLayer.cpp
179 src/armnn/layers/BatchNormalizationLayer.hpp
180 src/armnn/layers/BatchNormalizationLayer.cpp
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000181 src/armnn/layers/BatchToSpaceNdLayer.hpp
182 src/armnn/layers/BatchToSpaceNdLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100183 src/armnn/layers/ConstantLayer.hpp
184 src/armnn/layers/ConstantLayer.cpp
185 src/armnn/layers/Convolution2dLayer.hpp
186 src/armnn/layers/Convolution2dLayer.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100187 src/armnn/layers/ConvertFp16ToFp32Layer.hpp
188 src/armnn/layers/ConvertFp16ToFp32Layer.cpp
189 src/armnn/layers/ConvertFp32ToFp16Layer.hpp
190 src/armnn/layers/ConvertFp32ToFp16Layer.cpp
Nattapat Chaimanowonga9a1cf12018-12-03 16:06:49 +0000191 src/armnn/layers/DebugLayer.hpp
192 src/armnn/layers/DebugLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100193 src/armnn/layers/DepthwiseConvolution2dLayer.hpp
194 src/armnn/layers/DepthwiseConvolution2dLayer.cpp
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000195 src/armnn/layers/ElementwiseBaseLayer.hpp
196 src/armnn/layers/ElementwiseBaseLayer.cpp
FrancisMurtagh20995952018-12-17 12:11:36 +0000197 src/armnn/layers/EqualLayer.hpp
198 src/armnn/layers/EqualLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100199 src/armnn/layers/FakeQuantizationLayer.hpp
200 src/armnn/layers/FakeQuantizationLayer.cpp
201 src/armnn/layers/FloorLayer.hpp
202 src/armnn/layers/FloorLayer.cpp
203 src/armnn/layers/FullyConnectedLayer.hpp
204 src/armnn/layers/FullyConnectedLayer.cpp
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000205 src/armnn/layers/GreaterLayer.cpp
206 src/armnn/layers/GreaterLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100207 src/armnn/layers/InputLayer.hpp
208 src/armnn/layers/InputLayer.cpp
209 src/armnn/layers/L2NormalizationLayer.hpp
210 src/armnn/layers/L2NormalizationLayer.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100211 src/armnn/layers/LstmLayer.cpp
212 src/armnn/layers/LstmLayer.hpp
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000213 src/armnn/layers/MaximumLayer.cpp
214 src/armnn/layers/MaximumLayer.hpp
narpra0132b90462018-09-13 11:07:48 +0100215 src/armnn/layers/MeanLayer.hpp
216 src/armnn/layers/MeanLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100217 src/armnn/layers/MemCopyLayer.hpp
218 src/armnn/layers/MemCopyLayer.cpp
219 src/armnn/layers/MergerLayer.hpp
220 src/armnn/layers/MergerLayer.cpp
kevmay0190539692018-11-29 08:40:19 +0000221 src/armnn/layers/MinimumLayer.cpp
222 src/armnn/layers/MinimumLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100223 src/armnn/layers/MultiplicationLayer.hpp
224 src/armnn/layers/MultiplicationLayer.cpp
225 src/armnn/layers/NormalizationLayer.hpp
226 src/armnn/layers/NormalizationLayer.cpp
227 src/armnn/layers/OutputLayer.hpp
228 src/armnn/layers/OutputLayer.cpp
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100229 src/armnn/layers/PadLayer.hpp
230 src/armnn/layers/PadLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100231 src/armnn/layers/PermuteLayer.hpp
232 src/armnn/layers/PermuteLayer.cpp
233 src/armnn/layers/Pooling2dLayer.hpp
234 src/armnn/layers/Pooling2dLayer.cpp
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100235 src/armnn/layers/DivisionLayer.cpp
236 src/armnn/layers/DivisionLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100237 src/armnn/layers/ReshapeLayer.hpp
238 src/armnn/layers/ReshapeLayer.cpp
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +0000239 src/armnn/layers/SpaceToBatchNdLayer.hpp
240 src/armnn/layers/SpaceToBatchNdLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100241 src/armnn/layers/ResizeBilinearLayer.hpp
242 src/armnn/layers/ResizeBilinearLayer.cpp
243 src/armnn/layers/SoftmaxLayer.hpp
244 src/armnn/layers/SoftmaxLayer.cpp
245 src/armnn/layers/SplitterLayer.hpp
246 src/armnn/layers/SplitterLayer.cpp
Conor Kennedy430b5d82018-11-14 15:28:28 +0000247 src/armnn/layers/StridedSliceLayer.cpp
248 src/armnn/layers/StridedSliceLayer.hpp
David Beckc2044fe2018-09-05 15:00:38 +0100249 src/armnn/layers/SubtractionLayer.cpp
250 src/armnn/layers/SubtractionLayer.hpp
telsoa014fcda012018-03-09 14:13:49 +0000251 src/armnn/Descriptors.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100252 src/armnn/DeviceSpec.hpp
telsoa014fcda012018-03-09 14:13:49 +0000253 src/armnn/Exceptions.cpp
telsoa014fcda012018-03-09 14:13:49 +0000254 src/armnn/Graph.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100255 src/armnn/Graph.hpp
256 src/armnn/IGraphObservable.hpp
257 src/armnn/Instrument.hpp
258 src/armnn/InternalTypes.cpp
259 src/armnn/InternalTypes.hpp
260 src/armnn/JsonPrinter.cpp
261 src/armnn/JsonPrinter.hpp
262 src/armnn/Layer.cpp
263 src/armnn/LayerFwd.hpp
264 src/armnn/Layer.hpp
265 src/armnn/LayersFwd.hpp
266 src/armnn/LayerSupportCommon.hpp
267 src/armnn/LayerSupport.cpp
268 src/armnn/LoadedNetwork.cpp
269 src/armnn/LoadedNetwork.hpp
telsoa014fcda012018-03-09 14:13:49 +0000270 src/armnn/Network.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100271 src/armnn/Network.hpp
Aron Virginas-Tar5fdfe822018-11-21 13:14:42 +0000272 src/armnn/NetworkUtils.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100273 src/armnn/NetworkUtils.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100274 src/armnn/Observable.cpp
275 src/armnn/Observable.hpp
276 src/armnn/Optimizer.cpp
277 src/armnn/Optimizer.hpp
278 src/armnn/Profiling.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100279 src/armnn/ProfilingEvent.cpp
280 src/armnn/ProfilingEvent.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100281 src/armnn/Profiling.hpp
282 src/armnn/Runtime.cpp
283 src/armnn/Runtime.hpp
284 src/armnn/SerializeLayerParameters.cpp
285 src/armnn/SerializeLayerParameters.hpp
286 src/armnn/SubGraph.cpp
287 src/armnn/SubGraph.hpp
288 src/armnn/SubGraphSelector.cpp
289 src/armnn/SubGraphSelector.hpp
telsoa014fcda012018-03-09 14:13:49 +0000290 src/armnn/Tensor.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100291 src/armnn/TypeUtils.hpp
telsoa014fcda012018-03-09 14:13:49 +0000292 src/armnn/Utils.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100293 src/armnn/WallClockTimer.cpp
294 src/armnn/WallClockTimer.hpp
telsoa014fcda012018-03-09 14:13:49 +0000295 src/armnn/optimizations/All.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100296 src/armnn/optimizations/ConvertConstants.hpp
telsoa014fcda012018-03-09 14:13:49 +0000297 src/armnn/optimizations/MovePermuteUp.hpp
298 src/armnn/optimizations/Optimization.hpp
299 src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
300 src/armnn/optimizations/OptimizeInversePermutes.hpp
301 src/armnn/optimizations/PermuteAsReshape.hpp
302 src/armnn/optimizations/SquashEqualSiblings.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100303 src/armnn/optimizations/OptimizeInverseConversions.hpp
304 src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
keidav01738c2e62018-12-11 16:14:20 +0000305 src/armnn/optimizations/AddDebug.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100306 third-party/half/half.hpp
telsoa014fcda012018-03-09 14:13:49 +0000307 )
308
telsoa014fcda012018-03-09 14:13:49 +0000309# Files used for Streamline-based profiling backend
310if(PROFILING_BACKEND_STREAMLINE)
311 list(APPEND armnn_sources
312 ${GATOR_ROOT}/annotate/streamline_annotate.h
313 ${GATOR_ROOT}/annotate/streamline_annotate.c)
314endif()
315
David Beck10b4dfd2018-09-19 12:03:20 +0100316# the backends under src/backends extend the list of
David Beck3cc9a622018-10-12 10:38:31 +0100317# object libs armnn to include in the build
David Beck10b4dfd2018-09-19 12:03:20 +0100318include(src/backends/backends.cmake)
David Beck3cc9a622018-10-12 10:38:31 +0100319foreach(lib ${armnnLibraries})
320 message("Adding object library dependency to armnn: ${lib}")
321 list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
322endforeach()
323
324add_library_ex(armnn SHARED ${armnn_sources})
David Beck10b4dfd2018-09-19 12:03:20 +0100325
telsoa014fcda012018-03-09 14:13:49 +0000326target_include_directories(armnn PRIVATE src/armnn)
327target_include_directories(armnn PRIVATE src/armnnUtils)
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000328target_include_directories(armnn PRIVATE src/backends)
David Beck3cc9a622018-10-12 10:38:31 +0100329
330target_link_libraries(armnn armnnUtils)
331
telsoa014fcda012018-03-09 14:13:49 +0000332target_link_libraries(armnn ${CMAKE_DL_LIBS})
333
334install(TARGETS armnn DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
335if(BUILD_CAFFE_PARSER)
336 install(TARGETS armnnCaffeParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
337endif()
telsoa01c577f2c2018-08-31 09:22:23 +0100338if(BUILD_ONNX_PARSER)
339 install(TARGETS armnnOnnxParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
340endif()
surmeh01bceff2f2018-03-29 16:29:27 +0100341if(BUILD_TF_PARSER)
342 install(TARGETS armnnTfParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
343endif()
telsoa01c577f2c2018-08-31 09:22:23 +0100344if(BUILD_TF_LITE_PARSER)
345 install(TARGETS armnnTfLiteParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
346endif()
telsoa014fcda012018-03-09 14:13:49 +0000347install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_PREFIX}/include)
348
349target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
350
351if(ARMCOMPUTENEON OR ARMCOMPUTECL)
352 target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
353endif()
354if(ARMCOMPUTECL)
355 target_link_libraries(armnn ${OPENCL_LIBRARIES})
356endif()
357
358if(PROFILING_BACKEND_STREAMLINE)
359 target_link_libraries(armnn pthread)
360endif()
361
telsoa014fcda012018-03-09 14:13:49 +0000362if(BUILD_UNIT_TESTS)
363 set(unittest_sources)
364 list(APPEND unittest_sources
telsoa014fcda012018-03-09 14:13:49 +0000365 src/armnn/test/CreateWorkload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100366 src/armnn/test/CsvReaderTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100367 src/armnn/test/EndToEndTest.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100368 src/armnn/test/FloatingPointConverterTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100369 src/armnn/test/GraphTests.cpp
narpra016f37f832018-12-21 18:30:00 +0000370 src/armnn/test/GraphUtils.cpp
telsoa014fcda012018-03-09 14:13:49 +0000371 src/armnn/test/GraphUtils.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100372 src/armnn/test/InstrumentTests.cpp
Éanna Ó Catháin262553e2018-11-14 11:26:23 +0000373 src/armnn/test/LayerValidateOutputTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100374 src/armnn/test/NetworkTests.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100375 src/armnn/test/ObservableTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100376 src/armnn/test/OptimizerTests.cpp
377 src/armnn/test/OptionalTest.cpp
378 src/armnn/test/ProfilerTests.cpp
379 src/armnn/test/ProfilingEventTest.cpp
380 src/armnn/test/RuntimeTests.cpp
381 src/armnn/test/RuntimeTests.hpp
382 src/armnn/test/SubGraphTests.cpp
383 src/armnn/test/TensorHelpers.hpp
384 src/armnn/test/TensorTest.cpp
385 src/armnn/test/UnitTests.cpp
386 src/armnn/test/UnitTests.hpp
387 src/armnn/test/UtilsTests.cpp)
telsoa014fcda012018-03-09 14:13:49 +0000388
surmeh01bceff2f2018-03-29 16:29:27 +0100389 if(BUILD_TF_PARSER)
390 list(APPEND unittest_sources
391 src/armnnTfParser/test/Activations.cpp
392 src/armnnTfParser/test/Addition.cpp
Ferran Balaguerfbdad032018-12-28 18:15:24 +0000393 src/armnnTfParser/test/AddN.cpp
narpra016f37f832018-12-21 18:30:00 +0000394 src/armnnTfParser/test/Assert.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100395 src/armnnTfParser/test/BiasAdd.cpp
396 src/armnnTfParser/test/BroadcastForAdd.cpp
397 src/armnnTfParser/test/Convolution2d.cpp
398 src/armnnTfParser/test/Concat.cpp
399 src/armnnTfParser/test/ConcatOfConcats.cpp
400 src/armnnTfParser/test/DepthwiseConvolution2d.cpp
jimfly0184c70e62018-12-19 13:14:46 +0000401 src/armnnTfParser/test/Equal.cpp
Conor Kennedyc2130a02018-12-05 11:05:54 +0000402 src/armnnTfParser/test/ExpandDims.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100403 src/armnnTfParser/test/FusedBatchNorm.cpp
jimfly01a06bf312018-12-18 16:24:51 +0000404 src/armnnTfParser/test/Greater.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100405 src/armnnTfParser/test/Identity.cpp
406 src/armnnTfParser/test/LocalResponseNormalization.cpp
Sadik Armagan975c09a2018-12-04 10:02:08 +0000407 src/armnnTfParser/test/Maximum.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100408 src/armnnTfParser/test/MaximumForLeakyRelu.cpp
Nattapat Chaimanowong24df8222018-12-04 13:47:02 +0000409 src/armnnTfParser/test/Minimum.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100410 src/armnnTfParser/test/Multiplication.cpp
jimfly01f6ba7472018-12-04 10:09:52 +0000411 src/armnnTfParser/test/Pad.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100412 src/armnnTfParser/test/MultiOutput.cpp
413 src/armnnTfParser/test/PassThru.cpp
414 src/armnnTfParser/test/Pooling.cpp
saoste01bbd40612018-08-28 15:41:51 +0100415 src/armnnTfParser/test/RealDiv.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100416 src/armnnTfParser/test/Reshape.cpp
417 src/armnnTfParser/test/ResizeBilinear.cpp
418 src/armnnTfParser/test/Shape.cpp
419 src/armnnTfParser/test/Softmax.cpp
420 src/armnnTfParser/test/TestDependencies.cpp
421 src/armnnTfParser/test/FullyConnected.cpp
422 src/armnnTfParser/test/Constant.cpp
423 src/armnnTfParser/test/TestMultiInputsOutputs.cpp
Sadik Armagan2ad6cb42018-12-27 11:23:44 +0000424 src/armnnTfParser/test/Split.cpp
jimfly0123be07e2018-12-04 17:47:22 +0000425 src/armnnTfParser/test/Squeeze.cpp
426 src/armnnTfParser/test/Sub.cpp)
surmeh01bceff2f2018-03-29 16:29:27 +0100427 endif()
428
telsoa01c577f2c2018-08-31 09:22:23 +0100429 if(BUILD_TF_LITE_PARSER)
430 list(APPEND unittest_sources
431 src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp
Sadik Armagan479045b2018-10-01 11:51:37 +0100432 src/armnnTfLiteParser/test/Activations.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100433 src/armnnTfLiteParser/test/AvgPool2D.cpp
Sadik Armagan479045b2018-10-01 11:51:37 +0100434 src/armnnTfLiteParser/test/Concatenation.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100435 src/armnnTfLiteParser/test/Conv2D.cpp
436 src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
Sadik Armagan8853c1f2018-10-22 09:04:18 +0100437 src/armnnTfLiteParser/test/FullyConnected.cpp
Nattapat Chaimanowongb66504b2018-10-17 15:19:14 +0100438 src/armnnTfLiteParser/test/MaxPool2D.cpp
Sadik Armagan479045b2018-10-01 11:51:37 +0100439 src/armnnTfLiteParser/test/Reshape.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100440 src/armnnTfLiteParser/test/Softmax.cpp
441 src/armnnTfLiteParser/test/Squeeze.cpp
442 src/armnnTfLiteParser/test/LoadModel.cpp
443 src/armnnTfLiteParser/test/GetBuffer.cpp
444 src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
445 src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
446 src/armnnTfLiteParser/test/GetTensorIds.cpp
447 src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
448 src/armnnTfLiteParser/test/GetInputsOutputs.cpp
449 )
450 endif()
451
telsoa014fcda012018-03-09 14:13:49 +0000452 if(BUILD_CAFFE_PARSER)
453 list(APPEND unittest_sources
454 src/armnnCaffeParser/test/TestAdd.cpp
455 src/armnnCaffeParser/test/TestConcat.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100456 src/armnnCaffeParser/test/TestConvolution.cpp
telsoa014fcda012018-03-09 14:13:49 +0000457 src/armnnCaffeParser/test/TestDropout.cpp
458 src/armnnCaffeParser/test/TestInputs.cpp
459 src/armnnCaffeParser/test/TestMul.cpp
460 src/armnnCaffeParser/test/TestPooling2d.cpp
461 src/armnnCaffeParser/test/TestInPlace.cpp
462 src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
463 src/armnnCaffeParser/test/TestSplit.cpp
464 )
465 endif()
466
telsoa01c577f2c2018-08-31 09:22:23 +0100467 if(BUILD_ONNX_PARSER)
468 list(APPEND unittest_sources
469 src/armnnOnnxParser/test/Constructor.cpp
470 src/armnnOnnxParser/test/CreateNetwork.cpp
471 src/armnnOnnxParser/test/ProtoxtFixture.cpp
472 src/armnnOnnxParser/test/Const.cpp
473 src/armnnOnnxParser/test/Pooling.cpp
474 src/armnnOnnxParser/test/Reshape.cpp
475 src/armnnOnnxParser/test/Relu.cpp
476 src/armnnOnnxParser/test/Conv2D.cpp
477 src/armnnOnnxParser/test/Addition.cpp
478 src/armnnOnnxParser/test/FullyConnected.cpp
479 src/armnnOnnxParser/test/GetInputsOutputs.cpp
480 src/armnnOnnxParser/test/BatchNorm.cpp
481 src/armnnOnnxParser/test/DepthConv.cpp
482 )
483 endif()
484
Matteo Martincigh4423ac72018-10-11 17:39:30 +0100485 foreach(lib ${armnnUnitTestLibraries})
David Beck3cc9a622018-10-12 10:38:31 +0100486 message("Adding object library dependency to UnitTests: ${lib}")
Matteo Martincigh4423ac72018-10-11 17:39:30 +0100487 list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
488 endforeach()
489
490 add_executable(UnitTests ${unittest_sources})
telsoa014fcda012018-03-09 14:13:49 +0000491 target_include_directories(UnitTests PRIVATE src/armnn)
492 target_include_directories(UnitTests PRIVATE src/armnnUtils)
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000493 target_include_directories(UnitTests PRIVATE src/backends)
telsoa014fcda012018-03-09 14:13:49 +0000494
telsoa01c577f2c2018-08-31 09:22:23 +0100495 if(VALGRIND_FOUND)
496 if(HEAP_PROFILING OR LEAK_CHECKING)
497 message("Valgrind is disabled for heap profiling and leak checking builds.")
498 else()
499 # Valgrind works with gperftools version number <= 2.4
500 target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
501 endif()
telsoa014fcda012018-03-09 14:13:49 +0000502 endif()
503
telsoa014fcda012018-03-09 14:13:49 +0000504 target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
telsoa01c577f2c2018-08-31 09:22:23 +0100505 target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
telsoa014fcda012018-03-09 14:13:49 +0000506
Matteo Martincigh4423ac72018-10-11 17:39:30 +0100507 target_link_libraries(UnitTests armnn)
508 target_link_libraries(UnitTests armnnUtils)
509
surmeh01bceff2f2018-03-29 16:29:27 +0100510 if(BUILD_TF_PARSER)
511 target_link_libraries(UnitTests armnnTfParser)
512 endif()
513
telsoa014fcda012018-03-09 14:13:49 +0000514 if(BUILD_CAFFE_PARSER)
515 target_link_libraries(UnitTests armnnCaffeParser)
516 endif()
517
telsoa01c577f2c2018-08-31 09:22:23 +0100518 if(BUILD_TF_LITE_PARSER)
519 target_link_libraries(UnitTests armnnTfLiteParser)
520 endif()
521
522 if(BUILD_ONNX_PARSER)
523 target_link_libraries(UnitTests armnnOnnxParser)
524 endif()
525
telsoa014fcda012018-03-09 14:13:49 +0000526 addDllCopyCommands(UnitTests)
527endif()