blob: 950ad4a946a04a0e462f6c9f1b12b88bde09a969 [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#
Jim Flynn3091b062019-02-15 14:45:04 +00005cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake
telsoa014fcda012018-03-09 14:13:49 +00006project(armnn)
7
8set(additional_cmake_files)
9list(APPEND additional_cmake_files
Guillaume Gardet5ae78292019-10-15 08:47:26 +020010 cmake/ArmnnVersion.cmake
telsoa014fcda012018-03-09 14:13:49 +000011 cmake/Utils.cmake
12 cmake/GlobalConfig.cmake
13 cmake/AddDllCopyCommands.cmake)
14
15foreach(cmake_file ${additional_cmake_files})
16 include(${cmake_file})
17endforeach()
18
Matteo Martincighe7d44982019-08-05 12:16:47 +010019if (DYNAMIC_BACKEND_PATHS)
20 # It's expected to have the format: DYNAMIC_BACKEND_PATHS="PATH_1:PATH_2...:PATH_N"
21 add_definitions('-DDYNAMIC_BACKEND_PATHS="${DYNAMIC_BACKEND_PATHS}"')
22endif()
23
Matthew Benthama7b79282019-06-03 13:00:50 +010024include(GNUInstallDirs)
25
Ruomei Yandb7c3912019-03-08 17:50:25 +000026add_subdirectory(samples)
27add_subdirectory(src/armnnTfLiteParser)
28add_subdirectory(src/armnnSerializer)
29
telsoa014fcda012018-03-09 14:13:49 +000030if (BUILD_TESTS)
31 add_subdirectory(tests)
32endif()
33
34# Include the additional cmake files in their own target so that they will appear nicely in IDEs
35add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files})
36
37set(armnnUtils_sources)
38list(APPEND armnnUtils_sources
Matteo Martincighe011d202019-11-28 11:35:47 +000039 include/armnnUtils/DataLayoutIndexed.hpp
40 include/armnnUtils/Permute.hpp
41 include/armnnUtils/FloatingPointConverter.hpp
42 include/armnnUtils/TensorUtils.hpp
Rob Hughesbdee4262020-01-07 17:05:24 +000043 src/armnnUtils/Filesystem.hpp
44 src/armnnUtils/Filesystem.cpp
45 src/armnnUtils/Processes.hpp
46 src/armnnUtils/Processes.cpp
telsoa014fcda012018-03-09 14:13:49 +000047 src/armnnUtils/GraphTopologicalSort.hpp
arovir01616e7752018-10-01 17:08:59 +010048 src/armnnUtils/Half.hpp
telsoa014fcda012018-03-09 14:13:49 +000049 src/armnnUtils/Permute.cpp
Matteo Martincigh21350152018-11-28 16:22:22 +000050 src/armnnUtils/DataLayoutIndexed.cpp
surmeh01bceff2f2018-03-29 16:29:27 +010051 src/armnnUtils/DotSerializer.cpp
52 src/armnnUtils/DotSerializer.hpp
surmeh013537c2c2018-05-18 16:31:43 +010053 src/armnnUtils/HeapProfiling.cpp
54 src/armnnUtils/HeapProfiling.hpp
55 src/armnnUtils/LeakChecking.cpp
56 src/armnnUtils/LeakChecking.hpp
Éanna Ó Catháina4247d52019-05-08 14:00:45 +010057 src/armnnUtils/ModelAccuracyChecker.cpp
58 src/armnnUtils/ModelAccuracyChecker.hpp
telsoa01c577f2c2018-08-31 09:22:23 +010059 src/armnnUtils/CsvReader.cpp
60 src/armnnUtils/CsvReader.hpp
61 src/armnnUtils/FloatingPointConverter.cpp
telsoa01c577f2c2018-08-31 09:22:23 +010062 src/armnnUtils/VerificationHelpers.hpp
63 src/armnnUtils/VerificationHelpers.cpp
Sadik Armagan479045b2018-10-01 11:51:37 +010064 src/armnnUtils/ParserHelper.hpp
65 src/armnnUtils/ParserHelper.cpp
Matthew Bentham4057d912019-01-21 15:45:51 +000066 src/armnnUtils/ParserPrototxtFixture.hpp
67 src/armnnUtils/PrototxtConversions.hpp
68 src/armnnUtils/PrototxtConversions.cpp
Aron Virginas-Tar48623a02019-10-22 10:00:28 +010069 src/armnnUtils/QuantizeHelper.hpp
Jim Flynn2fd61002019-05-03 12:54:26 +010070 src/armnnUtils/TensorIOUtils.hpp
Nina Drozdd41b2592018-11-19 13:03:36 +000071 src/armnnUtils/TensorUtils.cpp
Rob Hughes25b74362020-01-13 11:14:59 +000072 src/armnnUtils/NetworkSockets.hpp
73 src/armnnUtils/NetworkSockets.cpp
telsoa014fcda012018-03-09 14:13:49 +000074 )
telsoa014fcda012018-03-09 14:13:49 +000075
76add_library_ex(armnnUtils STATIC ${armnnUtils_sources})
Jim Flynnf92dfce2019-05-02 11:33:25 +010077target_include_directories(armnnUtils PRIVATE src/backends)
telsoa014fcda012018-03-09 14:13:49 +000078
79if(BUILD_CAFFE_PARSER)
80 # ArmNN Parser source files required for all build options
81 set(armnn_caffe_parser_sources)
82 list(APPEND armnn_caffe_parser_sources
83 include/armnnCaffeParser/ICaffeParser.hpp
telsoa01c577f2c2018-08-31 09:22:23 +010084 src/armnnCaffeParser/RecordByRecordCaffeParser.hpp
85 src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
telsoa014fcda012018-03-09 14:13:49 +000086 src/armnnCaffeParser/CaffeParser.hpp
87 src/armnnCaffeParser/CaffeParser.cpp
88 ${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc
89 )
90 # The generated Caffe protobuf .cc file is not warning clean and we can't fix them.
91 if(COMPILER_IS_GNU_LIKE)
92 set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
93 endif()
94
95 add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})
96 set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")
97
98 target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)
99
Derek Lamberti08446972019-11-26 16:38:31 +0000100 target_link_libraries(armnnCaffeParser ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})
telsoa014fcda012018-03-09 14:13:49 +0000101
102 target_link_libraries(armnnCaffeParser armnn)
103 target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})
Matteo Martincighbccc3452019-10-18 14:59:19 +0100104 set_target_properties(armnnCaffeParser PROPERTIES VERSION ${GENERIC_LIB_VERSION} SOVERSION ${GENERIC_LIB_SOVERSION})
telsoa01c577f2c2018-08-31 09:22:23 +0100105
106endif()
107
108if(BUILD_ONNX_PARSER)
109 set(armnn_onnx_parser_sources)
110 list(APPEND armnn_onnx_parser_sources
111 include/armnnOnnxParser/IOnnxParser.hpp
112 src/armnnOnnxParser/OnnxParser.hpp
113 src/armnnOnnxParser/OnnxParser.cpp
114 ${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc
115 )
116 # The generated onnx protobuf .cc files are not warning clean and we can't fix them.
117 if(COMPILER_IS_GNU_LIKE)
118 set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")
119 endif()
120
121 add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})
122
123 target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)
124
125 target_link_libraries(armnnOnnxParser armnn)
126
127 # Protobuf
128 target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})
Matteo Martincighbccc3452019-10-18 14:59:19 +0100129 set_target_properties(armnnOnnxParser PROPERTIES VERSION ${GENERIC_LIB_VERSION} SOVERSION ${GENERIC_LIB_SOVERSION})
telsoa014fcda012018-03-09 14:13:49 +0000130endif()
131
surmeh01bceff2f2018-03-29 16:29:27 +0100132if(BUILD_TF_PARSER)
133 set(armnn_tf_parser_sources)
134 list(APPEND armnn_tf_parser_sources
135 include/armnnTfParser/ITfParser.hpp
136 src/armnnTfParser/TfParser.hpp
137 src/armnnTfParser/TfParser.cpp
138 ${TF_PROTOBUFS}
139 )
140 # The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.
141 if(COMPILER_IS_GNU_LIKE)
Derek Lambertiba25aee2019-12-10 22:20:54 +0000142 set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-unused-variable -Wno-unused-parameter -Wno-conversion -Wno-sign-conversion")
surmeh01bceff2f2018-03-29 16:29:27 +0100143 endif()
144
145 add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})
146
147 target_include_directories(armnnTfParser PRIVATE src/armnnUtils)
148
149 target_link_libraries(armnnTfParser armnn)
150
151 # Protobuf (use the specific version tensorflow wants)
152 target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})
Matteo Martincighbccc3452019-10-18 14:59:19 +0100153 set_target_properties(armnnTfParser PROPERTIES VERSION ${GENERIC_LIB_VERSION} SOVERSION ${GENERIC_LIB_SOVERSION})
surmeh01bceff2f2018-03-29 16:29:27 +0100154endif()
155
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100156if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
Jim Flynn3091b062019-02-15 14:45:04 +0000157 if(NOT BUILD_ARMNN_SERIALIZER)
Ruomei Yandb7c3912019-03-08 17:50:25 +0000158 message(ERROR, "In order to build the ArmNN Quantization Tool you must set BUILD_ARMNN_SERIALIZER = YES")
Jim Flynn3091b062019-02-15 14:45:04 +0000159 endif()
160
Nina Drozd59e15b02019-04-25 15:45:20 +0100161 set(armnn_quantizer_sources)
162 list(APPEND armnn_quantizer_sources
163 src/armnnQuantizer/QuantizationDataSet.hpp
164 src/armnnQuantizer/QuantizationDataSet.cpp
165 src/armnnQuantizer/QuantizationInput.hpp
166 src/armnnQuantizer/QuantizationInput.cpp
Jim Flynn3091b062019-02-15 14:45:04 +0000167 src/armnnQuantizer/CommandLineProcessor.hpp
168 src/armnnQuantizer/CommandLineProcessor.cpp
Nina Drozd59e15b02019-04-25 15:45:20 +0100169 )
170
171 add_library_ex(armnnQuantizer SHARED ${armnn_quantizer_sources})
172
173 target_include_directories(armnnQuantizer PRIVATE include/armnnDeserializer)
174 target_include_directories(armnnQuantizer PRIVATE src/armnnUtils)
175 target_include_directories(armnnQuantizer PRIVATE src/armnn)
176
177 include_directories(SYSTEM "${FLATBUFFERS_INCLUDE_PATH}")
178 set_target_properties(armnnQuantizer PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR})
179
180 target_link_libraries(armnnQuantizer
Guillaume GARDETe315c8f2019-06-05 10:58:47 +0200181 armnn
Nina Drozd59e15b02019-04-25 15:45:20 +0100182 ${Boost_SYSTEM_LIBRARY}
183 ${Boost_PROGRAM_OPTIONS_LIBRARY}
184 ${Boost_FILESYSTEM_LIBRARY}
Nina Drozd59e15b02019-04-25 15:45:20 +0100185 ${Boost_THREAD_LIBRARY} )
186
187 add_executable_ex(ArmnnQuantizer
Jim Flynn3091b062019-02-15 14:45:04 +0000188 src/armnnQuantizer/ArmNNQuantizerMain.cpp)
189
Ruomei Yandb7c3912019-03-08 17:50:25 +0000190 target_include_directories(ArmnnQuantizer PRIVATE include/armnnDeserializer)
Nina Drozd59e15b02019-04-25 15:45:20 +0100191 target_include_directories(ArmnnQuantizer PRIVATE src/armnn)
Jim Flynn3091b062019-02-15 14:45:04 +0000192
193 target_link_libraries(ArmnnQuantizer
194 ${Boost_SYSTEM_LIBRARY}
195 ${Boost_PROGRAM_OPTIONS_LIBRARY}
196 ${Boost_FILESYSTEM_LIBRARY}
Jim Flynn3091b062019-02-15 14:45:04 +0000197 ${Boost_THREAD_LIBRARY} )
198
199 target_link_libraries(ArmnnQuantizer
Nina Drozd59e15b02019-04-25 15:45:20 +0100200 armnnQuantizer
Jim Flynn3091b062019-02-15 14:45:04 +0000201 armnnSerializer
202 armnn
Nina Drozd59e15b02019-04-25 15:45:20 +0100203 armnnUtils
Jim Flynn3091b062019-02-15 14:45:04 +0000204 ${FLATBUFFERS_LIBRARY})
205
206 if(Threads_FOUND AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android)))
207 target_link_libraries(ArmnnQuantizer pthread)
208 endif()
209
Matteo Martincigh7be81f62019-10-21 11:33:50 +0100210 set_target_properties(armnnQuantizer PROPERTIES VERSION ${GENERIC_LIB_VERSION} SOVERSION ${GENERIC_LIB_SOVERSION})
Matteo Martincighbccc3452019-10-18 14:59:19 +0100211
Jim Flynn3091b062019-02-15 14:45:04 +0000212endif()
213
214
telsoa014fcda012018-03-09 14:13:49 +0000215list(APPEND armnn_sources
216 include/armnn/ArmNN.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100217 include/armnn/BackendHelper.hpp
David Beck9df2d952018-10-10 15:11:44 +0100218 include/armnn/BackendId.hpp
Derek Lamberti836b27b2019-11-20 10:51:57 +0000219 include/armnn/BackendOptions.hpp
Matteo Martincighc601aa62019-10-29 15:03:22 +0000220 include/armnn/BackendRegistry.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100221 include/armnn/Conversion.hpp
222 include/armnn/Deprecated.hpp
telsoa014fcda012018-03-09 14:13:49 +0000223 include/armnn/Descriptors.hpp
224 include/armnn/DescriptorsFwd.hpp
David Beck9df2d952018-10-10 15:11:44 +0100225 include/armnn/Exceptions.hpp
David Becke97c6e02018-10-03 13:09:28 +0100226 include/armnn/ILayerSupport.hpp
jimfly015f4e41f2019-01-23 16:10:17 +0000227 include/armnn/ILayerVisitor.hpp
telsoa014fcda012018-03-09 14:13:49 +0000228 include/armnn/INetwork.hpp
David Beck9df2d952018-10-10 15:11:44 +0100229 include/armnn/IProfiler.hpp
230 include/armnn/IRuntime.hpp
231 include/armnn/LayerSupport.hpp
Mike Kelly8c1701a2019-02-11 17:01:27 +0000232 include/armnn/LayerVisitorBase.hpp
Derek Lamberti08446972019-11-26 16:38:31 +0000233 include/armnn/Logging.hpp
David Beck9df2d952018-10-10 15:11:44 +0100234 include/armnn/LstmParams.hpp
Derek Lambertif674aa02019-08-01 15:56:25 +0100235 include/armnn/MemorySources.hpp
David Beck9df2d952018-10-10 15:11:44 +0100236 include/armnn/NetworkFwd.hpp
David Beck5eec11d2018-10-04 15:43:17 +0100237 include/armnn/Optional.hpp
James Conroyee18dc82019-07-17 11:27:46 +0100238 include/armnn/QuantizedLstmParams.hpp
telsoa014fcda012018-03-09 14:13:49 +0000239 include/armnn/Tensor.hpp
240 include/armnn/TensorFwd.hpp
241 include/armnn/Types.hpp
242 include/armnn/TypesUtils.hpp
243 include/armnn/Utils.hpp
telsoa014fcda012018-03-09 14:13:49 +0000244 include/armnn/Version.hpp
Colm Donelan5ccb33d2020-01-24 16:27:02 +0000245 include/armnn/profiling/IProfilingGuidGenerator.hpp
246 include/armnn/profiling/ISendTimelinePacket.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100247 src/armnn/layers/LayerCloneBase.hpp
248 src/armnn/layers/LayerWithParameters.hpp
249 src/armnn/layers/ActivationLayer.hpp
250 src/armnn/layers/ActivationLayer.cpp
251 src/armnn/layers/AdditionLayer.hpp
252 src/armnn/layers/AdditionLayer.cpp
Nikhil Rajee391d52019-09-05 17:50:44 +0100253 src/armnn/layers/ArgMinMaxLayer.hpp
254 src/armnn/layers/ArgMinMaxLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100255 src/armnn/layers/BatchNormalizationLayer.hpp
256 src/armnn/layers/BatchNormalizationLayer.cpp
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000257 src/armnn/layers/BatchToSpaceNdLayer.hpp
258 src/armnn/layers/BatchToSpaceNdLayer.cpp
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100259 src/armnn/layers/ComparisonLayer.hpp
260 src/armnn/layers/ComparisonLayer.cpp
Jim Flynne242f2d2019-05-22 14:24:13 +0100261 src/armnn/layers/ConcatLayer.hpp
262 src/armnn/layers/ConcatLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100263 src/armnn/layers/ConstantLayer.hpp
264 src/armnn/layers/ConstantLayer.cpp
265 src/armnn/layers/Convolution2dLayer.hpp
266 src/armnn/layers/Convolution2dLayer.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100267 src/armnn/layers/ConvertFp16ToFp32Layer.hpp
268 src/armnn/layers/ConvertFp16ToFp32Layer.cpp
269 src/armnn/layers/ConvertFp32ToFp16Layer.hpp
270 src/armnn/layers/ConvertFp32ToFp16Layer.cpp
Nattapat Chaimanowonga9a1cf12018-12-03 16:06:49 +0000271 src/armnn/layers/DebugLayer.hpp
272 src/armnn/layers/DebugLayer.cpp
Aron Virginas-Tardd6247f2019-09-19 14:31:17 +0100273 src/armnn/layers/DepthToSpaceLayer.hpp
274 src/armnn/layers/DepthToSpaceLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100275 src/armnn/layers/DepthwiseConvolution2dLayer.hpp
276 src/armnn/layers/DepthwiseConvolution2dLayer.cpp
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +0000277 src/armnn/layers/DequantizeLayer.hpp
278 src/armnn/layers/DequantizeLayer.cpp
Narumol Prangnawarat94dd5d82019-01-23 18:06:26 +0000279 src/armnn/layers/DetectionPostProcessLayer.hpp
280 src/armnn/layers/DetectionPostProcessLayer.cpp
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000281 src/armnn/layers/ElementwiseBaseLayer.hpp
282 src/armnn/layers/ElementwiseBaseLayer.cpp
josh minor4a3c6102020-01-06 16:40:46 -0600283 src/armnn/layers/ElementwiseUnaryLayer.hpp
284 src/armnn/layers/ElementwiseUnaryLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100285 src/armnn/layers/FakeQuantizationLayer.hpp
286 src/armnn/layers/FakeQuantizationLayer.cpp
287 src/armnn/layers/FloorLayer.hpp
288 src/armnn/layers/FloorLayer.cpp
289 src/armnn/layers/FullyConnectedLayer.hpp
290 src/armnn/layers/FullyConnectedLayer.cpp
narpra01b89b05f2019-01-16 09:53:09 +0000291 src/armnn/layers/GatherLayer.cpp
292 src/armnn/layers/GatherLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100293 src/armnn/layers/InputLayer.hpp
294 src/armnn/layers/InputLayer.cpp
Kevin Mayce5045a2019-10-02 14:07:47 +0100295 src/armnn/layers/InstanceNormalizationLayer.hpp
296 src/armnn/layers/InstanceNormalizationLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100297 src/armnn/layers/L2NormalizationLayer.hpp
298 src/armnn/layers/L2NormalizationLayer.cpp
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +0100299 src/armnn/layers/LogSoftmaxLayer.hpp
300 src/armnn/layers/LogSoftmaxLayer.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100301 src/armnn/layers/LstmLayer.cpp
302 src/armnn/layers/LstmLayer.hpp
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000303 src/armnn/layers/MaximumLayer.cpp
304 src/armnn/layers/MaximumLayer.hpp
narpra0132b90462018-09-13 11:07:48 +0100305 src/armnn/layers/MeanLayer.hpp
306 src/armnn/layers/MeanLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100307 src/armnn/layers/MemCopyLayer.hpp
308 src/armnn/layers/MemCopyLayer.cpp
Derek Lambertif674aa02019-08-01 15:56:25 +0100309 src/armnn/layers/MemImportLayer.hpp
310 src/armnn/layers/MemImportLayer.cpp
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +0100311 src/armnn/layers/MergeLayer.hpp
312 src/armnn/layers/MergeLayer.cpp
kevmay0190539692018-11-29 08:40:19 +0000313 src/armnn/layers/MinimumLayer.cpp
314 src/armnn/layers/MinimumLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100315 src/armnn/layers/MultiplicationLayer.hpp
316 src/armnn/layers/MultiplicationLayer.cpp
317 src/armnn/layers/NormalizationLayer.hpp
318 src/armnn/layers/NormalizationLayer.cpp
319 src/armnn/layers/OutputLayer.hpp
320 src/armnn/layers/OutputLayer.cpp
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100321 src/armnn/layers/PadLayer.hpp
322 src/armnn/layers/PadLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100323 src/armnn/layers/PermuteLayer.hpp
324 src/armnn/layers/PermuteLayer.cpp
325 src/armnn/layers/Pooling2dLayer.hpp
326 src/armnn/layers/Pooling2dLayer.cpp
Derek Lambertia9cca6a2019-03-25 15:41:58 +0000327 src/armnn/layers/QuantizeLayer.cpp
328 src/armnn/layers/QuantizeLayer.hpp
James Conroyee18dc82019-07-17 11:27:46 +0100329 src/armnn/layers/QuantizedLstmLayer.hpp
330 src/armnn/layers/QuantizedLstmLayer.cpp
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100331 src/armnn/layers/DivisionLayer.cpp
332 src/armnn/layers/DivisionLayer.hpp
Matteo Martincigh49124022019-01-11 13:25:59 +0000333 src/armnn/layers/PreCompiledLayer.hpp
334 src/armnn/layers/PreCompiledLayer.cpp
Matteo Martincigh0e406ee2019-06-12 15:42:18 +0100335 src/armnn/layers/PreluLayer.hpp
336 src/armnn/layers/PreluLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100337 src/armnn/layers/ReshapeLayer.hpp
338 src/armnn/layers/ReshapeLayer.cpp
Teresa Charlina9075df2019-06-27 15:41:57 +0100339 src/armnn/layers/ResizeLayer.hpp
340 src/armnn/layers/ResizeLayer.cpp
Aron Virginas-Tar636ab402019-09-16 14:27:45 +0100341 src/armnn/layers/SliceLayer.cpp
342 src/armnn/layers/SliceLayer.hpp
surmeh013537c2c2018-05-18 16:31:43 +0100343 src/armnn/layers/SoftmaxLayer.hpp
344 src/armnn/layers/SoftmaxLayer.cpp
Matteo Martincigh0e406ee2019-06-12 15:42:18 +0100345 src/armnn/layers/SpaceToBatchNdLayer.hpp
346 src/armnn/layers/SpaceToBatchNdLayer.cpp
347 src/armnn/layers/SpaceToDepthLayer.hpp
348 src/armnn/layers/SpaceToDepthLayer.cpp
surmeh013537c2c2018-05-18 16:31:43 +0100349 src/armnn/layers/SplitterLayer.hpp
350 src/armnn/layers/SplitterLayer.cpp
Matthew Jackson2b8c1da2019-07-04 14:59:16 +0100351 src/armnn/layers/StackLayer.hpp
352 src/armnn/layers/StackLayer.cpp
Derek Lamberti013c3902019-10-21 10:46:16 +0100353 src/armnn/layers/StandInLayer.cpp
354 src/armnn/layers/StandInLayer.hpp
Conor Kennedy430b5d82018-11-14 15:28:28 +0000355 src/armnn/layers/StridedSliceLayer.cpp
356 src/armnn/layers/StridedSliceLayer.hpp
David Beckc2044fe2018-09-05 15:00:38 +0100357 src/armnn/layers/SubtractionLayer.cpp
358 src/armnn/layers/SubtractionLayer.hpp
Sadik Armaganeff363d2019-04-05 15:25:46 +0100359 src/armnn/layers/SwitchLayer.cpp
360 src/armnn/layers/SwitchLayer.hpp
Aron Virginas-Tar639fb042019-06-20 14:28:19 +0100361 src/armnn/layers/TransposeConvolution2dLayer.cpp
362 src/armnn/layers/TransposeConvolution2dLayer.hpp
Matteo Martincighc601aa62019-10-29 15:03:22 +0000363 src/armnn/BackendRegistry.cpp
Matteo Martincigh49124022019-01-11 13:25:59 +0000364 src/armnn/BackendSettings.hpp
Ferran Balaguerf54c9462019-07-10 12:43:58 +0100365 src/armnn/BackendHelper.cpp
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +0000366 src/armnn/CompatibleTypes.hpp
telsoa014fcda012018-03-09 14:13:49 +0000367 src/armnn/Descriptors.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100368 src/armnn/DeviceSpec.hpp
Rob Hughesa91479c2020-01-07 09:40:12 +0000369 src/armnn/DllExport.hpp
Jim Flynnf92dfce2019-05-02 11:33:25 +0100370 src/armnn/DynamicQuantizationVisitor.cpp
371 src/armnn/DynamicQuantizationVisitor.hpp
telsoa014fcda012018-03-09 14:13:49 +0000372 src/armnn/Exceptions.cpp
Teresa Charlin4de9f672019-04-10 13:59:49 +0100373 src/armnn/ExecutionFrame.cpp
374 src/armnn/ExecutionFrame.hpp
telsoa014fcda012018-03-09 14:13:49 +0000375 src/armnn/Graph.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100376 src/armnn/Graph.hpp
377 src/armnn/IGraphObservable.hpp
378 src/armnn/Instrument.hpp
379 src/armnn/InternalTypes.cpp
380 src/armnn/InternalTypes.hpp
Derek Lambertiff05cc52019-04-26 13:05:17 +0100381 src/armnn/ISubgraphViewConverter.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100382 src/armnn/JsonPrinter.cpp
383 src/armnn/JsonPrinter.hpp
384 src/armnn/Layer.cpp
385 src/armnn/LayerFwd.hpp
386 src/armnn/Layer.hpp
387 src/armnn/LayersFwd.hpp
388 src/armnn/LayerSupportCommon.hpp
389 src/armnn/LayerSupport.cpp
390 src/armnn/LoadedNetwork.cpp
391 src/armnn/LoadedNetwork.hpp
Matthew Benthamd3a07832020-01-15 17:21:38 +0000392 src/armnn/Logging.cpp
telsoa014fcda012018-03-09 14:13:49 +0000393 src/armnn/Network.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100394 src/armnn/Network.hpp
Nattapat Chaimanowong7ac07f32019-03-20 11:51:14 +0000395 src/armnn/NetworkQuantizationScheme.hpp
Derek Lamberti27d83072019-02-05 16:00:08 +0000396 src/armnn/NetworkQuantizer.cpp
397 src/armnn/NetworkQuantizer.hpp
Matteo Martincigha8d572d2019-02-07 17:51:09 +0000398 src/armnn/NetworkQuantizerUtils.cpp
399 src/armnn/NetworkQuantizerUtils.hpp
Aron Virginas-Tar5fdfe822018-11-21 13:14:42 +0000400 src/armnn/NetworkUtils.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100401 src/armnn/NetworkUtils.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100402 src/armnn/Observable.cpp
403 src/armnn/Observable.hpp
404 src/armnn/Optimizer.cpp
405 src/armnn/Optimizer.hpp
Matteo Martincighe5b8eb92019-11-28 15:45:42 +0000406 src/armnn/OutputHandler.cpp
407 src/armnn/OutputHandler.hpp
Matteo Martincigha8d572d2019-02-07 17:51:09 +0000408 src/armnn/OverrideInputRangeVisitor.cpp
409 src/armnn/OverrideInputRangeVisitor.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100410 src/armnn/Profiling.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100411 src/armnn/ProfilingEvent.cpp
412 src/armnn/ProfilingEvent.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100413 src/armnn/Profiling.hpp
Derek Lamberti27d83072019-02-05 16:00:08 +0000414 src/armnn/QuantizerVisitor.cpp
415 src/armnn/QuantizerVisitor.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100416 src/armnn/Runtime.cpp
417 src/armnn/Runtime.hpp
Derek Lamberti8a4ca102019-02-08 17:54:20 +0000418 src/armnn/RangeTracker.cpp
419 src/armnn/RangeTracker.hpp
Aron Virginas-Tard4f0fea2019-04-09 14:08:06 +0100420 src/armnn/ResolveType.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100421 src/armnn/SerializeLayerParameters.cpp
422 src/armnn/SerializeLayerParameters.hpp
Derek Lamberti27d83072019-02-05 16:00:08 +0000423 src/armnn/StaticRangeVisitor.cpp
424 src/armnn/StaticRangeVisitor.hpp
Derek Lambertiff05cc52019-04-26 13:05:17 +0100425 src/armnn/SubgraphView.cpp
426 src/armnn/SubgraphView.hpp
427 src/armnn/SubgraphViewSelector.cpp
428 src/armnn/SubgraphViewSelector.hpp
telsoa014fcda012018-03-09 14:13:49 +0000429 src/armnn/Tensor.cpp
Aron Virginas-Tard4f0fea2019-04-09 14:08:06 +0100430 src/armnn/TypesUtils.cpp
telsoa014fcda012018-03-09 14:13:49 +0000431 src/armnn/Utils.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100432 src/armnn/WallClockTimer.cpp
433 src/armnn/WallClockTimer.hpp
Nina Drozd861985f2019-04-18 14:48:51 +0100434 src/armnn/optimizations/AddDebug.hpp
telsoa014fcda012018-03-09 14:13:49 +0000435 src/armnn/optimizations/All.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100436 src/armnn/optimizations/ConvertConstants.hpp
Nina Drozd861985f2019-04-18 14:48:51 +0100437 src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp
438 src/armnn/optimizations/FoldPadIntoConvolution2d.hpp
telsoa014fcda012018-03-09 14:13:49 +0000439 src/armnn/optimizations/MovePermuteUp.hpp
440 src/armnn/optimizations/Optimization.hpp
441 src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp
Nina Drozd861985f2019-04-18 14:48:51 +0100442 src/armnn/optimizations/OptimizeInverseConversions.hpp
telsoa014fcda012018-03-09 14:13:49 +0000443 src/armnn/optimizations/OptimizeInversePermutes.hpp
Rob Hughes3a7d3a72019-09-24 16:59:56 +0100444 src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp
445 src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.cpp
telsoa014fcda012018-03-09 14:13:49 +0000446 src/armnn/optimizations/PermuteAsReshape.hpp
447 src/armnn/optimizations/SquashEqualSiblings.hpp
Narumol Prangnawarat7be47ef2019-09-27 18:00:11 +0100448 src/profiling/BufferManager.cpp
449 src/profiling/BufferManager.hpp
Matteo Martincigh8a837172019-10-04 17:01:07 +0100450 src/profiling/CommandHandler.cpp
451 src/profiling/CommandHandler.hpp
Francis Murtagh11f99b42019-08-16 11:28:52 +0100452 src/profiling/CommandHandlerFunctor.cpp
453 src/profiling/CommandHandlerFunctor.hpp
Francis Murtagh1f7db452019-08-14 09:49:34 +0100454 src/profiling/CommandHandlerKey.cpp
455 src/profiling/CommandHandlerKey.hpp
Francis Murtagh94d79152019-08-16 17:45:07 +0100456 src/profiling/CommandHandlerRegistry.cpp
457 src/profiling/CommandHandlerRegistry.hpp
Sadik Armaganb5f01b22019-09-18 17:29:00 +0100458 src/profiling/ConnectionAcknowledgedCommandHandler.cpp
459 src/profiling/ConnectionAcknowledgedCommandHandler.hpp
Aron Virginas-Tar4e5fc1f2019-08-22 18:10:52 +0100460 src/profiling/CounterDirectory.cpp
461 src/profiling/CounterDirectory.hpp
David Monahande803072020-01-30 12:44:23 +0000462 src/profiling/CounterIdMap.cpp
463 src/profiling/CounterIdMap.hpp
Keith Davis3201eea2019-10-24 17:30:41 +0100464 src/profiling/DirectoryCaptureCommandHandler.cpp
465 src/profiling/DirectoryCaptureCommandHandler.hpp
Nikhil Rajd88e47c2019-08-19 10:04:23 +0100466 src/profiling/EncodeVersion.hpp
Keith Davis3201eea2019-10-24 17:30:41 +0100467 src/profiling/FileOnlyProfilingConnection.cpp
468 src/profiling/FileOnlyProfilingConnection.hpp
Francis Murtagh68f78d82019-09-04 16:42:29 +0100469 src/profiling/Holder.cpp
470 src/profiling/Holder.hpp
Narumol Prangnawarat404b2752019-09-24 17:23:16 +0100471 src/profiling/IBufferManager.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100472 src/profiling/ICounterDirectory.hpp
Jim Flynn97897022020-02-02 12:52:59 +0000473 src/profiling/ICounterRegistry.hpp
Matteo Martincighe0e6efc2019-10-04 17:17:42 +0100474 src/profiling/ICounterValues.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100475 src/profiling/ISendCounterPacket.hpp
Narumol Prangnawarat404b2752019-09-24 17:23:16 +0100476 src/profiling/IPacketBuffer.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100477 src/profiling/IPeriodicCounterCapture.hpp
Aron Virginas-Tar1a0f6912019-08-23 15:18:44 +0100478 src/profiling/IProfilingConnection.hpp
Matteo Martincigh54fb9572019-10-02 12:50:57 +0100479 src/profiling/IProfilingConnectionFactory.hpp
Jim Flynnab845752019-10-25 13:17:30 +0100480 src/profiling/LabelsAndEventClasses.cpp
481 src/profiling/LabelsAndEventClasses.hpp
Nikhil Rajbc626052019-08-15 15:49:45 +0100482 src/profiling/Packet.hpp
Narumol Prangnawarat7be47ef2019-09-27 18:00:11 +0100483 src/profiling/PacketBuffer.cpp
484 src/profiling/PacketBuffer.hpp
Aron Virginas-Tare898db92019-08-22 12:56:34 +0100485 src/profiling/PacketVersionResolver.cpp
486 src/profiling/PacketVersionResolver.hpp
Francis Murtaghfcb8ef62019-09-20 15:40:09 +0100487 src/profiling/PeriodicCounterCapture.hpp
488 src/profiling/PeriodicCounterCapture.cpp
Ferran Balaguer1b941722019-08-28 16:57:18 +0100489 src/profiling/PeriodicCounterSelectionCommandHandler.cpp
490 src/profiling/PeriodicCounterSelectionCommandHandler.hpp
Matteo Martincigh994b5342019-10-11 17:19:56 +0100491 src/profiling/PerJobCounterSelectionCommandHandler.cpp
492 src/profiling/PerJobCounterSelectionCommandHandler.hpp
Aron Virginas-Tardfa14772019-09-24 18:24:47 +0100493 src/profiling/ProfilingConnectionDumpToFileDecorator.cpp
494 src/profiling/ProfilingConnectionDumpToFileDecorator.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100495 src/profiling/ProfilingConnectionFactory.cpp
496 src/profiling/ProfilingConnectionFactory.hpp
Narumol Prangnawarat15effd82019-10-22 14:17:11 +0100497 src/profiling/ProfilingGuidGenerator.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100498 src/profiling/ProfilingService.cpp
499 src/profiling/ProfilingService.hpp
500 src/profiling/ProfilingStateMachine.cpp
501 src/profiling/ProfilingStateMachine.hpp
502 src/profiling/ProfilingUtils.cpp
503 src/profiling/ProfilingUtils.hpp
Jim Flynn97897022020-02-02 12:52:59 +0000504 src/profiling/RegisterBackendCounters.cpp
505 src/profiling/RegisterBackendCounters.hpp
Narumol Prangnawarat48033692019-09-20 12:04:55 +0100506 src/profiling/RequestCounterDirectoryCommandHandler.cpp
507 src/profiling/RequestCounterDirectoryCommandHandler.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100508 src/profiling/SendCounterPacket.cpp
509 src/profiling/SendCounterPacket.hpp
Sadik Armagan7bbdf9d2019-10-24 10:26:05 +0100510 src/profiling/SendTimelinePacket.cpp
511 src/profiling/SendTimelinePacket.hpp
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100512 src/profiling/SocketProfilingConnection.cpp
513 src/profiling/SocketProfilingConnection.hpp
Sadik Armagan7bbdf9d2019-10-24 10:26:05 +0100514 src/profiling/TimelinePacketWriterFactory.cpp
515 src/profiling/TimelinePacketWriterFactory.hpp
Matteo Martincigh830101c2019-10-22 11:07:45 +0100516 src/profiling/TimelineUtilityMethods.cpp
517 src/profiling/TimelineUtilityMethods.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100518 third-party/half/half.hpp
telsoa014fcda012018-03-09 14:13:49 +0000519 )
520
telsoa014fcda012018-03-09 14:13:49 +0000521# Files used for Streamline-based profiling backend
522if(PROFILING_BACKEND_STREAMLINE)
523 list(APPEND armnn_sources
524 ${GATOR_ROOT}/annotate/streamline_annotate.h
525 ${GATOR_ROOT}/annotate/streamline_annotate.c)
526endif()
527
David Beck10b4dfd2018-09-19 12:03:20 +0100528# the backends under src/backends extend the list of
David Beck3cc9a622018-10-12 10:38:31 +0100529# object libs armnn to include in the build
David Beck10b4dfd2018-09-19 12:03:20 +0100530include(src/backends/backends.cmake)
David Beck3cc9a622018-10-12 10:38:31 +0100531foreach(lib ${armnnLibraries})
Rob Hughes721b82f2019-11-15 09:04:17 +0000532 message(STATUS "Adding object library dependency to armnn: ${lib}")
David Beck3cc9a622018-10-12 10:38:31 +0100533 list(APPEND armnn_sources $<TARGET_OBJECTS:${lib}>)
534endforeach()
535
536add_library_ex(armnn SHARED ${armnn_sources})
David Beck10b4dfd2018-09-19 12:03:20 +0100537
Rob Hughesa91479c2020-01-07 09:40:12 +0000538target_compile_definitions(armnn PRIVATE "ARMNN_COMPILING_DLL")
539
telsoa014fcda012018-03-09 14:13:49 +0000540target_include_directories(armnn PRIVATE src/armnn)
541target_include_directories(armnn PRIVATE src/armnnUtils)
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000542target_include_directories(armnn PRIVATE src/backends)
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100543target_include_directories(armnn PRIVATE src/profiling)
David Beck3cc9a622018-10-12 10:38:31 +0100544
545target_link_libraries(armnn armnnUtils)
546
telsoa014fcda012018-03-09 14:13:49 +0000547target_link_libraries(armnn ${CMAKE_DL_LIBS})
Rob Hughes25b74362020-01-13 11:14:59 +0000548if ("${CMAKE_SYSTEM_NAME}" STREQUAL Windows)
549 target_link_libraries(armnn Ws2_32.lib)
550endif()
telsoa014fcda012018-03-09 14:13:49 +0000551
Matthew Benthama7b79282019-06-03 13:00:50 +0100552install(TARGETS armnn
Rob Hughes91e1d892019-08-23 10:11:58 +0100553 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
554 RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
telsoa014fcda012018-03-09 14:13:49 +0000555if(BUILD_CAFFE_PARSER)
Matthew Benthama7b79282019-06-03 13:00:50 +0100556 install(TARGETS armnnCaffeParser
Rob Hughes91e1d892019-08-23 10:11:58 +0100557 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
558 RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
telsoa014fcda012018-03-09 14:13:49 +0000559endif()
telsoa01c577f2c2018-08-31 09:22:23 +0100560if(BUILD_ONNX_PARSER)
Matthew Benthama7b79282019-06-03 13:00:50 +0100561 install(TARGETS armnnOnnxParser
Rob Hughes91e1d892019-08-23 10:11:58 +0100562 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
563 RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
telsoa01c577f2c2018-08-31 09:22:23 +0100564endif()
surmeh01bceff2f2018-03-29 16:29:27 +0100565if(BUILD_TF_PARSER)
Matthew Benthama7b79282019-06-03 13:00:50 +0100566 install(TARGETS armnnTfParser
Rob Hughes91e1d892019-08-23 10:11:58 +0100567 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
568 RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
surmeh01bceff2f2018-03-29 16:29:27 +0100569endif()
Ruomei Yandb7c3912019-03-08 17:50:25 +0000570
Matthew Benthama7b79282019-06-03 13:00:50 +0100571install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
telsoa014fcda012018-03-09 14:13:49 +0000572
Derek Lamberti08446972019-11-26 16:38:31 +0000573target_link_libraries(armnn ${Boost_THREAD_LIBRARY}
Matteo Martincighe7d44982019-08-05 12:16:47 +0100574 ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
telsoa014fcda012018-03-09 14:13:49 +0000575
576if(ARMCOMPUTENEON OR ARMCOMPUTECL)
577 target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})
578endif()
telsoa014fcda012018-03-09 14:13:49 +0000579
Rob Hughes199c98e2019-11-13 11:41:36 +0000580if(PROFILING_BACKEND_STREAMLINE AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android)))
telsoa014fcda012018-03-09 14:13:49 +0000581 target_link_libraries(armnn pthread)
582endif()
583
Matteo Martincighbccc3452019-10-18 14:59:19 +0100584set_target_properties(armnn PROPERTIES VERSION ${GENERIC_LIB_VERSION} SOVERSION ${GENERIC_LIB_SOVERSION})
585
telsoa014fcda012018-03-09 14:13:49 +0000586if(BUILD_UNIT_TESTS)
587 set(unittest_sources)
588 list(APPEND unittest_sources
jimfly01aebf2e72019-02-05 09:20:11 +0000589 src/armnn/test/ConstTensorLayerVisitor.hpp
590 src/armnn/test/ConstTensorLayerVisitor.cpp
telsoa014fcda012018-03-09 14:13:49 +0000591 src/armnn/test/CreateWorkload.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100592 src/armnn/test/CsvReaderTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100593 src/armnn/test/EndToEndTest.cpp
Teresa Charlin4de9f672019-04-10 13:59:49 +0100594 src/armnn/test/ExecutionFrameTest.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100595 src/armnn/test/FloatingPointConverterTest.cpp
Matthew Benthamd78b891d2019-04-30 10:17:40 +0100596 src/armnn/test/FlowControl.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100597 src/armnn/test/GraphTests.cpp
narpra016f37f832018-12-21 18:30:00 +0000598 src/armnn/test/GraphUtils.cpp
telsoa014fcda012018-03-09 14:13:49 +0000599 src/armnn/test/GraphUtils.hpp
telsoa01c577f2c2018-08-31 09:22:23 +0100600 src/armnn/test/InstrumentTests.cpp
Matteo Martincighbf0e7222019-06-20 17:17:45 +0100601 src/armnn/test/InferOutputTests.cpp
602 src/armnn/test/InferOutputTests.hpp
Éanna Ó Catháina4247d52019-05-08 14:00:45 +0100603 src/armnn/test/ModelAccuracyCheckerTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100604 src/armnn/test/NetworkTests.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100605 src/armnn/test/ObservableTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100606 src/armnn/test/OptimizerTests.cpp
Rob Hughes95e73d72019-09-24 09:34:53 +0100607 src/armnn/test/optimizations/ConvertConstantsFloatToHalfTests.cpp
608 src/armnn/test/optimizations/ConvertConstantsHalfToFloatTests.cpp
609 src/armnn/test/optimizations/Fp32NetworkToFp16ConverterTests.cpp
610 src/armnn/test/optimizations/InsertDebugLayerTests.cpp
611 src/armnn/test/optimizations/MovePermuteUpTests.cpp
612 src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp
613 src/armnn/test/optimizations/OptimizeInverseConversionsTests.cpp
614 src/armnn/test/optimizations/OptimizeInversePermutesTests.cpp
Rob Hughes3a7d3a72019-09-24 16:59:56 +0100615 src/armnn/test/optimizations/PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp
Rob Hughes95e73d72019-09-24 09:34:53 +0100616 src/armnn/test/optimizations/PermuteAsReshapeTests.cpp
617 src/armnn/test/optimizations/SquashEqualSiblingsTests.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100618 src/armnn/test/OptionalTest.cpp
619 src/armnn/test/ProfilerTests.cpp
620 src/armnn/test/ProfilingEventTest.cpp
Derek Lambertiff05cc52019-04-26 13:05:17 +0100621 src/armnn/test/SubgraphViewTests.cpp
Derek Lamberti84da38b2019-06-13 11:40:08 +0100622 src/armnn/test/TensorHandleStrategyTest.cpp
David Beckf98d21a2018-10-26 16:03:03 +0100623 src/armnn/test/TensorHelpers.hpp
624 src/armnn/test/TensorTest.cpp
FrancisMurtagh531ef942019-02-01 11:48:58 +0000625 src/armnn/test/TestInputOutputLayerVisitor.cpp
626 src/armnn/test/TestInputOutputLayerVisitor.hpp
FrancisMurtagh1306b192019-01-28 14:03:21 +0000627 src/armnn/test/TestLayerVisitor.cpp
628 src/armnn/test/TestLayerVisitor.hpp
FrancisMurtagh28d3d632019-01-29 12:15:23 +0000629 src/armnn/test/TestNameOnlyLayerVisitor.cpp
630 src/armnn/test/TestNameOnlyLayerVisitor.hpp
FrancisMurtaghf08876f2019-02-04 15:41:17 +0000631 src/armnn/test/TestNameAndDescriptorLayerVisitor.hpp
632 src/armnn/test/TestNameAndDescriptorLayerVisitor.cpp
Matteo Martincighbf0e7222019-06-20 17:17:45 +0100633 src/armnn/test/TestUtils.cpp
634 src/armnn/test/TestUtils.hpp
David Beckf98d21a2018-10-26 16:03:03 +0100635 src/armnn/test/UnitTests.cpp
636 src/armnn/test/UnitTests.hpp
Ferran Balaguer51dd62f2019-01-11 19:29:18 +0000637 src/armnn/test/UtilsTests.cpp
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100638 src/armnnUtils/test/QuantizeHelperTest.cpp
Matthew Bentham4057d912019-01-21 15:45:51 +0000639 src/armnnUtils/test/PrototxtConversionsTest.cpp
Ferran Balaguer51dd62f2019-01-11 19:29:18 +0000640 src/armnnUtils/test/ParserHelperTest.cpp
Narumol Prangnawarat02807852019-09-11 16:43:09 +0100641 src/armnnUtils/test/TensorUtilsTest.cpp
Narumol Prangnawarat7be47ef2019-09-27 18:00:11 +0100642 src/profiling/test/BufferTests.cpp
Keith Davis3201eea2019-10-24 17:30:41 +0100643 src/profiling/test/FileOnlyProfilingDecoratorTests.cpp
Aron Virginas-Tardfa14772019-09-24 18:24:47 +0100644 src/profiling/test/ProfilingConnectionDumpToFileDecoratorTests.cpp
Narumol Prangnawarat15effd82019-10-22 14:17:11 +0100645 src/profiling/test/ProfilingGuidTest.cpp
Francis Murtagh1f7db452019-08-14 09:49:34 +0100646 src/profiling/test/ProfilingTests.cpp
Matteo Martincighd0613b52019-10-09 16:47:04 +0100647 src/profiling/test/ProfilingTests.hpp
Narumol Prangnawaratdf31cfe2019-11-22 11:26:06 +0000648 src/profiling/test/ProfilingTestUtils.cpp
649 src/profiling/test/ProfilingTestUtils.hpp
Ferran Balagueraf5c46b2019-08-30 15:49:15 +0100650 src/profiling/test/SendCounterPacketTests.cpp
Matteo Martincighd0613b52019-10-09 16:47:04 +0100651 src/profiling/test/SendCounterPacketTests.hpp
Sadik Armagan7bbdf9d2019-10-24 10:26:05 +0100652 src/profiling/test/SendTimelinePacketTests.cpp
Matteo Martincigh0aed4f92019-10-01 14:25:34 +0100653 src/profiling/test/TimelinePacketTests.cpp
Matteo Martincigh830101c2019-10-22 11:07:45 +0100654 src/profiling/test/TimelineUtilityMethodsTests.cpp
Ferran Balaguer51dd62f2019-01-11 19:29:18 +0000655 )
telsoa014fcda012018-03-09 14:13:49 +0000656
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100657 if(ARMNNREF)
Matteo Martincighe67edb22019-08-14 14:05:46 +0100658 list(APPEND unittest_sources
659 src/armnn/test/DebugCallbackTest.cpp
660 src/armnn/test/QuantizerTest.cpp
661 src/armnn/test/RuntimeTests.cpp
662 src/armnn/test/RuntimeTests.hpp
663 )
664 endif()
665
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100666 if(BUILD_TF_PARSER AND ARMNNREF)
surmeh01bceff2f2018-03-29 16:29:27 +0100667 list(APPEND unittest_sources
668 src/armnnTfParser/test/Activations.cpp
669 src/armnnTfParser/test/Addition.cpp
Ferran Balaguerfbdad032018-12-28 18:15:24 +0000670 src/armnnTfParser/test/AddN.cpp
narpra016f37f832018-12-21 18:30:00 +0000671 src/armnnTfParser/test/Assert.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100672 src/armnnTfParser/test/BiasAdd.cpp
673 src/armnnTfParser/test/BroadcastForAdd.cpp
674 src/armnnTfParser/test/Convolution2d.cpp
675 src/armnnTfParser/test/Concat.cpp
676 src/armnnTfParser/test/ConcatOfConcats.cpp
677 src/armnnTfParser/test/DepthwiseConvolution2d.cpp
jimfly0184c70e62018-12-19 13:14:46 +0000678 src/armnnTfParser/test/Equal.cpp
Conor Kennedyc2130a02018-12-05 11:05:54 +0000679 src/armnnTfParser/test/ExpandDims.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100680 src/armnnTfParser/test/FusedBatchNorm.cpp
FrancisMurtagh94412af2019-01-24 10:53:39 +0000681 src/armnnTfParser/test/Gather.cpp
jimfly01a06bf312018-12-18 16:24:51 +0000682 src/armnnTfParser/test/Greater.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100683 src/armnnTfParser/test/Identity.cpp
684 src/armnnTfParser/test/LocalResponseNormalization.cpp
Sadik Armagan975c09a2018-12-04 10:02:08 +0000685 src/armnnTfParser/test/Maximum.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100686 src/armnnTfParser/test/MaximumForLeakyRelu.cpp
Ferran Balaguer51dd62f2019-01-11 19:29:18 +0000687 src/armnnTfParser/test/Mean.cpp
Nattapat Chaimanowong24df8222018-12-04 13:47:02 +0000688 src/armnnTfParser/test/Minimum.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100689 src/armnnTfParser/test/Multiplication.cpp
jimfly01f6ba7472018-12-04 10:09:52 +0000690 src/armnnTfParser/test/Pad.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100691 src/armnnTfParser/test/MultiOutput.cpp
692 src/armnnTfParser/test/PassThru.cpp
693 src/armnnTfParser/test/Pooling.cpp
saoste01bbd40612018-08-28 15:41:51 +0100694 src/armnnTfParser/test/RealDiv.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100695 src/armnnTfParser/test/Reshape.cpp
696 src/armnnTfParser/test/ResizeBilinear.cpp
Mohamed Nour Abouelseoud7a8892f2019-01-09 14:19:58 +0000697 src/armnnTfParser/test/Rsqrt.cpp
surmeh01bceff2f2018-03-29 16:29:27 +0100698 src/armnnTfParser/test/Shape.cpp
699 src/armnnTfParser/test/Softmax.cpp
700 src/armnnTfParser/test/TestDependencies.cpp
701 src/armnnTfParser/test/FullyConnected.cpp
702 src/armnnTfParser/test/Constant.cpp
703 src/armnnTfParser/test/TestMultiInputsOutputs.cpp
Sadik Armagan2ad6cb42018-12-27 11:23:44 +0000704 src/armnnTfParser/test/Split.cpp
jimfly0123be07e2018-12-04 17:47:22 +0000705 src/armnnTfParser/test/Squeeze.cpp
Ferran Balaguer51dd62f2019-01-11 19:29:18 +0000706 src/armnnTfParser/test/Sub.cpp
707 )
Matteo Martincighe67edb22019-08-14 14:05:46 +0100708
surmeh01bceff2f2018-03-29 16:29:27 +0100709 endif()
710
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100711 if(BUILD_TF_LITE_PARSER AND ARMNNREF)
telsoa01c577f2c2018-08-31 09:22:23 +0100712 list(APPEND unittest_sources
Sadik Armagan479045b2018-10-01 11:51:37 +0100713 src/armnnTfLiteParser/test/Activations.cpp
Bruno Goncalvesd4ac6a42018-12-18 12:56:22 -0200714 src/armnnTfLiteParser/test/Addition.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100715 src/armnnTfLiteParser/test/AvgPool2D.cpp
Bruno Goncalvesdb947e22019-02-08 18:52:21 -0200716 src/armnnTfLiteParser/test/BatchToSpaceND.cpp
Sadik Armagan479045b2018-10-01 11:51:37 +0100717 src/armnnTfLiteParser/test/Concatenation.cpp
Bruno Goncalves3d7efe92018-12-27 14:21:43 -0200718 src/armnnTfLiteParser/test/Constant.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100719 src/armnnTfLiteParser/test/Conv2D.cpp
720 src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp
Finn Williamsed66d142019-12-06 09:55:55 +0000721 src/armnnTfLiteParser/test/Dequantize.cpp
keidav011b3e2ea2019-02-21 10:07:37 +0000722 src/armnnTfLiteParser/test/DetectionPostProcess.cpp
Sadik Armagan8853c1f2018-10-22 09:04:18 +0100723 src/armnnTfLiteParser/test/FullyConnected.cpp
Matthew Jackson28c94572019-07-18 10:47:03 +0100724 src/armnnTfLiteParser/test/L2Normalization.cpp
Bruno Goncalvesb8d805e2019-02-12 22:57:13 -0200725 src/armnnTfLiteParser/test/Maximum.cpp
Nattapat Chaimanowongb66504b2018-10-17 15:19:14 +0100726 src/armnnTfLiteParser/test/MaxPool2D.cpp
Bruno Goncalves2235cee2018-12-19 12:51:45 -0200727 src/armnnTfLiteParser/test/Mean.cpp
Bruno Goncalves8f6d7a72019-02-12 22:58:18 -0200728 src/armnnTfLiteParser/test/Minimum.cpp
Bruno Goncalvesf803f782018-12-18 13:40:30 -0200729 src/armnnTfLiteParser/test/Multiplication.cpp
Matthew Jacksonbcca1f42019-07-16 11:39:21 +0100730 src/armnnTfLiteParser/test/Pack.cpp
Bruno Goncalves6c2355b2018-12-19 12:52:01 -0200731 src/armnnTfLiteParser/test/Pad.cpp
Sadik Armagan479045b2018-10-01 11:51:37 +0100732 src/armnnTfLiteParser/test/Reshape.cpp
Bruno Goncalves3f58ddb2019-02-07 18:40:11 -0200733 src/armnnTfLiteParser/test/ResizeBilinear.cpp
Sadik Armagana3b31f02019-12-05 09:08:53 +0000734 src/armnnTfLiteParser/test/ResizeNearestNeighbor.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100735 src/armnnTfLiteParser/test/Softmax.cpp
Bruno Goncalvesbaded142019-02-08 19:02:48 -0200736 src/armnnTfLiteParser/test/SpaceToBatchND.cpp
josh minorba424d22019-11-13 10:55:17 -0600737 src/armnnTfLiteParser/test/Slice.cpp
Nina Drozd0324f482019-04-08 10:52:10 +0100738 src/armnnTfLiteParser/test/Split.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100739 src/armnnTfLiteParser/test/Squeeze.cpp
Bruno Goncalves451d95b2019-02-12 22:59:22 -0200740 src/armnnTfLiteParser/test/StridedSlice.cpp
Nina Drozd0324f482019-04-08 10:52:10 +0100741 src/armnnTfLiteParser/test/Sub.cpp
Matthew Jackson74bf7da2019-08-16 16:51:42 +0100742 src/armnnTfLiteParser/test/TransposeConv.cpp
Keith Davis4cd29a02019-09-09 14:49:20 +0100743 src/armnnTfLiteParser/test/Transpose.cpp
Nina Drozd200e3802019-04-15 09:47:39 +0100744 src/armnnTfLiteParser/test/Unpack.cpp
Aron Virginas-Tarc975f922019-10-23 17:38:17 +0100745 src/armnnTfLiteParser/test/Unsupported.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100746 src/armnnTfLiteParser/test/LoadModel.cpp
747 src/armnnTfLiteParser/test/GetBuffer.cpp
748 src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp
749 src/armnnTfLiteParser/test/InputOutputTensorNames.cpp
750 src/armnnTfLiteParser/test/GetTensorIds.cpp
751 src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp
752 src/armnnTfLiteParser/test/GetInputsOutputs.cpp
753 )
Rob Hughesa91479c2020-01-07 09:40:12 +0000754
755 # Generate SchemaText.cpp file which contains the TfLite schema text as a
Rob Hughesff3c4262019-12-20 17:43:16 +0000756 # static C-array of bytes. This is needed at runtime for TfLite parser tests.
757 add_custom_command(
758 OUTPUT SchemaText.cpp
759 COMMAND cp ${TF_LITE_SCHEMA_INCLUDE_PATH}/schema.fbs g_TfLiteSchemaText
760 COMMAND xxd -i g_TfLiteSchemaText SchemaText.cpp
761 WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
Rob Hughesa91479c2020-01-07 09:40:12 +0000762 DEPENDS ${TF_LITE_SCHEMA_INCLUDE_PATH}/schema.fbs
Rob Hughesff3c4262019-12-20 17:43:16 +0000763 )
764 list(APPEND unittest_sources ${CMAKE_CURRENT_BINARY_DIR}/SchemaText.cpp)
telsoa01c577f2c2018-08-31 09:22:23 +0100765 endif()
766
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100767 if(BUILD_CAFFE_PARSER AND ARMNNREF)
telsoa014fcda012018-03-09 14:13:49 +0000768 list(APPEND unittest_sources
769 src/armnnCaffeParser/test/TestAdd.cpp
770 src/armnnCaffeParser/test/TestConcat.cpp
telsoa01c577f2c2018-08-31 09:22:23 +0100771 src/armnnCaffeParser/test/TestConvolution.cpp
telsoa014fcda012018-03-09 14:13:49 +0000772 src/armnnCaffeParser/test/TestDropout.cpp
773 src/armnnCaffeParser/test/TestInputs.cpp
774 src/armnnCaffeParser/test/TestMul.cpp
775 src/armnnCaffeParser/test/TestPooling2d.cpp
776 src/armnnCaffeParser/test/TestInPlace.cpp
777 src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
778 src/armnnCaffeParser/test/TestSplit.cpp
779 )
780 endif()
781
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100782 if(BUILD_ONNX_PARSER AND ARMNNREF)
Matteo Martincighe67edb22019-08-14 14:05:46 +0100783 list(APPEND unittest_sources
784 src/armnnOnnxParser/test/Constructor.cpp
785 src/armnnOnnxParser/test/CreateNetwork.cpp
786 src/armnnOnnxParser/test/ProtoxtFixture.cpp
787 src/armnnOnnxParser/test/Const.cpp
788 src/armnnOnnxParser/test/Pooling.cpp
789 src/armnnOnnxParser/test/Reshape.cpp
790 src/armnnOnnxParser/test/Relu.cpp
791 src/armnnOnnxParser/test/Conv2D.cpp
792 src/armnnOnnxParser/test/Addition.cpp
793 src/armnnOnnxParser/test/FullyConnected.cpp
794 src/armnnOnnxParser/test/GetInputsOutputs.cpp
795 src/armnnOnnxParser/test/BatchNorm.cpp
796 src/armnnOnnxParser/test/DepthConv.cpp
797 )
798 endif()
799
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100800 if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
Nina Drozd59e15b02019-04-25 15:45:20 +0100801 list(APPEND unittest_sources
802 src/armnnQuantizer/test/QuantizationDataSetTests.cpp
803 )
804 endif()
805
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100806 if(BUILD_ARMNN_SERIALIZER AND ARMNNREF)
Kevin May43a799c2019-02-08 16:31:42 +0000807 enable_language(ASM)
808 list(APPEND unittest_sources
Ruomei Yandb7c3912019-03-08 17:50:25 +0000809 src/armnnSerializer/test/ActivationSerializationTests.cpp
810 src/armnnSerializer/test/SerializerTests.cpp
FinnWilliamsArm4ffcc8f2019-09-05 14:34:20 +0100811 src/armnnDeserializer/test/DeserializeAbs.cpp
Ruomei Yandb7c3912019-03-08 17:50:25 +0000812 src/armnnDeserializer/test/DeserializeActivation.cpp
813 src/armnnDeserializer/test/DeserializeAdd.cpp
814 src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp
815 src/armnnDeserializer/test/DeserializeBatchNormalization.cpp
Aron Virginas-Tar422f2fb2019-10-21 14:09:11 +0100816 src/armnnDeserializer/test/DeserializeComparison.cpp
Ruomei Yandb7c3912019-03-08 17:50:25 +0000817 src/armnnDeserializer/test/DeserializeConstant.cpp
818 src/armnnDeserializer/test/DeserializeConvolution2d.cpp
Aron Virginas-Tar11b2eca2019-09-24 11:01:51 +0100819 src/armnnDeserializer/test/DeserializeDepthToSpace.cpp
Ruomei Yandb7c3912019-03-08 17:50:25 +0000820 src/armnnDeserializer/test/DeserializeDivision.cpp
Ruomei Yandb7c3912019-03-08 17:50:25 +0000821 src/armnnDeserializer/test/DeserializeFloor.cpp
822 src/armnnDeserializer/test/DeserializeFullyConnected.cpp
823 src/armnnDeserializer/test/DeserializeGather.cpp
Aron Virginas-Tar92bbcae2019-10-09 10:36:34 +0100824 src/armnnDeserializer/test/DeserializeInstanceNormalization.cpp
Ferran Balaguer0dcffec2019-06-18 16:25:06 +0100825 src/armnnDeserializer/test/DeserializeL2Normalization.cpp
Aron Virginas-Tarc949ab52019-10-14 16:43:13 +0100826 src/armnnDeserializer/test/DeserializeLogSoftmax.cpp
Ruomei Yandb7c3912019-03-08 17:50:25 +0000827 src/armnnDeserializer/test/DeserializeMean.cpp
828 src/armnnDeserializer/test/DeserializeMultiplication.cpp
829 src/armnnDeserializer/test/DeserializeNormalization.cpp
830 src/armnnDeserializer/test/DeserializePad.cpp
831 src/armnnDeserializer/test/DeserializePermute.cpp
832 src/armnnDeserializer/test/DeserializePooling2d.cpp
833 src/armnnDeserializer/test/DeserializeReshape.cpp
834 src/armnnDeserializer/test/DeserializeResizeBilinear.cpp
835 src/armnnDeserializer/test/DeserializeRsqrt.cpp
Aron Virginas-Tara2f4b4b2019-09-20 11:21:14 +0100836 src/armnnDeserializer/test/DeserializeSlice.cpp
Ruomei Yandb7c3912019-03-08 17:50:25 +0000837 src/armnnDeserializer/test/DeserializeSpaceToBatchNd.cpp
838 src/armnnDeserializer/test/DeserializeStridedSlice.cpp
839 src/armnnDeserializer/test/DeserializeSubtraction.cpp
840 src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp
841 src/armnnDeserializer/test/SchemaSerialize.s
842 )
Derek Lamberti0028d1b2019-02-20 13:57:42 +0000843 set_source_files_properties(src/armnnDeserializer/test/SchemaSerialize.s PROPERTIES COMPILE_FLAGS "-x assembler-with-cpp")
Kevin May43a799c2019-02-08 16:31:42 +0000844 endif()
845
Colm Donelana21620d2019-10-11 13:09:49 +0100846 if(BUILD_GATORD_MOCK)
847 list(APPEND unittest_sources
848 tests/profiling/gatordmock/tests/GatordMockTests.cpp
Finn Williamse63a0262019-10-22 10:30:49 +0100849 tests/profiling/timelineDecoder/tests/TimelineTests.cpp
Colm Donelana21620d2019-10-11 13:09:49 +0100850 )
851 endif()
852
Matteo Martincigh4423ac72018-10-11 17:39:30 +0100853 foreach(lib ${armnnUnitTestLibraries})
Rob Hughes721b82f2019-11-15 09:04:17 +0000854 message(STATUS "Adding object library dependency to UnitTests: ${lib}")
Matteo Martincigh4423ac72018-10-11 17:39:30 +0100855 list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)
856 endforeach()
857
858 add_executable(UnitTests ${unittest_sources})
telsoa014fcda012018-03-09 14:13:49 +0000859 target_include_directories(UnitTests PRIVATE src/armnn)
860 target_include_directories(UnitTests PRIVATE src/armnnUtils)
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000861 target_include_directories(UnitTests PRIVATE src/backends)
Matteo Martincigh6db5f202019-09-05 12:02:04 +0100862 target_include_directories(UnitTests PRIVATE src/profiling)
telsoa014fcda012018-03-09 14:13:49 +0000863
telsoa01c577f2c2018-08-31 09:22:23 +0100864 if(VALGRIND_FOUND)
865 if(HEAP_PROFILING OR LEAK_CHECKING)
Rob Hughes721b82f2019-11-15 09:04:17 +0000866 message(STATUS "Valgrind is disabled for heap profiling and leak checking builds.")
telsoa01c577f2c2018-08-31 09:22:23 +0100867 else()
868 # Valgrind works with gperftools version number <= 2.4
869 target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")
870 endif()
telsoa014fcda012018-03-09 14:13:49 +0000871 endif()
872
telsoa014fcda012018-03-09 14:13:49 +0000873 target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})
telsoa01c577f2c2018-08-31 09:22:23 +0100874 target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})
telsoa014fcda012018-03-09 14:13:49 +0000875
Matteo Martincigh4423ac72018-10-11 17:39:30 +0100876 target_link_libraries(UnitTests armnn)
877 target_link_libraries(UnitTests armnnUtils)
878
surmeh01bceff2f2018-03-29 16:29:27 +0100879 if(BUILD_TF_PARSER)
880 target_link_libraries(UnitTests armnnTfParser)
881 endif()
882
Colm Donelana21620d2019-10-11 13:09:49 +0100883 if(BUILD_GATORD_MOCK)
884 target_link_libraries(UnitTests gatordMockService)
885 endif()
886
Ruomei Yandb7c3912019-03-08 17:50:25 +0000887 if(BUILD_TF_LITE_PARSER)
888 target_include_directories(UnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
889 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
890 target_link_libraries(UnitTests armnnTfLiteParser)
891 endif()
892
telsoa014fcda012018-03-09 14:13:49 +0000893 if(BUILD_CAFFE_PARSER)
894 target_link_libraries(UnitTests armnnCaffeParser)
895 endif()
896
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100897 if(BUILD_ARMNN_SERIALIZER AND ARMNNREF)
Mike Kelly8c1701a2019-02-11 17:01:27 +0000898 target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer)
Ruomei Yandb7c3912019-03-08 17:50:25 +0000899 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
Kevin May43a799c2019-02-08 16:31:42 +0000900 target_link_libraries(UnitTests armnnSerializer)
901 endif()
902
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100903 if(BUILD_ARMNN_QUANTIZER AND ARMNNREF)
Nina Drozd59e15b02019-04-25 15:45:20 +0100904 target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnQuantizer)
905 target_include_directories(UnitTests SYSTEM PRIVATE "${FLATBUFFERS_INCLUDE_PATH}")
906 target_link_libraries(UnitTests armnnQuantizer armnnSerializer)
907 endif()
908
telsoa01c577f2c2018-08-31 09:22:23 +0100909 if(BUILD_ONNX_PARSER)
910 target_link_libraries(UnitTests armnnOnnxParser)
911 endif()
912
telsoa014fcda012018-03-09 14:13:49 +0000913 addDllCopyCommands(UnitTests)
914endif()
Nattapat Chaimanowong4fbae332019-02-14 15:28:02 +0000915
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100916if (BUILD_ARMNN_SERIALIZER AND (BUILD_TF_PARSER OR BUILD_TF_LITE_PARSER OR BUILD_ONNX_PARSER OR BUILD_CAFFE_PARSER) AND ARMNNREF)
Nattapat Chaimanowong4fbae332019-02-14 15:28:02 +0000917 set(ArmnnConverter_sources
918 src/armnnConverter/ArmnnConverter.cpp)
919
920 add_executable_ex(ArmnnConverter ${ArmnnConverter_sources})
921 target_include_directories(ArmnnConverter PRIVATE src/armnn)
922 target_include_directories(ArmnnConverter PRIVATE src/armnnUtils)
923
Sadik Armagan232cfc22019-03-13 18:33:10 +0000924 if(BUILD_CAFFE_PARSER)
925 target_link_libraries(ArmnnConverter armnnCaffeParser)
926 endif()
927
928 if(BUILD_ONNX_PARSER)
929 target_link_libraries(ArmnnConverter armnnOnnxParser)
930 endif()
931
932 if(BUILD_TF_PARSER)
933 target_link_libraries(ArmnnConverter armnnTfParser)
934 endif()
935
936 if(BUILD_TF_LITE_PARSER)
937 target_link_libraries(ArmnnConverter armnnTfLiteParser)
938 endif()
939
Nattapat Chaimanowong4fbae332019-02-14 15:28:02 +0000940 target_link_libraries(ArmnnConverter armnnSerializer)
941
942 target_link_libraries(ArmnnConverter armnn)
943 target_link_libraries(ArmnnConverter ${CMAKE_THREAD_LIBS_INIT})
944 target_link_libraries(ArmnnConverter
945 ${Boost_SYSTEM_LIBRARY}
946 ${Boost_FILESYSTEM_LIBRARY}
947 ${Boost_PROGRAM_OPTIONS_LIBRARY})
948 addDllCopyCommands(ArmnnConverter)
949endif()
Colm Donelan366023f2019-09-05 10:03:56 +0100950
951if(BUILD_GATORD_MOCK)
952 set(gatord_mock_sources)
953 list(APPEND gatord_mock_sources
Colm Donelana21620d2019-10-11 13:09:49 +0100954 tests/profiling/gatordmock/CommandFileParser.hpp
955 tests/profiling/gatordmock/CommandFileParser.cpp
956 tests/profiling/gatordmock/CommandLineProcessor.hpp
957 tests/profiling/gatordmock/CommandLineProcessor.cpp
958 tests/profiling/gatordmock/GatordMockService.hpp
959 tests/profiling/gatordmock/GatordMockService.cpp
Colm Donelana21620d2019-10-11 13:09:49 +0100960 tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.cpp
961 tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp
Colm Donelanb682d842019-10-16 12:24:20 +0100962 tests/profiling/gatordmock/PeriodicCounterSelectionResponseHandler.cpp
963 tests/profiling/gatordmock/PeriodicCounterSelectionResponseHandler.hpp
Matteo Martincighcdfb9412019-11-08 11:23:06 +0000964 tests/profiling/gatordmock/StreamMetadataCommandHandler.cpp
965 tests/profiling/gatordmock/StreamMetadataCommandHandler.hpp
Finn Williamse63a0262019-10-22 10:30:49 +0100966 tests/profiling/timelineDecoder/ITimelineDecoder.h
967 tests/profiling/timelineDecoder/TimelineCaptureCommandHandler.cpp
968 tests/profiling/timelineDecoder/TimelineCaptureCommandHandler.hpp
969 tests/profiling/timelineDecoder/TimelineDecoder.cpp
970 tests/profiling/timelineDecoder/TimelineDirectoryCaptureCommandHandler.cpp
971 tests/profiling/timelineDecoder/TimelineDirectoryCaptureCommandHandler.hpp
972 tests/profiling/timelineDecoder/tests/TimelineTestFunctions.hpp
Colm Donelan366023f2019-09-05 10:03:56 +0100973 )
974
Matteo Martincigh65984272019-10-17 13:26:21 +0100975 include_directories(${Boost_INCLUDE_DIRS} src/profiling)
Matteo Martincighcdfb9412019-11-08 11:23:06 +0000976 include_directories(${Boost_INCLUDE_DIRS} tests/profiling)
977 include_directories(${Boost_INCLUDE_DIRS} tests/profiling/gatordmock)
978 include_directories(${Boost_INCLUDE_DIRS} tests/profiling/timelineDecoder)
Colm Donelan366023f2019-09-05 10:03:56 +0100979
Colm Donelana21620d2019-10-11 13:09:49 +0100980 add_library_ex(gatordMockService STATIC ${gatord_mock_sources})
Rob Hughes25b74362020-01-13 11:14:59 +0000981 target_include_directories(gatordMockService PRIVATE src/armnnUtils)
Colm Donelan366023f2019-09-05 10:03:56 +0100982
Rob Hughes5edefa02020-01-10 09:15:30 +0000983 add_executable_ex(GatordMock tests/profiling/gatordmock/GatordMockMain.cpp)
Rob Hughes25b74362020-01-13 11:14:59 +0000984 target_include_directories(GatordMock PRIVATE src/armnnUtils)
Colm Donelana21620d2019-10-11 13:09:49 +0100985
Rob Hughes5edefa02020-01-10 09:15:30 +0000986 target_link_libraries(GatordMock
Matteo Martincigh65984272019-10-17 13:26:21 +0100987 armnn
988 gatordMockService
989 ${Boost_PROGRAM_OPTIONS_LIBRARY}
990 ${Boost_SYSTEM_LIBRARY})
Colm Donelan366023f2019-09-05 10:03:56 +0100991
Rob Hughes199c98e2019-11-13 11:41:36 +0000992 if(Threads_FOUND AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android)))
Rob Hughes5edefa02020-01-10 09:15:30 +0000993 target_link_libraries(GatordMock pthread)
Rob Hughes199c98e2019-11-13 11:41:36 +0000994 endif()
995
Matteo Martincigh994b5342019-10-11 17:19:56 +0100996endif()