blob: 9c2685c96d886f3ca1b7b3b4ab8b2ab67525c0c5 [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +00001cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake.
2project(armnn)
3
4set(additional_cmake_files)
5list(APPEND additional_cmake_files
6 cmake/Utils.cmake
7 cmake/GlobalConfig.cmake
8 cmake/AddDllCopyCommands.cmake)
9
10foreach(cmake_file ${additional_cmake_files})
11 include(${cmake_file})
12endforeach()
13
14if (BUILD_TESTS)
15 add_subdirectory(tests)
16endif()
17
telsoa01c577f2c2018-08-31 09:22:23 +010018add_subdirectory(samples)
19
telsoa014fcda012018-03-09 14:13:49 +000020# Include the additional cmake files in their own target so that they will appear nicely in IDEs
21add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files})
22
23set(armnnUtils_sources)
24list(APPEND armnnUtils_sources
25 src/armnnUtils/GraphTopologicalSort.hpp
26 src/armnnUtils/Logging.hpp
27 src/armnnUtils/Permute.hpp
28 src/armnnUtils/Logging.cpp
29 src/armnnUtils/Permute.cpp
surmeh01bceff2f2018-03-29 16:29:27 +010030 src/armnnUtils/DotSerializer.cpp
31 src/armnnUtils/DotSerializer.hpp
surmeh013537c2c2018-05-18 16:31:43 +010032 src/armnnUtils/HeapProfiling.cpp
33 src/armnnUtils/HeapProfiling.hpp
34 src/armnnUtils/LeakChecking.cpp
35 src/armnnUtils/LeakChecking.hpp
telsoa01c577f2c2018-08-31 09:22:23 +010036 src/armnnUtils/CsvReader.cpp
37 src/armnnUtils/CsvReader.hpp
38 src/armnnUtils/FloatingPointConverter.cpp
39 src/armnnUtils/FloatingPointConverter.hpp
40 src/armnnUtils/VerificationHelpers.hpp
41 src/armnnUtils/VerificationHelpers.cpp
telsoa014fcda012018-03-09 14:13:49 +000042 )
surmeh01bceff2f2018-03-29 16:29:27 +010043if(BUILD_TF_PARSER OR BUILD_CAFFE_PARSER)
telsoa014fcda012018-03-09 14:13:49 +000044 list(APPEND armnnUtils_sources
45 src/armnnUtils/ParserPrototxtFixture.hpp
46 )
47endif()
48
49add_library_ex(armnnUtils STATIC ${armnnUtils_sources})
50
51if(BUILD_CAFFE_PARSER)
52 # ArmNN Parser source files required for all build options
53 set(armnn_caffe_parser_sources)
54 list(APPEND armnn_caffe_parser_sources
55 include/armnnCaffeParser/ICaffeParser.hpp
telsoa01c577f2c2018-08-31 09:22:23 +010056 src/armnnCaffeParser/RecordByRecordCaffeParser.hpp
57 src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
telsoa014fcda012018-03-09 14:13:49 +000058 src/armnnCaffeParser/CaffeParser.hpp
59 src/armnnCaffeParser/CaffeParser.cpp
60 ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc
61 )
62 # The generated Caffe protobuf .cc file is not warning clean and we can't fix them.
63 if(COMPILER_IS_GNU_LIKE)
64 set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
65 endif()
66
67 add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})
68 set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")
69
70 target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)
71
72 target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
73
74 target_link_libraries(armnnCaffeParser armnn)
75 target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})
telsoa01c577f2c2018-08-31 09:22:23 +010076
77endif()
78
79if(BUILD_ONNX_PARSER)
80 set(armnn_onnx_parser_sources)
81 list(APPEND armnn_onnx_parser_sources
82 include/armnnOnnxParser/IOnnxParser.hpp
83 src/armnnOnnxParser/OnnxParser.hpp
84 src/armnnOnnxParser/OnnxParser.cpp
85 ${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc
86 )
87 # The generated onnx protobuf .cc files are not warning clean and we can't fix them.
88 if(COMPILER_IS_GNU_LIKE)
89 set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
90 endif()
91
92 add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})
93
94 target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)
95
96 target_link_libraries(armnnOnnxParser armnn)
97
98 # Protobuf
99 target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})
telsoa014fcda012018-03-09 14:13:49 +0000100endif()
101
surmeh01bceff2f2018-03-29 16:29:27 +0100102if(BUILD_TF_PARSER)
103 set(armnn_tf_parser_sources)
104 list(APPEND armnn_tf_parser_sources
105 include/armnnTfParser/ITfParser.hpp
106 src/armnnTfParser/TfParser.hpp
107 src/armnnTfParser/TfParser.cpp
108 ${TF_PROTOBUFS}
109 )
110 # The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.
111 if(COMPILER_IS_GNU_LIKE)
112 set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
113 endif()
114
115 add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})
116
117 target_include_directories(armnnTfParser PRIVATE src/armnnUtils)
118
119 target_link_libraries(armnnTfParser armnn)
120
121 # Protobuf (use the specific version tensorflow wants)
122 target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})
123endif()
124
telsoa01c577f2c2018-08-31 09:22:23 +0100125if(BUILD_TF_LITE_PARSER)
126 set(armnn_tf_lite_parser_sources)
127 list(APPEND armnn_tf_lite_parser_sources
128 include/armnnTfLiteParser/ITfLiteParser.hpp
129 src/armnnTfLiteParser/TfLiteParser.hpp
130 src/armnnTfLiteParser/TfLiteParser.cpp
131 )
132
133 add_library_ex(armnnTfLiteParser SHARED ${armnn_tf_lite_parser_sources})
134
135 target_include_directories(armnnTfLiteParser PRIVATE src/armnnUtils)
136
137 target_link_libraries(armnnTfLiteParser ${Boost_FILESYSTEM_LIBRARY} ${Boost_THREAD_LIBRARY})
138 target_link_libraries(armnnTfLiteParser armnn ${FLATBUFFERS_LIBRARY})
139endif()
140
telsoa014fcda012018-03-09 14:13:49 +0000141# ArmNN source files required for all build options
telsoa01c577f2c2018-08-31 09:22:23 +0100142include_directories(SYSTEM third-party)
143
telsoa014fcda012018-03-09 14:13:49 +0000144list(APPEND armnn_sources
145 include/armnn/ArmNN.hpp
146 include/armnn/Descriptors.hpp
147 include/armnn/DescriptorsFwd.hpp
148 include/armnn/IRuntime.hpp
149 include/armnn/INetwork.hpp
150 include/armnn/Tensor.hpp
151 include/armnn/TensorFwd.hpp
152 include/armnn/Types.hpp
153 include/armnn/TypesUtils.hpp
154 include/armnn/Utils.hpp
155 include/armnn/LayerSupport.hpp
156 include/armnn/Version.hpp
157 src/armnn/backends/ClWorkloadFactory.hpp
158 src/armnn/backends/ClWorkloadFactory.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100159 src/armnn/backends/ClContextControl.hpp
160 src/armnn/backends/ClContextControl.cpp
telsoa014fcda012018-03-09 14:13:49 +0000161 src/armnn/backends/ClLayerSupport.cpp
162 src/armnn/backends/ClLayerSupport.hpp
163 src/armnn/backends/CpuTensorHandleFwd.hpp
164 src/armnn/backends/CpuTensorHandle.hpp
165 src/armnn/backends/CpuTensorHandle.cpp
166 src/armnn/backends/RefWorkloadFactory.cpp
167 src/armnn/backends/RefWorkloadFactory.hpp
168 src/armnn/backends/RefLayerSupport.cpp
169 src/armnn/backends/RefLayerSupport.hpp
170 src/armnn/backends/MakeWorkloadHelper.hpp
171 src/armnn/backends/NeonWorkloadFactory.cpp
172 src/armnn/backends/NeonWorkloadFactory.hpp
173 src/armnn/backends/NeonLayerSupport.cpp
174 src/armnn/backends/NeonLayerSupport.hpp
175 src/armnn/backends/Workload.hpp
176 src/armnn/backends/WorkloadDataFwd.hpp
177 src/armnn/backends/WorkloadData.hpp
178 src/armnn/backends/WorkloadData.cpp
179 src/armnn/backends/WorkloadFactory.hpp
180 src/armnn/backends/WorkloadFactory.cpp
181 src/armnn/backends/WorkloadInfo.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100182 src/armnn/backends/WorkloadUtils.hpp
telsoa014fcda012018-03-09 14:13:49 +0000183 src/armnn/backends/MemCopyWorkload.cpp
184 src/armnn/backends/MemCopyWorkload.hpp
185 src/armnn/backends/RefWorkloads/Broadcast.hpp
186 src/armnn/backends/RefWorkloads/Broadcast.cpp
187 src/armnn/backends/RefWorkloads/RefMergerUint8Workload.cpp
188 src/armnn/backends/RefWorkloads/RefConstantUint8Workload.hpp
David Beck591cdb72018-09-11 16:37:14 +0100189 src/armnn/backends/RefWorkloads/ArithmeticFunction.cpp
190 src/armnn/backends/RefWorkloads/ArithmeticFunction.hpp
David Beckf195f032018-09-06 16:46:34 +0100191 src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp
192 src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.hpp
193 src/armnn/backends/RefWorkloads/RefAdditionFloat32Workload.cpp
194 src/armnn/backends/RefWorkloads/RefAdditionFloat32Workload.hpp
David Beckf195f032018-09-06 16:46:34 +0100195 src/armnn/backends/RefWorkloads/RefSubtractionUint8Workload.cpp
196 src/armnn/backends/RefWorkloads/RefSubtractionUint8Workload.hpp
197 src/armnn/backends/RefWorkloads/RefSubtractionFloat32Workload.cpp
198 src/armnn/backends/RefWorkloads/RefSubtractionFloat32Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000199 src/armnn/backends/RefWorkloads/ConvImpl.hpp
200 src/armnn/backends/RefWorkloads/RefResizeBilinearUint8Workload.cpp
201 src/armnn/backends/RefWorkloads/RefMultiplicationUint8Workload.hpp
202 src/armnn/backends/RefWorkloads/FullyConnected.cpp
203 src/armnn/backends/RefWorkloads/RefFullyConnectedFloat32Workload.cpp
204 src/armnn/backends/RefWorkloads/RefSoftmaxFloat32Workload.cpp
205 src/armnn/backends/RefWorkloads/RefBaseConstantWorkload.hpp
206 src/armnn/backends/RefWorkloads/RefConvolution2dUint8Workload.hpp
207 src/armnn/backends/RefWorkloads/RefSplitterUint8Workload.hpp
208 src/armnn/backends/RefWorkloads/RefActivationFloat32Workload.cpp
209 src/armnn/backends/RefWorkloads/RefBatchNormalizationUint8Workload.cpp
210 src/armnn/backends/RefWorkloads/RefResizeBilinearUint8Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000211 src/armnn/backends/RefWorkloads/RefMultiplicationUint8Workload.cpp
212 src/armnn/backends/RefWorkloads/RefL2NormalizationFloat32Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000213 src/armnn/backends/RefWorkloads/RefActivationUint8Workload.hpp
214 src/armnn/backends/RefWorkloads/RefBaseConstantWorkload.cpp
telsoa014fcda012018-03-09 14:13:49 +0000215 src/armnn/backends/RefWorkloads/RefResizeBilinearFloat32Workload.cpp
216 src/armnn/backends/RefWorkloads/RefBatchNormalizationFloat32Workload.cpp
217 src/armnn/backends/RefWorkloads/RefPooling2dFloat32Workload.hpp
218 src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp
219 src/armnn/backends/RefWorkloads/RefMergerUint8Workload.hpp
220 src/armnn/backends/RefWorkloads/RefFullyConnectedFloat32Workload.hpp
221 src/armnn/backends/RefWorkloads/Softmax.hpp
222 src/armnn/backends/RefWorkloads/RefMergerFloat32Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000223 src/armnn/backends/RefWorkloads/RefFakeQuantizationFloat32Workload.cpp
224 src/armnn/backends/RefWorkloads/TensorBufferArrayView.hpp
225 src/armnn/backends/RefWorkloads/ResizeBilinear.cpp
226 src/armnn/backends/RefWorkloads/RefBatchNormalizationFloat32Workload.hpp
227 src/armnn/backends/RefWorkloads/Splitter.hpp
228 src/armnn/backends/RefWorkloads/RefFullyConnectedUint8Workload.hpp
229 src/armnn/backends/RefWorkloads/RefSoftmaxUint8Workload.cpp
230 src/armnn/backends/RefWorkloads/RefReshapeFloat32Workload.hpp
231 src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dUint8Workload.hpp
232 src/armnn/backends/RefWorkloads/FullyConnected.hpp
233 src/armnn/backends/RefWorkloads/RefFloorFloat32Workload.hpp
234 src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dFloat32Workload.cpp
235 src/armnn/backends/RefWorkloads/RefPooling2dUint8Workload.cpp
236 src/armnn/backends/RefWorkloads/RefFloorFloat32Workload.cpp
237 src/armnn/backends/RefWorkloads/ConvImpl.cpp
238 src/armnn/backends/RefWorkloads/RefSoftmaxFloat32Workload.hpp
239 src/armnn/backends/RefWorkloads/RefSoftmaxUint8Workload.hpp
240 src/armnn/backends/RefWorkloads/RefReshapeUint8Workload.hpp
241 src/armnn/backends/RefWorkloads/Activation.cpp
242 src/armnn/backends/RefWorkloads/RefResizeBilinearFloat32Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000243 src/armnn/backends/RefWorkloads/RefReshapeUint8Workload.cpp
244 src/armnn/backends/RefWorkloads/RefMultiplicationFloat32Workload.hpp
245 src/armnn/backends/RefWorkloads/RefL2NormalizationFloat32Workload.cpp
246 src/armnn/backends/RefWorkloads/RefConvolution2dFloat32Workload.cpp
247 src/armnn/backends/RefWorkloads/RefBatchNormalizationUint8Workload.hpp
248 src/armnn/backends/RefWorkloads/RefConvolution2dUint8Workload.cpp
249 src/armnn/backends/RefWorkloads/ResizeBilinear.hpp
250 src/armnn/backends/RefWorkloads/RefNormalizationFloat32Workload.hpp
251 src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dFloat32Workload.hpp
252 src/armnn/backends/RefWorkloads/RefSplitterFloat32Workload.cpp
253 src/armnn/backends/RefWorkloads/RefActivationUint8Workload.cpp
254 src/armnn/backends/RefWorkloads/RefPooling2dUint8Workload.hpp
255 src/armnn/backends/RefWorkloads/BatchNormImpl.hpp
256 src/armnn/backends/RefWorkloads/RefSplitterUint8Workload.cpp
257 src/armnn/backends/RefWorkloads/Activation.hpp
258 src/armnn/backends/RefWorkloads/RefPooling2dFloat32Workload.cpp
259 src/armnn/backends/RefWorkloads/Merger.hpp
260 src/armnn/backends/RefWorkloads/RefSplitterFloat32Workload.hpp
261 src/armnn/backends/RefWorkloads/RefConstantFloat32Workload.hpp
262 src/armnn/backends/RefWorkloads/RefActivationFloat32Workload.hpp
263 src/armnn/backends/RefWorkloads/RefMultiplicationFloat32Workload.cpp
264 src/armnn/backends/RefWorkloads/RefReshapeFloat32Workload.cpp
265 src/armnn/backends/RefWorkloads/RefNormalizationFloat32Workload.cpp
266 src/armnn/backends/RefWorkloads/Softmax.cpp
267 src/armnn/backends/RefWorkloads/RefDepthwiseConvolution2dUint8Workload.cpp
268 src/armnn/backends/RefWorkloads/RefConstantUint8Workload.cpp
269 src/armnn/backends/RefWorkloads/RefConstantFloat32Workload.cpp
270 src/armnn/backends/RefWorkloads/Pooling2d.cpp
telsoa014fcda012018-03-09 14:13:49 +0000271 src/armnn/backends/RefWorkloads/RefConvolution2dFloat32Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000272 src/armnn/backends/RefWorkloads/RefMergerFloat32Workload.cpp
273 src/armnn/backends/RefWorkloads/Pooling2d.hpp
274 src/armnn/backends/RefWorkloads/RefFullyConnectedUint8Workload.cpp
275 src/armnn/backends/RefWorkloads/RefFakeQuantizationFloat32Workload.hpp
276 src/armnn/backends/RefWorkloads/RefPermuteWorkload.hpp
277 src/armnn/backends/RefWorkloads/RefPermuteWorkload.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100278 src/armnn/backends/RefWorkloads/RefLstmFloat32Workload.cpp
279 src/armnn/backends/RefWorkloads/RefLstmFloat32Workload.hpp
280 src/armnn/backends/RefWorkloads/RefConvertFp16ToFp32Workload.cpp
281 src/armnn/backends/RefWorkloads/RefConvertFp16ToFp32Workload.hpp
282 src/armnn/backends/RefWorkloads/RefConvertFp32ToFp16Workload.cpp
283 src/armnn/backends/RefWorkloads/RefConvertFp32ToFp16Workload.hpp
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100284 src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.cpp
285 src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.hpp
286 src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp
287 src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100288 src/armnn/layers/LayerCloneBase.hpp
289 src/armnn/layers/LayerWithParameters.hpp
290 src/armnn/layers/ActivationLayer.hpp
291 src/armnn/layers/ActivationLayer.cpp
292 src/armnn/layers/AdditionLayer.hpp
293 src/armnn/layers/AdditionLayer.cpp
Les Bell033626d2018-09-03 16:24:52 +0100294 src/armnn/layers/ArithmeticBaseLayer.hpp
295 src/armnn/layers/ArithmeticBaseLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100296 src/armnn/layers/BatchNormalizationLayer.hpp
297 src/armnn/layers/BatchNormalizationLayer.cpp
298 src/armnn/layers/ConstantLayer.hpp
299 src/armnn/layers/ConstantLayer.cpp
300 src/armnn/layers/Convolution2dLayer.hpp
301 src/armnn/layers/Convolution2dLayer.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100302 src/armnn/layers/ConvertFp16ToFp32Layer.hpp
303 src/armnn/layers/ConvertFp16ToFp32Layer.cpp
304 src/armnn/layers/ConvertFp32ToFp16Layer.hpp
305 src/armnn/layers/ConvertFp32ToFp16Layer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100306 src/armnn/layers/DepthwiseConvolution2dLayer.hpp
307 src/armnn/layers/DepthwiseConvolution2dLayer.cpp
308 src/armnn/layers/FakeQuantizationLayer.hpp
309 src/armnn/layers/FakeQuantizationLayer.cpp
310 src/armnn/layers/FloorLayer.hpp
311 src/armnn/layers/FloorLayer.cpp
312 src/armnn/layers/FullyConnectedLayer.hpp
313 src/armnn/layers/FullyConnectedLayer.cpp
314 src/armnn/layers/InputLayer.hpp
315 src/armnn/layers/InputLayer.cpp
316 src/armnn/layers/L2NormalizationLayer.hpp
317 src/armnn/layers/L2NormalizationLayer.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100318 src/armnn/layers/LstmLayer.cpp
319 src/armnn/layers/LstmLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100320 src/armnn/layers/MemCopyLayer.hpp
321 src/armnn/layers/MemCopyLayer.cpp
322 src/armnn/layers/MergerLayer.hpp
323 src/armnn/layers/MergerLayer.cpp
324 src/armnn/layers/MultiplicationLayer.hpp
325 src/armnn/layers/MultiplicationLayer.cpp
326 src/armnn/layers/NormalizationLayer.hpp
327 src/armnn/layers/NormalizationLayer.cpp
328 src/armnn/layers/OutputLayer.hpp
329 src/armnn/layers/OutputLayer.cpp
330 src/armnn/layers/PermuteLayer.hpp
331 src/armnn/layers/PermuteLayer.cpp
332 src/armnn/layers/Pooling2dLayer.hpp
333 src/armnn/layers/Pooling2dLayer.cpp
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100334 src/armnn/layers/DivisionLayer.cpp
335 src/armnn/layers/DivisionLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100336 src/armnn/layers/ReshapeLayer.hpp
337 src/armnn/layers/ReshapeLayer.cpp
338 src/armnn/layers/ResizeBilinearLayer.hpp
339 src/armnn/layers/ResizeBilinearLayer.cpp
340 src/armnn/layers/SoftmaxLayer.hpp
341 src/armnn/layers/SoftmaxLayer.cpp
342 src/armnn/layers/SplitterLayer.hpp
343 src/armnn/layers/SplitterLayer.cpp
David Beckc2044fe2018-09-05 15:00:38 +0100344 src/armnn/layers/SubtractionLayer.cpp
345 src/armnn/layers/SubtractionLayer.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100346 src/armnn/Half.hpp
telsoa014fcda012018-03-09 14:13:49 +0000347 src/armnn/InternalTypes.hpp
348 src/armnn/InternalTypes.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100349 src/armnn/JsonPrinter.hpp
350 src/armnn/JsonPrinter.cpp
telsoa014fcda012018-03-09 14:13:49 +0000351 src/armnn/LayerFwd.hpp
352 src/armnn/Layer.hpp
353 src/armnn/Layer.cpp
354 src/armnn/LayersFwd.hpp
telsoa014fcda012018-03-09 14:13:49 +0000355 src/armnn/Runtime.hpp
356 src/armnn/Runtime.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100357 src/armnn/SerializeLayerParameters.cpp
358 src/armnn/SerializeLayerParameters.hpp
telsoa014fcda012018-03-09 14:13:49 +0000359 src/armnn/Descriptors.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100360 src/armnn/DeviceSpec.hpp
telsoa014fcda012018-03-09 14:13:49 +0000361 src/armnn/LoadedNetwork.hpp
362 src/armnn/LoadedNetwork.cpp
363 src/armnn/Exceptions.cpp
364 src/armnn/Graph.hpp
365 src/armnn/Graph.cpp
366 src/armnn/Network.hpp
367 src/armnn/Network.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100368 src/armnn/NetworkUtils.hpp
telsoa014fcda012018-03-09 14:13:49 +0000369 src/armnn/backends/OutputHandler.hpp
370 src/armnn/backends/OutputHandler.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100371 src/armnn/ProfilingEvent.cpp
372 src/armnn/ProfilingEvent.hpp
telsoa014fcda012018-03-09 14:13:49 +0000373 src/armnn/Profiling.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100374 src/armnn/Instrument.hpp
375 src/armnn/WallClockTimer.hpp
376 src/armnn/WallClockTimer.cpp
telsoa014fcda012018-03-09 14:13:49 +0000377 src/armnn/Tensor.cpp
378 src/armnn/Utils.cpp
379 src/armnn/LayerSupport.cpp
380 src/armnn/LayerSupportCommon.hpp
381 src/armnn/optimizations/All.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100382 src/armnn/optimizations/ConvertConstants.hpp
telsoa014fcda012018-03-09 14:13:49 +0000383 src/armnn/optimizations/MovePermuteUp.hpp
384 src/armnn/optimizations/Optimization.hpp
385 src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
386 src/armnn/optimizations/OptimizeInversePermutes.hpp
387 src/armnn/optimizations/PermuteAsReshape.hpp
388 src/armnn/optimizations/SquashEqualSiblings.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100389 src/armnn/optimizations/OptimizeInverseConversions.hpp
390 src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
telsoa014fcda012018-03-09 14:13:49 +0000391 src/armnn/Optimizer.hpp
392 src/armnn/Optimizer.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100393 third-party/half/half.hpp
394 src/armnn/IGraphObservable.hpp
395 src/armnn/Observable.hpp
396 src/armnn/Observable.cpp
telsoa014fcda012018-03-09 14:13:49 +0000397 )
398
399if(ARMCOMPUTENEON)
400 # Additionally include source files for ARM Compute NEON backend
401 list(APPEND armnn_sources
arovir019e53a352018-08-31 15:26:35 +0100402 src/armnn/backends/NeonWorkloads/NeonActivationFloatWorkload.cpp
403 src/armnn/backends/NeonWorkloads/NeonActivationFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000404 src/armnn/backends/NeonWorkloads/NeonActivationUint8Workload.cpp
405 src/armnn/backends/NeonWorkloads/NeonActivationUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100406 src/armnn/backends/NeonWorkloads/NeonAdditionFloatWorkload.cpp
407 src/armnn/backends/NeonWorkloads/NeonAdditionFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000408 src/armnn/backends/NeonWorkloads/NeonBaseConstantWorkload.hpp
409 src/armnn/backends/NeonWorkloads/NeonBaseMergerWorkload.hpp
410 src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100411 src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloatWorkload.cpp
412 src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloatWorkload.hpp
413 src/armnn/backends/NeonWorkloads/NeonConstantFloatWorkload.cpp
414 src/armnn/backends/NeonWorkloads/NeonConstantFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000415 src/armnn/backends/NeonWorkloads/NeonConstantUint8Workload.cpp
416 src/armnn/backends/NeonWorkloads/NeonConstantUint8Workload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100417 src/armnn/backends/NeonWorkloads/NeonConvertFp16ToFp32Workload.cpp
418 src/armnn/backends/NeonWorkloads/NeonConvertFp16ToFp32Workload.hpp
419 src/armnn/backends/NeonWorkloads/NeonConvertFp32ToFp16Workload.cpp
420 src/armnn/backends/NeonWorkloads/NeonConvertFp32ToFp16Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000421 src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.cpp
422 src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100423 src/armnn/backends/NeonWorkloads/NeonConvolution2dFloatWorkload.cpp
424 src/armnn/backends/NeonWorkloads/NeonConvolution2dFloatWorkload.hpp
surmeh01bceff2f2018-03-29 16:29:27 +0100425 src/armnn/backends/NeonWorkloads/NeonConvolution2dUint8Workload.cpp
426 src/armnn/backends/NeonWorkloads/NeonConvolution2dUint8Workload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100427 src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.cpp
428 src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100429 src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionFloatWorkload.cpp
430 src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000431 src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.cpp
432 src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100433 src/armnn/backends/NeonWorkloads/NeonFloorFloatWorkload.cpp
434 src/armnn/backends/NeonWorkloads/NeonFloorFloatWorkload.hpp
435 src/armnn/backends/NeonWorkloads/NeonFullyConnectedFloatWorkload.cpp
436 src/armnn/backends/NeonWorkloads/NeonFullyConnectedFloatWorkload.hpp
437 src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloatWorkload.cpp
438 src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloatWorkload.hpp
439 src/armnn/backends/NeonWorkloads/NeonLstmFloatWorkload.cpp
440 src/armnn/backends/NeonWorkloads/NeonLstmFloatWorkload.hpp
441 src/armnn/backends/NeonWorkloads/NeonMergerFloatWorkload.cpp
442 src/armnn/backends/NeonWorkloads/NeonMergerFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000443 src/armnn/backends/NeonWorkloads/NeonMergerUint8Workload.cpp
444 src/armnn/backends/NeonWorkloads/NeonMergerUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100445 src/armnn/backends/NeonWorkloads/NeonMultiplicationFloatWorkload.cpp
446 src/armnn/backends/NeonWorkloads/NeonMultiplicationFloatWorkload.hpp
447 src/armnn/backends/NeonWorkloads/NeonNormalizationFloatWorkload.cpp
448 src/armnn/backends/NeonWorkloads/NeonNormalizationFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000449 src/armnn/backends/NeonWorkloads/NeonPermuteWorkload.cpp
450 src/armnn/backends/NeonWorkloads/NeonPermuteWorkload.hpp
451 src/armnn/backends/NeonWorkloads/NeonPooling2dBaseWorkload.cpp
452 src/armnn/backends/NeonWorkloads/NeonPooling2dBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100453 src/armnn/backends/NeonWorkloads/NeonPooling2dFloatWorkload.cpp
454 src/armnn/backends/NeonWorkloads/NeonPooling2dFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000455 src/armnn/backends/NeonWorkloads/NeonPooling2dUint8Workload.cpp
456 src/armnn/backends/NeonWorkloads/NeonPooling2dUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100457 src/armnn/backends/NeonWorkloads/NeonReshapeFloatWorkload.cpp
458 src/armnn/backends/NeonWorkloads/NeonReshapeFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000459 src/armnn/backends/NeonWorkloads/NeonReshapeUint8Workload.cpp
460 src/armnn/backends/NeonWorkloads/NeonReshapeUint8Workload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100461 src/armnn/backends/NeonWorkloads/NeonSoftmaxBaseWorkload.cpp
462 src/armnn/backends/NeonWorkloads/NeonSoftmaxBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100463 src/armnn/backends/NeonWorkloads/NeonSoftmaxFloatWorkload.cpp
464 src/armnn/backends/NeonWorkloads/NeonSoftmaxFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000465 src/armnn/backends/NeonWorkloads/NeonSoftmaxUint8Workload.cpp
466 src/armnn/backends/NeonWorkloads/NeonSoftmaxUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100467 src/armnn/backends/NeonWorkloads/NeonSplitterFloatWorkload.cpp
468 src/armnn/backends/NeonWorkloads/NeonSplitterFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000469 src/armnn/backends/NeonWorkloads/NeonSplitterUint8Workload.cpp
470 src/armnn/backends/NeonWorkloads/NeonSplitterUint8Workload.hpp
471 src/armnn/backends/NeonWorkloadUtils.cpp
472 src/armnn/backends/NeonWorkloadUtils.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100473 src/armnn/backends/NeonTensorHandle.hpp
474 src/armnn/NeonInterceptorScheduler.hpp
475 src/armnn/NeonInterceptorScheduler.cpp
476 src/armnn/NeonTimer.hpp
477 src/armnn/NeonTimer.cpp)
telsoa014fcda012018-03-09 14:13:49 +0000478endif()
479if(ARMCOMPUTECL)
480 # Additionally include source files for ARM Compute OpenCL backend
481 list(APPEND armnn_sources
arovir019e53a352018-08-31 15:26:35 +0100482 src/armnn/backends/ClWorkloads/ClActivationFloatWorkload.cpp
483 src/armnn/backends/ClWorkloads/ClActivationFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000484 src/armnn/backends/ClWorkloads/ClActivationUint8Workload.cpp
485 src/armnn/backends/ClWorkloads/ClActivationUint8Workload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100486 src/armnn/backends/ClWorkloads/ClAdditionBaseWorkload.cpp
487 src/armnn/backends/ClWorkloads/ClAdditionBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100488 src/armnn/backends/ClWorkloads/ClAdditionFloatWorkload.cpp
489 src/armnn/backends/ClWorkloads/ClAdditionFloatWorkload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100490 src/armnn/backends/ClWorkloads/ClAdditionUint8Workload.cpp
491 src/armnn/backends/ClWorkloads/ClAdditionUint8Workload.hpp
David Beck4a8692c2018-09-07 16:19:24 +0100492 src/armnn/backends/ClWorkloads/ClSubtractionBaseWorkload.cpp
493 src/armnn/backends/ClWorkloads/ClSubtractionBaseWorkload.hpp
494 src/armnn/backends/ClWorkloads/ClSubtractionFloatWorkload.cpp
495 src/armnn/backends/ClWorkloads/ClSubtractionFloatWorkload.hpp
496 src/armnn/backends/ClWorkloads/ClSubtractionUint8Workload.cpp
497 src/armnn/backends/ClWorkloads/ClSubtractionUint8Workload.hpp
498 src/armnn/backends/ClWorkloads/ClConvertFp16ToFp32Workload.cpp
499 src/armnn/backends/ClWorkloads/ClConvertFp16ToFp32Workload.hpp
500 src/armnn/backends/ClWorkloads/ClConvertFp32ToFp16Workload.cpp
501 src/armnn/backends/ClWorkloads/ClConvertFp32ToFp16Workload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000502 src/armnn/backends/ClWorkloads/ClBaseConstantWorkload.cpp
503 src/armnn/backends/ClWorkloads/ClBaseConstantWorkload.hpp
504 src/armnn/backends/ClWorkloads/ClBaseMergerWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100505 src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.cpp
506 src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.hpp
507 src/armnn/backends/ClWorkloads/ClConstantFloatWorkload.cpp
508 src/armnn/backends/ClWorkloads/ClConstantFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000509 src/armnn/backends/ClWorkloads/ClConstantUint8Workload.cpp
510 src/armnn/backends/ClWorkloads/ClConstantUint8Workload.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100511 src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.cpp
512 src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100513 src/armnn/backends/ClWorkloads/ClConvolution2dFloatWorkload.cpp
514 src/armnn/backends/ClWorkloads/ClConvolution2dFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000515 src/armnn/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp
516 src/armnn/backends/ClWorkloads/ClConvolution2dUint8Workload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100517 src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.cpp
518 src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100519 src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionFloatWorkload.cpp
520 src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000521 src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.cpp
522 src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100523 src/armnn/backends/ClWorkloads/ClFloorFloatWorkload.cpp
524 src/armnn/backends/ClWorkloads/ClFloorFloatWorkload.hpp
525 src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.cpp
526 src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.hpp
527 src/armnn/backends/ClWorkloads/ClL2NormalizationFloatWorkload.cpp
528 src/armnn/backends/ClWorkloads/ClL2NormalizationFloatWorkload.hpp
529 src/armnn/backends/ClWorkloads/ClLstmFloatWorkload.cpp
530 src/armnn/backends/ClWorkloads/ClLstmFloatWorkload.hpp
531 src/armnn/backends/ClWorkloads/ClMergerFloatWorkload.cpp
532 src/armnn/backends/ClWorkloads/ClMergerFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000533 src/armnn/backends/ClWorkloads/ClMergerUint8Workload.cpp
534 src/armnn/backends/ClWorkloads/ClMergerUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100535 src/armnn/backends/ClWorkloads/ClMultiplicationFloatWorkload.cpp
536 src/armnn/backends/ClWorkloads/ClMultiplicationFloatWorkload.hpp
537 src/armnn/backends/ClWorkloads/ClNormalizationFloatWorkload.cpp
538 src/armnn/backends/ClWorkloads/ClNormalizationFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000539 src/armnn/backends/ClWorkloads/ClPermuteWorkload.cpp
540 src/armnn/backends/ClWorkloads/ClPermuteWorkload.hpp
541 src/armnn/backends/ClWorkloads/ClPooling2dBaseWorkload.cpp
542 src/armnn/backends/ClWorkloads/ClPooling2dBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100543 src/armnn/backends/ClWorkloads/ClPooling2dFloatWorkload.cpp
544 src/armnn/backends/ClWorkloads/ClPooling2dFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000545 src/armnn/backends/ClWorkloads/ClPooling2dUint8Workload.cpp
546 src/armnn/backends/ClWorkloads/ClPooling2dUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100547 src/armnn/backends/ClWorkloads/ClReshapeFloatWorkload.cpp
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100548 src/armnn/backends/ClWorkloads/ClDivisionFloatWorkload.hpp
549 src/armnn/backends/ClWorkloads/ClDivisionFloatWorkload.cpp
arovir019e53a352018-08-31 15:26:35 +0100550 src/armnn/backends/ClWorkloads/ClReshapeFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000551 src/armnn/backends/ClWorkloads/ClReshapeUint8Workload.cpp
552 src/armnn/backends/ClWorkloads/ClReshapeUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100553 src/armnn/backends/ClWorkloads/ClResizeBilinearFloatWorkload.cpp
554 src/armnn/backends/ClWorkloads/ClResizeBilinearFloatWorkload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100555 src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.cpp
556 src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp
arovir019e53a352018-08-31 15:26:35 +0100557 src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.cpp
558 src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000559 src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.cpp
560 src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp
arovir019e53a352018-08-31 15:26:35 +0100561 src/armnn/backends/ClWorkloads/ClSplitterFloatWorkload.cpp
562 src/armnn/backends/ClWorkloads/ClSplitterFloatWorkload.hpp
telsoa014fcda012018-03-09 14:13:49 +0000563 src/armnn/backends/ClWorkloads/ClSplitterUint8Workload.cpp
564 src/armnn/backends/ClWorkloads/ClSplitterUint8Workload.hpp
565 src/armnn/backends/ClWorkloadUtils.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100566 src/armnn/backends/ClTensorHandle.hpp
567 src/armnn/OpenClTimer.cpp
568 src/armnn/OpenClTimer.hpp)
telsoa014fcda012018-03-09 14:13:49 +0000569endif()
570# Files shared by all ARM Compute backends
571if(ARMCOMPUTENEON OR ARMCOMPUTECL)
572 list(APPEND armnn_sources
573 src/armnn/backends/ArmComputeTensorUtils.hpp
574 src/armnn/backends/ArmComputeTensorUtils.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100575 src/armnn/backends/ArmComputeUtils.hpp
576 src/armnn/memory/IMemoryPool.hpp
577 src/armnn/memory/BlobMemoryPool.cpp
578 src/armnn/memory/BlobMemoryPool.hpp
579 src/armnn/memory/BlobLifetimeManager.cpp
580 src/armnn/memory/BlobLifetimeManager.hpp
581 src/armnn/memory/PoolManager.cpp
582 src/armnn/memory/PoolManager.hpp
583 src/armnn/memory/BaseMemoryManager.hpp
584 src/armnn/memory/BaseMemoryManager.cpp
585 src/armnn/memory/OffsetMemoryPool.cpp
586 src/armnn/memory/OffsetMemoryPool.hpp
587 src/armnn/memory/OffsetLifetimeManager.cpp
588 src/armnn/memory/OffsetLifetimeManager.hpp)
telsoa014fcda012018-03-09 14:13:49 +0000589endif()
590
591# Files used for Streamline-based profiling backend
592if(PROFILING_BACKEND_STREAMLINE)
593 list(APPEND armnn_sources
594 ${GATOR_ROOT}/annotate/streamline_annotate.h
595 ${GATOR_ROOT}/annotate/streamline_annotate.c)
596endif()
597
598add_library_ex(armnn SHARED ${armnn_sources})
599target_include_directories(armnn PRIVATE src/armnn)
600target_include_directories(armnn PRIVATE src/armnnUtils)
601target_link_libraries(armnn armnnUtils)
602target_link_libraries(armnn ${CMAKE_DL_LIBS})
603
telsoa01c577f2c2018-08-31 09:22:23 +0100604
telsoa014fcda012018-03-09 14:13:49 +0000605install(TARGETS armnn DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
606if(BUILD_CAFFE_PARSER)
607 install(TARGETS armnnCaffeParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
608endif()
telsoa01c577f2c2018-08-31 09:22:23 +0100609if(BUILD_ONNX_PARSER)
610 install(TARGETS armnnOnnxParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
611endif()
surmeh01bceff2f2018-03-29 16:29:27 +0100612if(BUILD_TF_PARSER)
613 install(TARGETS armnnTfParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
614endif()
telsoa01c577f2c2018-08-31 09:22:23 +0100615if(BUILD_TF_LITE_PARSER)
616 install(TARGETS armnnTfLiteParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)
617endif()
telsoa014fcda012018-03-09 14:13:49 +0000618install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_PREFIX}/include)
619
620target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
621
622if(ARMCOMPUTENEON OR ARMCOMPUTECL)
623 target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
624endif()
625if(ARMCOMPUTECL)
626 target_link_libraries(armnn ${OPENCL_LIBRARIES})
627endif()
628
629if(PROFILING_BACKEND_STREAMLINE)
630 target_link_libraries(armnn pthread)
631endif()
632
telsoa014fcda012018-03-09 14:13:49 +0000633if(BUILD_UNIT_TESTS)
634 set(unittest_sources)
635 list(APPEND unittest_sources
636 src/armnn/test/UnitTests.cpp
637 src/armnn/test/UnitTests.hpp
638 src/armnn/test/EndToEndTest.cpp
639 src/armnn/test/UtilsTests.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100640 src/armnn/test/JsonPrinterTests.cpp
telsoa014fcda012018-03-09 14:13:49 +0000641 src/armnn/test/GraphTests.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100642 src/armnn/test/OptimizerTests.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100643 src/armnn/test/ProfilerTests.cpp
telsoa014fcda012018-03-09 14:13:49 +0000644 src/armnn/test/RuntimeTests.cpp
645 src/armnn/test/CreateWorkload.hpp
646 src/armnn/test/TensorTest.cpp
647 src/armnn/test/TensorHelpers.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100648 src/armnn/test/CsvReaderTest.cpp
649 src/armnn/test/NetworkTests.cpp
650 src/armnn/test/FloatingPointConverterTest.cpp
651 src/armnn/test/ProfilingEventTest.cpp
telsoa014fcda012018-03-09 14:13:49 +0000652 src/armnn/test/GraphUtils.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100653 src/armnn/test/InstrumentTests.cpp
654 src/armnn/test/ObservableTest.cpp
telsoa014fcda012018-03-09 14:13:49 +0000655 src/armnn/backends/test/IsLayerSupportedTest.cpp
656 src/armnn/backends/test/IsLayerSupportedTestImpl.hpp
657 src/armnn/backends/test/Reference.cpp
658 src/armnn/backends/test/WorkloadDataValidation.cpp
659 src/armnn/backends/test/TensorCopyUtils.hpp
660 src/armnn/backends/test/TensorCopyUtils.cpp
661 src/armnn/backends/test/LayerTests.hpp
662 src/armnn/backends/test/LayerTests.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100663 src/armnn/backends/test/LayerReleaseConstantDataTest.cpp
telsoa014fcda012018-03-09 14:13:49 +0000664 src/armnn/backends/test/Conv2dTestImpl.hpp
665 src/armnn/backends/test/ActivationTestImpl.hpp
666 src/armnn/backends/test/ActivationFixture.hpp
667 src/armnn/backends/test/Pooling2dTestImpl.hpp
668 src/armnn/backends/test/ReshapeTestImpl.hpp
669 src/armnn/backends/test/PermuteTestImpl.hpp
670 src/armnn/backends/test/FullyConnectedTestImpl.hpp
671 src/armnn/backends/test/SplitterTestImpl.hpp
672 src/armnn/backends/test/NormTestImpl.hpp
673 src/armnn/backends/test/BatchNormTestImpl.hpp
674 src/armnn/backends/test/WorkloadTestUtils.hpp
675 src/armnn/backends/test/CreateWorkloadRef.cpp
676 src/armnn/backends/test/QuantizeHelper.hpp)
677
678 if(ARMCOMPUTENEON)
679 list(APPEND unittest_sources
680 src/armnn/backends/test/ArmComputeNeon.cpp
681 src/armnn/backends/test/CreateWorkloadNeon.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100682 src/armnn/test/CreateWorkloadClNeon.hpp
683 src/armnn/test/NeonTimerTest.cpp)
telsoa014fcda012018-03-09 14:13:49 +0000684 endif()
685
686 if(ARMCOMPUTECL)
687 list(APPEND unittest_sources
688 src/armnn/backends/test/ArmComputeCl.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100689 src/armnn/backends/test/ClContextControlFixture.hpp
telsoa014fcda012018-03-09 14:13:49 +0000690 src/armnn/backends/test/CreateWorkloadCl.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100691 src/armnn/test/CreateWorkloadClNeon.hpp
692 src/armnn/test/OpenClTimerTest.cpp
693 src/armnn/test/FP16SupportTest.cpp)
telsoa014fcda012018-03-09 14:13:49 +0000694 endif()
695
696 if(ARMCOMPUTENEON OR ARMCOMPUTECL)
697 list(APPEND unittest_sources
698 src/armnn/backends/test/MemCopyTests.cpp)
699 endif()
700
surmeh01bceff2f2018-03-29 16:29:27 +0100701 if(BUILD_TF_PARSER)
702 list(APPEND unittest_sources
703 src/armnnTfParser/test/Activations.cpp
704 src/armnnTfParser/test/Addition.cpp
705 src/armnnTfParser/test/BiasAdd.cpp
706 src/armnnTfParser/test/BroadcastForAdd.cpp
707 src/armnnTfParser/test/Convolution2d.cpp
708 src/armnnTfParser/test/Concat.cpp
709 src/armnnTfParser/test/ConcatOfConcats.cpp
710 src/armnnTfParser/test/DepthwiseConvolution2d.cpp
711 src/armnnTfParser/test/FusedBatchNorm.cpp
712 src/armnnTfParser/test/Identity.cpp
713 src/armnnTfParser/test/LocalResponseNormalization.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100714 src/armnnTfParser/test/MaximumForLeakyRelu.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100715 src/armnnTfParser/test/Multiplication.cpp
716 src/armnnTfParser/test/MultiOutput.cpp
717 src/armnnTfParser/test/PassThru.cpp
718 src/armnnTfParser/test/Pooling.cpp
saoste01bbd40612018-08-28 15:41:51 +0100719 src/armnnTfParser/test/RealDiv.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100720 src/armnnTfParser/test/Reshape.cpp
721 src/armnnTfParser/test/ResizeBilinear.cpp
722 src/armnnTfParser/test/Shape.cpp
723 src/armnnTfParser/test/Softmax.cpp
724 src/armnnTfParser/test/TestDependencies.cpp
725 src/armnnTfParser/test/FullyConnected.cpp
726 src/armnnTfParser/test/Constant.cpp
727 src/armnnTfParser/test/TestMultiInputsOutputs.cpp
728 src/armnnTfParser/test/Squeeze.cpp)
729 endif()
730
telsoa01c577f2c2018-08-31 09:22:23 +0100731 if(BUILD_TF_LITE_PARSER)
732 list(APPEND unittest_sources
733 src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp
734 src/armnnTfLiteParser/test/AvgPool2D.cpp
735 src/armnnTfLiteParser/test/Conv2D.cpp
736 src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
737 src/armnnTfLiteParser/test/Softmax.cpp
738 src/armnnTfLiteParser/test/Squeeze.cpp
739 src/armnnTfLiteParser/test/LoadModel.cpp
740 src/armnnTfLiteParser/test/GetBuffer.cpp
741 src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
742 src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
743 src/armnnTfLiteParser/test/GetTensorIds.cpp
744 src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
745 src/armnnTfLiteParser/test/GetInputsOutputs.cpp
746 )
747 endif()
748
telsoa014fcda012018-03-09 14:13:49 +0000749 if(BUILD_CAFFE_PARSER)
750 list(APPEND unittest_sources
751 src/armnnCaffeParser/test/TestAdd.cpp
752 src/armnnCaffeParser/test/TestConcat.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100753 src/armnnCaffeParser/test/TestConvolution.cpp
telsoa014fcda012018-03-09 14:13:49 +0000754 src/armnnCaffeParser/test/TestDropout.cpp
755 src/armnnCaffeParser/test/TestInputs.cpp
756 src/armnnCaffeParser/test/TestMul.cpp
757 src/armnnCaffeParser/test/TestPooling2d.cpp
758 src/armnnCaffeParser/test/TestInPlace.cpp
759 src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
760 src/armnnCaffeParser/test/TestSplit.cpp
761 )
762 endif()
763
telsoa01c577f2c2018-08-31 09:22:23 +0100764 if(BUILD_ONNX_PARSER)
765 list(APPEND unittest_sources
766 src/armnnOnnxParser/test/Constructor.cpp
767 src/armnnOnnxParser/test/CreateNetwork.cpp
768 src/armnnOnnxParser/test/ProtoxtFixture.cpp
769 src/armnnOnnxParser/test/Const.cpp
770 src/armnnOnnxParser/test/Pooling.cpp
771 src/armnnOnnxParser/test/Reshape.cpp
772 src/armnnOnnxParser/test/Relu.cpp
773 src/armnnOnnxParser/test/Conv2D.cpp
774 src/armnnOnnxParser/test/Addition.cpp
775 src/armnnOnnxParser/test/FullyConnected.cpp
776 src/armnnOnnxParser/test/GetInputsOutputs.cpp
777 src/armnnOnnxParser/test/BatchNorm.cpp
778 src/armnnOnnxParser/test/DepthConv.cpp
779 )
780 endif()
781
telsoa014fcda012018-03-09 14:13:49 +0000782 add_executable_ex(UnitTests ${unittest_sources})
783 target_include_directories(UnitTests PRIVATE src/armnn)
784 target_include_directories(UnitTests PRIVATE src/armnnUtils)
785
telsoa01c577f2c2018-08-31 09:22:23 +0100786 if(VALGRIND_FOUND)
787 if(HEAP_PROFILING OR LEAK_CHECKING)
788 message("Valgrind is disabled for heap profiling and leak checking builds.")
789 else()
790 # Valgrind works with gperftools version number <= 2.4
791 target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
792 endif()
telsoa014fcda012018-03-09 14:13:49 +0000793 endif()
794
795 target_link_libraries(UnitTests armnn)
surmeh013537c2c2018-05-18 16:31:43 +0100796 target_link_libraries(UnitTests armnnUtils)
telsoa014fcda012018-03-09 14:13:49 +0000797 target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
telsoa01c577f2c2018-08-31 09:22:23 +0100798 target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
telsoa014fcda012018-03-09 14:13:49 +0000799
surmeh01bceff2f2018-03-29 16:29:27 +0100800 if(BUILD_TF_PARSER)
801 target_link_libraries(UnitTests armnnTfParser)
802 endif()
803
telsoa014fcda012018-03-09 14:13:49 +0000804 if(BUILD_CAFFE_PARSER)
805 target_link_libraries(UnitTests armnnCaffeParser)
806 endif()
807
telsoa01c577f2c2018-08-31 09:22:23 +0100808 if(BUILD_TF_LITE_PARSER)
809 target_link_libraries(UnitTests armnnTfLiteParser)
810 endif()
811
812 if(BUILD_ONNX_PARSER)
813 target_link_libraries(UnitTests armnnOnnxParser)
814 endif()
815
telsoa014fcda012018-03-09 14:13:49 +0000816 addDllCopyCommands(UnitTests)
817endif()