Blame - CMakeLists.txt - ml/armnn

2018-09-19 12:03:20 +0100

[diff] [blame]

1

#

2

3

# SPDX-License-Identifier: MIT

4

#

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

5

cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake.

6

project(armnn)

7

8

set(additional_cmake_files)

9

list(APPEND additional_cmake_files

10

cmake/Utils.cmake

11

cmake/GlobalConfig.cmake

12

cmake/AddDllCopyCommands.cmake)

13

14

foreach(cmake_file ${additional_cmake_files})

15

include(${cmake_file})

endforeach()

if (BUILD_TESTS)

add_subdirectory(tests)

20

endif()

21

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

22

add_subdirectory(samples)

23

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

24

# Include the additional cmake files in their own target so that they will appear nicely in IDEs

25

add_custom_target(AdditionalCMakeFiles SOURCES ${additional_cmake_files})

26

27

set(armnnUtils_sources)

28

list(APPEND armnnUtils_sources

29

src/armnnUtils/GraphTopologicalSort.hpp

arovir01

616e775

2018-10-01 17:08:59 +0100

[diff] [blame]

30

src/armnnUtils/Half.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

31

src/armnnUtils/Logging.hpp

32

src/armnnUtils/Permute.hpp

33

src/armnnUtils/Logging.cpp

34

src/armnnUtils/Permute.cpp

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

35

src/armnnUtils/DotSerializer.cpp

36

src/armnnUtils/DotSerializer.hpp

surmeh01

2018-05-18 16:31:43 +0100

[diff] [blame]

37

src/armnnUtils/HeapProfiling.cpp

38

src/armnnUtils/HeapProfiling.hpp

39

src/armnnUtils/LeakChecking.cpp

40

src/armnnUtils/LeakChecking.hpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

41

src/armnnUtils/CsvReader.cpp

42

src/armnnUtils/CsvReader.hpp

43

src/armnnUtils/FloatingPointConverter.cpp

44

src/armnnUtils/FloatingPointConverter.hpp

45

src/armnnUtils/VerificationHelpers.hpp

46

src/armnnUtils/VerificationHelpers.cpp

Sadik Armagan

2018-10-01 11:51:37 +0100

[diff] [blame]

47

src/armnnUtils/ParserHelper.hpp

48

src/armnnUtils/ParserHelper.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

49

)

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

50

if(BUILD_TF_PARSER OR BUILD_CAFFE_PARSER)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

51

list(APPEND armnnUtils_sources

52

src/armnnUtils/ParserPrototxtFixture.hpp

)

endif()

add_library_ex(armnnUtils STATIC ${armnnUtils_sources})

57

58

if(BUILD_CAFFE_PARSER)

59

# ArmNN Parser source files required for all build options

60

set(armnn_caffe_parser_sources)

61

list(APPEND armnn_caffe_parser_sources

62

include/armnnCaffeParser/ICaffeParser.hpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

63

src/armnnCaffeParser/RecordByRecordCaffeParser.hpp

64

src/armnnCaffeParser/RecordByRecordCaffeParser.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

65

src/armnnCaffeParser/CaffeParser.hpp

66

src/armnnCaffeParser/CaffeParser.cpp

67

${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc

68

)

69

# The generated Caffe protobuf .cc file is not warning clean and we can't fix them.

70

if(COMPILER_IS_GNU_LIKE)

71

set_source_files_properties(${CAFFE_GENERATED_SOURCES}/caffe/proto/caffe.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")

72

endif()

73

74

add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})

75

set_target_properties(armnnCaffeParser PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_ADDITIONAL_COMPILE_FLAGS}")

76

77

target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)

78

79

target_link_libraries(armnnCaffeParser ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})

80

81

target_link_libraries(armnnCaffeParser armnn)

82

target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

endif()

if(BUILD_ONNX_PARSER)

87

set(armnn_onnx_parser_sources)

88

list(APPEND armnn_onnx_parser_sources

89

include/armnnOnnxParser/IOnnxParser.hpp

90

src/armnnOnnxParser/OnnxParser.hpp

91

src/armnnOnnxParser/OnnxParser.cpp

92

${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc

93

)

94

# The generated onnx protobuf .cc files are not warning clean and we can't fix them.

95

if(COMPILER_IS_GNU_LIKE)

96

set_source_files_properties(${ONNX_GENERATED_SOURCES}/onnx/onnx.pb.cc PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")

97

endif()

98

99

add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})

100

101

target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)

102

103

target_link_libraries(armnnOnnxParser armnn)

104

105

# Protobuf

106

target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

107

endif()

108

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

109

if(BUILD_TF_PARSER)

110

set(armnn_tf_parser_sources)

111

list(APPEND armnn_tf_parser_sources

112

include/armnnTfParser/ITfParser.hpp

113

src/armnnTfParser/TfParser.hpp

114

src/armnnTfParser/TfParser.cpp

115

${TF_PROTOBUFS}

116

)

117

# The generated tensorflow protobuf .cc files are not warning clean and we can't fix them.

118

if(COMPILER_IS_GNU_LIKE)

119

set_source_files_properties(${TF_PROTOBUFS} PROPERTIES COMPILE_FLAGS "-Wno-conversion -Wno-sign-conversion")

120

endif()

121

122

add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})

123

124

target_include_directories(armnnTfParser PRIVATE src/armnnUtils)

125

126

target_link_libraries(armnnTfParser armnn)

127

128

# Protobuf (use the specific version tensorflow wants)

129

target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})

130

endif()

131

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

132

if(BUILD_TF_LITE_PARSER)

133

set(armnn_tf_lite_parser_sources)

134

list(APPEND armnn_tf_lite_parser_sources

135

include/armnnTfLiteParser/ITfLiteParser.hpp

136

src/armnnTfLiteParser/TfLiteParser.hpp

137

src/armnnTfLiteParser/TfLiteParser.cpp

138

)

139

140

add_library_ex(armnnTfLiteParser SHARED ${armnn_tf_lite_parser_sources})

141

142

target_include_directories(armnnTfLiteParser PRIVATE src/armnnUtils)

143

144

target_link_libraries(armnnTfLiteParser ${Boost_FILESYSTEM_LIBRARY} ${Boost_THREAD_LIBRARY})

145

target_link_libraries(armnnTfLiteParser armnn ${FLATBUFFERS_LIBRARY})

146

endif()

147

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

148

list(APPEND armnn_sources

149

include/armnn/ArmNN.hpp

David Beck

2018-10-10 15:11:44 +0100

[diff] [blame]

150

include/armnn/BackendId.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

151

include/armnn/Descriptors.hpp

152

include/armnn/DescriptorsFwd.hpp

David Beck

2018-10-10 15:11:44 +0100

[diff] [blame]

153

include/armnn/Exceptions.hpp

David Beck

e97c6e0

2018-10-03 13:09:28 +0100

[diff] [blame]

154

include/armnn/ILayerSupport.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

155

include/armnn/INetwork.hpp

David Beck

2018-10-10 15:11:44 +0100

[diff] [blame]

156

include/armnn/IProfiler.hpp

157

include/armnn/IRuntime.hpp

158

include/armnn/LayerSupport.hpp

159

include/armnn/LstmParams.hpp

160

include/armnn/NetworkFwd.hpp

David Beck

5eec11d

2018-10-04 15:43:17 +0100

[diff] [blame]

161

include/armnn/Optional.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

162

include/armnn/Tensor.hpp

163

include/armnn/TensorFwd.hpp

164

include/armnn/Types.hpp

165

include/armnn/TypesUtils.hpp

166

include/armnn/Utils.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

167

include/armnn/Version.hpp

surmeh01

2018-05-18 16:31:43 +0100

[diff] [blame]

168

src/armnn/layers/LayerCloneBase.hpp

169

src/armnn/layers/LayerWithParameters.hpp

170

src/armnn/layers/ActivationLayer.hpp

171

src/armnn/layers/ActivationLayer.cpp

172

src/armnn/layers/AdditionLayer.hpp

173

src/armnn/layers/AdditionLayer.cpp

Les Bell

033626d

2018-09-03 16:24:52 +0100

[diff] [blame]

174

src/armnn/layers/ArithmeticBaseLayer.hpp

175

src/armnn/layers/ArithmeticBaseLayer.cpp

surmeh01

2018-05-18 16:31:43 +0100

[diff] [blame]

176

src/armnn/layers/BatchNormalizationLayer.hpp

177

src/armnn/layers/BatchNormalizationLayer.cpp

178

src/armnn/layers/ConstantLayer.hpp

179

src/armnn/layers/ConstantLayer.cpp

180

src/armnn/layers/Convolution2dLayer.hpp

181

src/armnn/layers/Convolution2dLayer.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

182

src/armnn/layers/ConvertFp16ToFp32Layer.hpp

183

src/armnn/layers/ConvertFp16ToFp32Layer.cpp

184

src/armnn/layers/ConvertFp32ToFp16Layer.hpp

185

src/armnn/layers/ConvertFp32ToFp16Layer.cpp

surmeh01

2018-05-18 16:31:43 +0100

[diff] [blame]

186

src/armnn/layers/DepthwiseConvolution2dLayer.hpp

187

src/armnn/layers/DepthwiseConvolution2dLayer.cpp

188

src/armnn/layers/FakeQuantizationLayer.hpp

189

src/armnn/layers/FakeQuantizationLayer.cpp

190

src/armnn/layers/FloorLayer.hpp

191

src/armnn/layers/FloorLayer.cpp

192

src/armnn/layers/FullyConnectedLayer.hpp

193

src/armnn/layers/FullyConnectedLayer.cpp

194

src/armnn/layers/InputLayer.hpp

195

src/armnn/layers/InputLayer.cpp

196

src/armnn/layers/L2NormalizationLayer.hpp

197

src/armnn/layers/L2NormalizationLayer.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

198

src/armnn/layers/LstmLayer.cpp

199

src/armnn/layers/LstmLayer.hpp

narpra01

32b9046

2018-09-13 11:07:48 +0100

[diff] [blame]

200

src/armnn/layers/MeanLayer.hpp

201

src/armnn/layers/MeanLayer.cpp

surmeh01

2018-05-18 16:31:43 +0100

[diff] [blame]

202

src/armnn/layers/MemCopyLayer.hpp

203

src/armnn/layers/MemCopyLayer.cpp

204

src/armnn/layers/MergerLayer.hpp

205

src/armnn/layers/MergerLayer.cpp

206

src/armnn/layers/MultiplicationLayer.hpp

207

src/armnn/layers/MultiplicationLayer.cpp

208

src/armnn/layers/NormalizationLayer.hpp

209

src/armnn/layers/NormalizationLayer.cpp

210

src/armnn/layers/OutputLayer.hpp

211

src/armnn/layers/OutputLayer.cpp

Mohamed Nour Abouelseoud

5662c20

2018-09-24 13:30:09 +0100

[diff] [blame]

212

src/armnn/layers/PadLayer.hpp

213

src/armnn/layers/PadLayer.cpp

surmeh01

2018-05-18 16:31:43 +0100

[diff] [blame]

214

src/armnn/layers/PermuteLayer.hpp

215

src/armnn/layers/PermuteLayer.cpp

216

src/armnn/layers/Pooling2dLayer.hpp

217

src/armnn/layers/Pooling2dLayer.cpp

Francis Murtagh

e7a86a4

2018-08-29 12:42:10 +0100

[diff] [blame]

218

src/armnn/layers/DivisionLayer.cpp

219

src/armnn/layers/DivisionLayer.hpp

surmeh01

2018-05-18 16:31:43 +0100

[diff] [blame]

220

src/armnn/layers/ReshapeLayer.hpp

221

src/armnn/layers/ReshapeLayer.cpp

222

src/armnn/layers/ResizeBilinearLayer.hpp

223

src/armnn/layers/ResizeBilinearLayer.cpp

224

src/armnn/layers/SoftmaxLayer.hpp

225

src/armnn/layers/SoftmaxLayer.cpp

226

src/armnn/layers/SplitterLayer.hpp

227

src/armnn/layers/SplitterLayer.cpp

David Beck

c2044fe

2018-09-05 15:00:38 +0100

[diff] [blame]

228

src/armnn/layers/SubtractionLayer.cpp

229

src/armnn/layers/SubtractionLayer.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

230

src/armnn/InternalTypes.hpp

231

src/armnn/InternalTypes.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

232

src/armnn/JsonPrinter.hpp

233

src/armnn/JsonPrinter.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

234

src/armnn/LayerFwd.hpp

235

src/armnn/Layer.hpp

236

src/armnn/Layer.cpp

237

src/armnn/LayersFwd.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

238

src/armnn/Runtime.hpp

239

src/armnn/Runtime.cpp

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

240

src/armnn/SerializeLayerParameters.cpp

241

src/armnn/SerializeLayerParameters.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

242

src/armnn/Descriptors.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

243

src/armnn/DeviceSpec.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

244

src/armnn/LoadedNetwork.hpp

245

src/armnn/LoadedNetwork.cpp

246

src/armnn/Exceptions.cpp

247

src/armnn/Graph.hpp

248

src/armnn/Graph.cpp

249

src/armnn/Network.hpp

250

src/armnn/Network.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

251

src/armnn/NetworkUtils.hpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

252

src/armnn/ProfilingEvent.cpp

253

src/armnn/ProfilingEvent.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

254

src/armnn/Profiling.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

255

src/armnn/Instrument.hpp

256

src/armnn/WallClockTimer.hpp

257

src/armnn/WallClockTimer.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

258

src/armnn/Tensor.cpp

259

src/armnn/Utils.cpp

260

src/armnn/LayerSupport.cpp

261

src/armnn/LayerSupportCommon.hpp

262

src/armnn/optimizations/All.hpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

263

src/armnn/optimizations/ConvertConstants.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

264

src/armnn/optimizations/MovePermuteUp.hpp

265

src/armnn/optimizations/Optimization.hpp

266

src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp

267

src/armnn/optimizations/OptimizeInversePermutes.hpp

268

src/armnn/optimizations/PermuteAsReshape.hpp

269

src/armnn/optimizations/SquashEqualSiblings.hpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

270

src/armnn/optimizations/OptimizeInverseConversions.hpp

271

src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

272

src/armnn/Optimizer.hpp

273

src/armnn/Optimizer.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

274

third-party/half/half.hpp

275

src/armnn/IGraphObservable.hpp

276

src/armnn/Observable.hpp

277

src/armnn/Observable.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

)

if(ARMCOMPUTENEON)

# Additionally include source files for ARM Compute NEON backend

282

list(APPEND armnn_sources

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

283

src/armnn/NeonInterceptorScheduler.hpp

284

src/armnn/NeonInterceptorScheduler.cpp

285

src/armnn/NeonTimer.hpp

286

src/armnn/NeonTimer.cpp)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

287

endif()

288

if(ARMCOMPUTECL)

289

# Additionally include source files for ARM Compute OpenCL backend

290

list(APPEND armnn_sources

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

291

src/armnn/OpenClTimer.cpp

292

src/armnn/OpenClTimer.hpp)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

293

endif()

294

# Files shared by all ARM Compute backends

295

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

296

list(APPEND armnn_sources

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

297

src/armnn/memory/IMemoryPool.hpp

298

src/armnn/memory/BlobMemoryPool.cpp

299

src/armnn/memory/BlobMemoryPool.hpp

300

src/armnn/memory/BlobLifetimeManager.cpp

301

src/armnn/memory/BlobLifetimeManager.hpp

302

src/armnn/memory/PoolManager.cpp

303

src/armnn/memory/PoolManager.hpp

304

src/armnn/memory/BaseMemoryManager.hpp

305

src/armnn/memory/BaseMemoryManager.cpp

306

src/armnn/memory/OffsetMemoryPool.cpp

307

src/armnn/memory/OffsetMemoryPool.hpp

308

src/armnn/memory/OffsetLifetimeManager.cpp

309

src/armnn/memory/OffsetLifetimeManager.hpp)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

310

endif()

311

312

# Files used for Streamline-based profiling backend

313

if(PROFILING_BACKEND_STREAMLINE)

314

list(APPEND armnn_sources

315

${GATOR_ROOT}/annotate/streamline_annotate.h

316

${GATOR_ROOT}/annotate/streamline_annotate.c)

317

endif()

318

319

add_library_ex(armnn SHARED ${armnn_sources})

David Beck

2018-09-19 12:03:20 +0100

[diff] [blame]

320

321

# the backends under src/backends extend the list of

322

# static libs armnn to link against

323

list(APPEND armnnLibraries armnnUtils)

324

include(src/backends/backends.cmake)

325

326

target_include_directories(armnn PRIVATE src)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

327

target_include_directories(armnn PRIVATE src/armnn)

328

target_include_directories(armnn PRIVATE src/armnnUtils)

David Beck

2018-09-19 12:03:20 +0100

[diff] [blame]

329

foreach(lib ${armnnLibraries})

330

target_link_libraries(armnn ${lib})

331

message("Adding library dependency to armnn: ${lib}")

332

endforeach()

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

333

target_link_libraries(armnn ${CMAKE_DL_LIBS})

334

335

install(TARGETS armnn DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)

336

if(BUILD_CAFFE_PARSER)

337

install(TARGETS armnnCaffeParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)

338

endif()

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

339

if(BUILD_ONNX_PARSER)

340

install(TARGETS armnnOnnxParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)

341

endif()

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

342

if(BUILD_TF_PARSER)

343

install(TARGETS armnnTfParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)

344

endif()

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

345

if(BUILD_TF_LITE_PARSER)

346

install(TARGETS armnnTfLiteParser DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)

347

endif()

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

348

install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_PREFIX}/include)

349

350

target_link_libraries(armnn ${Boost_LOG_LIBRARY} ${Boost_THREAD_LIBRARY} ${Boost_SYSTEM_LIBRARY})

351

352

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

353

target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})

354

endif()

355

if(ARMCOMPUTECL)

356

target_link_libraries(armnn ${OPENCL_LIBRARIES})

357

endif()

358

359

if(PROFILING_BACKEND_STREAMLINE)

360

target_link_libraries(armnn pthread)

361

endif()

362

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

363

if(BUILD_UNIT_TESTS)

364

set(unittest_sources)

365

list(APPEND unittest_sources

366

src/armnn/test/UnitTests.cpp

367

src/armnn/test/UnitTests.hpp

368

src/armnn/test/EndToEndTest.cpp

369

src/armnn/test/UtilsTests.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

370

src/armnn/test/JsonPrinterTests.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

371

src/armnn/test/GraphTests.cpp

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

372

src/armnn/test/OptimizerTests.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

373

src/armnn/test/ProfilerTests.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

374

src/armnn/test/RuntimeTests.cpp

375

src/armnn/test/CreateWorkload.hpp

376

src/armnn/test/TensorTest.cpp

377

src/armnn/test/TensorHelpers.hpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

378

src/armnn/test/CsvReaderTest.cpp

379

src/armnn/test/NetworkTests.cpp

380

src/armnn/test/FloatingPointConverterTest.cpp

381

src/armnn/test/ProfilingEventTest.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

382

src/armnn/test/GraphUtils.hpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

383

src/armnn/test/InstrumentTests.cpp

384

src/armnn/test/ObservableTest.cpp

David Beck

5eec11d

2018-10-04 15:43:17 +0100

[diff] [blame]

385

src/armnn/test/OptionalTest.cpp

David Beck

2018-10-10 15:11:44 +0100

[diff] [blame]

386

src/backends/test/BackendIdTests.cpp

David Beck

32cbb0c

2018-10-09 15:46:08 +0100

[diff] [blame]

387

src/backends/test/BackendRegistryTests.cpp

David Beck

2018-09-19 12:03:20 +0100

[diff] [blame]

388

src/backends/test/IsLayerSupportedTestImpl.hpp

David Beck

2018-09-19 12:03:20 +0100

[diff] [blame]

389

src/backends/test/WorkloadDataValidation.cpp

390

src/backends/test/TensorCopyUtils.hpp

391

src/backends/test/TensorCopyUtils.cpp

392

src/backends/test/LayerTests.hpp

393

src/backends/test/LayerTests.cpp

394

src/backends/test/LayerReleaseConstantDataTest.cpp

395

src/backends/test/Conv2dTestImpl.hpp

396

src/backends/test/ActivationTestImpl.hpp

397

src/backends/test/ActivationFixture.hpp

398

src/backends/test/Pooling2dTestImpl.hpp

399

src/backends/test/ReshapeTestImpl.hpp

400

src/backends/test/PermuteTestImpl.hpp

401

src/backends/test/FullyConnectedTestImpl.hpp

402

src/backends/test/SplitterTestImpl.hpp

403

src/backends/test/NormTestImpl.hpp

404

src/backends/test/BatchNormTestImpl.hpp

405

src/backends/test/WorkloadTestUtils.hpp

David Beck

2018-09-19 12:03:20 +0100

[diff] [blame]

406

src/backends/test/QuantizeHelper.hpp)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

407

408

if(ARMCOMPUTENEON)

409

list(APPEND unittest_sources

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

410

src/armnn/test/CreateWorkloadClNeon.hpp

411

src/armnn/test/NeonTimerTest.cpp)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

endif()

if(ARMCOMPUTECL)

list(APPEND unittest_sources

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

416

src/armnn/test/CreateWorkloadClNeon.hpp

417

src/armnn/test/OpenClTimerTest.cpp

418

src/armnn/test/FP16SupportTest.cpp)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

419

endif()

420

421

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

422

list(APPEND unittest_sources

David Beck

2018-09-19 12:03:20 +0100

[diff] [blame]

423

src/backends/test/MemCopyTests.cpp)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

424

endif()

425

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

426

if(BUILD_TF_PARSER)

427

list(APPEND unittest_sources

428

src/armnnTfParser/test/Activations.cpp

429

src/armnnTfParser/test/Addition.cpp

430

src/armnnTfParser/test/BiasAdd.cpp

431

src/armnnTfParser/test/BroadcastForAdd.cpp

432

src/armnnTfParser/test/Convolution2d.cpp

433

src/armnnTfParser/test/Concat.cpp

434

src/armnnTfParser/test/ConcatOfConcats.cpp

435

src/armnnTfParser/test/DepthwiseConvolution2d.cpp

436

src/armnnTfParser/test/FusedBatchNorm.cpp

437

src/armnnTfParser/test/Identity.cpp

438

src/armnnTfParser/test/LocalResponseNormalization.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

439

src/armnnTfParser/test/MaximumForLeakyRelu.cpp

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

440

src/armnnTfParser/test/Multiplication.cpp

441

src/armnnTfParser/test/MultiOutput.cpp

442

src/armnnTfParser/test/PassThru.cpp

443

src/armnnTfParser/test/Pooling.cpp

saoste01

bbd4061

2018-08-28 15:41:51 +0100

[diff] [blame]

444

src/armnnTfParser/test/RealDiv.cpp

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

445

src/armnnTfParser/test/Reshape.cpp

446

src/armnnTfParser/test/ResizeBilinear.cpp

447

src/armnnTfParser/test/Shape.cpp

448

src/armnnTfParser/test/Softmax.cpp

449

src/armnnTfParser/test/TestDependencies.cpp

450

src/armnnTfParser/test/FullyConnected.cpp

451

src/armnnTfParser/test/Constant.cpp

452

src/armnnTfParser/test/TestMultiInputsOutputs.cpp

453

src/armnnTfParser/test/Squeeze.cpp)

454

endif()

455

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

456

if(BUILD_TF_LITE_PARSER)

457

list(APPEND unittest_sources

458

src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp

Sadik Armagan

2018-10-01 11:51:37 +0100

[diff] [blame]

459

src/armnnTfLiteParser/test/Activations.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

460

src/armnnTfLiteParser/test/AvgPool2D.cpp

Sadik Armagan

2018-10-01 11:51:37 +0100

[diff] [blame]

461

src/armnnTfLiteParser/test/Concatenation.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

462

src/armnnTfLiteParser/test/Conv2D.cpp

463

src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp

Sadik Armagan

2018-10-01 11:51:37 +0100

[diff] [blame]

464

src/armnnTfLiteParser/test/Reshape.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

465

src/armnnTfLiteParser/test/Softmax.cpp

466

src/armnnTfLiteParser/test/Squeeze.cpp

467

src/armnnTfLiteParser/test/LoadModel.cpp

468

src/armnnTfLiteParser/test/GetBuffer.cpp

469

src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp

470

src/armnnTfLiteParser/test/InputOutputTensorNames.cpp

471

src/armnnTfLiteParser/test/GetTensorIds.cpp

472

src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp

473

src/armnnTfLiteParser/test/GetInputsOutputs.cpp

)

endif()

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

477

if(BUILD_CAFFE_PARSER)

478

list(APPEND unittest_sources

479

src/armnnCaffeParser/test/TestAdd.cpp

480

src/armnnCaffeParser/test/TestConcat.cpp

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

481

src/armnnCaffeParser/test/TestConvolution.cpp

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

482

src/armnnCaffeParser/test/TestDropout.cpp

483

src/armnnCaffeParser/test/TestInputs.cpp

484

src/armnnCaffeParser/test/TestMul.cpp

485

src/armnnCaffeParser/test/TestPooling2d.cpp

486

src/armnnCaffeParser/test/TestInPlace.cpp

487

src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp

488

src/armnnCaffeParser/test/TestSplit.cpp

)

endif()

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

492

if(BUILD_ONNX_PARSER)

493

list(APPEND unittest_sources

494

src/armnnOnnxParser/test/Constructor.cpp

495

src/armnnOnnxParser/test/CreateNetwork.cpp

496

src/armnnOnnxParser/test/ProtoxtFixture.cpp

497

src/armnnOnnxParser/test/Const.cpp

498

src/armnnOnnxParser/test/Pooling.cpp

499

src/armnnOnnxParser/test/Reshape.cpp

500

src/armnnOnnxParser/test/Relu.cpp

501

src/armnnOnnxParser/test/Conv2D.cpp

502

src/armnnOnnxParser/test/Addition.cpp

503

src/armnnOnnxParser/test/FullyConnected.cpp

504

src/armnnOnnxParser/test/GetInputsOutputs.cpp

505

src/armnnOnnxParser/test/BatchNorm.cpp

506

src/armnnOnnxParser/test/DepthConv.cpp

)

endif()

Matteo Martincigh

2018-10-11 17:39:30 +0100

[diff] [blame^]

510

foreach(lib ${armnnUnitTestLibraries})

511

message("Adding library dependency to UnitTests: ${lib}")

512

list(APPEND unittest_sources $<TARGET_OBJECTS:${lib}>)

513

endforeach()

514

515

add_executable(UnitTests ${unittest_sources})

David Beck

2018-09-19 12:03:20 +0100

[diff] [blame]

516

target_include_directories(UnitTests PRIVATE src)

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

517

target_include_directories(UnitTests PRIVATE src/armnn)

518

target_include_directories(UnitTests PRIVATE src/armnnUtils)

519

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

520

if(VALGRIND_FOUND)

521

if(HEAP_PROFILING OR LEAK_CHECKING)

522

message("Valgrind is disabled for heap profiling and leak checking builds.")

523

else()

524

# Valgrind works with gperftools version number <= 2.4

525

target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")

526

endif()

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

527

endif()

528

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

529

target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

530

target_link_libraries(UnitTests ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY} ${Boost_SYSTEM_LIBRARY} ${Boost_FILESYSTEM_LIBRARY})

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

531

Matteo Martincigh

4423ac7

2018-10-11 17:39:30 +0100

[diff] [blame^]

532

target_link_libraries(UnitTests armnn)

533

target_link_libraries(UnitTests armnnUtils)

534

surmeh01

2018-03-29 16:29:27 +0100

[diff] [blame]

535

if(BUILD_TF_PARSER)

536

target_link_libraries(UnitTests armnnTfParser)

537

endif()

538

telsoa01

2018-03-09 14:13:49 +0000

[diff] [blame]

539

if(BUILD_CAFFE_PARSER)

540

target_link_libraries(UnitTests armnnCaffeParser)

541

endif()

542

telsoa01

2018-08-31 09:22:23 +0100

[diff] [blame]

543

if(BUILD_TF_LITE_PARSER)

544

target_link_libraries(UnitTests armnnTfLiteParser)

545

endif()

546

547

if(BUILD_ONNX_PARSER)

548

target_link_libraries(UnitTests armnnOnnxParser)

549

endif()

550

telsoa01