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

30

src/armnnUtils/Logging.hpp

31

src/armnnUtils/Permute.hpp

32

src/armnnUtils/Logging.cpp

33

src/armnnUtils/Permute.cpp

surmeh01

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

[diff] [blame]

34

src/armnnUtils/DotSerializer.cpp

35

src/armnnUtils/DotSerializer.hpp

surmeh01

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

[diff] [blame]

36

src/armnnUtils/HeapProfiling.cpp

37

src/armnnUtils/HeapProfiling.hpp

38

src/armnnUtils/LeakChecking.cpp

39

src/armnnUtils/LeakChecking.hpp

telsoa01

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

[diff] [blame]

40

src/armnnUtils/CsvReader.cpp

41

src/armnnUtils/CsvReader.hpp

42

src/armnnUtils/FloatingPointConverter.cpp

43

src/armnnUtils/FloatingPointConverter.hpp

44

src/armnnUtils/VerificationHelpers.hpp

45

src/armnnUtils/VerificationHelpers.cpp

telsoa01

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

[diff] [blame]

46

)

surmeh01

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

[diff] [blame]

47

if(BUILD_TF_PARSER OR BUILD_CAFFE_PARSER)

telsoa01

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

[diff] [blame]

48

list(APPEND armnnUtils_sources

49

src/armnnUtils/ParserPrototxtFixture.hpp

)

endif()

add_library_ex(armnnUtils STATIC ${armnnUtils_sources})

54

55

if(BUILD_CAFFE_PARSER)

56

# ArmNN Parser source files required for all build options

57

set(armnn_caffe_parser_sources)

58

list(APPEND armnn_caffe_parser_sources

59

include/armnnCaffeParser/ICaffeParser.hpp

telsoa01

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

[diff] [blame]

60

src/armnnCaffeParser/RecordByRecordCaffeParser.hpp

61

src/armnnCaffeParser/RecordByRecordCaffeParser.cpp

telsoa01

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

[diff] [blame]

62

src/armnnCaffeParser/CaffeParser.hpp

63

src/armnnCaffeParser/CaffeParser.cpp

64

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

65

)

66

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

67

if(COMPILER_IS_GNU_LIKE)

68

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

69

endif()

70

71

add_library_ex(armnnCaffeParser SHARED ${armnn_caffe_parser_sources})

72

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

73

74

target_include_directories(armnnCaffeParser PRIVATE src/armnnUtils)

75

76

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

77

78

target_link_libraries(armnnCaffeParser armnn)

79

target_link_libraries(armnnCaffeParser ${PROTOBUF_LIBRARIES})

telsoa01

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

[diff] [blame]

endif()

if(BUILD_ONNX_PARSER)

84

set(armnn_onnx_parser_sources)

85

list(APPEND armnn_onnx_parser_sources

86

include/armnnOnnxParser/IOnnxParser.hpp

87

src/armnnOnnxParser/OnnxParser.hpp

88

src/armnnOnnxParser/OnnxParser.cpp

89

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

90

)

91

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

92

if(COMPILER_IS_GNU_LIKE)

93

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

94

endif()

95

96

add_library_ex(armnnOnnxParser SHARED ${armnn_onnx_parser_sources})

97

98

target_include_directories(armnnOnnxParser PRIVATE src/armnnUtils)

99

100

target_link_libraries(armnnOnnxParser armnn)

101

102

# Protobuf

103

target_link_libraries(armnnOnnxParser ${PROTOBUF_LIBRARIES})

telsoa01

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

[diff] [blame]

104

endif()

105

surmeh01

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

[diff] [blame]

106

if(BUILD_TF_PARSER)

107

set(armnn_tf_parser_sources)

108

list(APPEND armnn_tf_parser_sources

109

include/armnnTfParser/ITfParser.hpp

110

src/armnnTfParser/TfParser.hpp

111

src/armnnTfParser/TfParser.cpp

112

${TF_PROTOBUFS}

113

)

114

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

115

if(COMPILER_IS_GNU_LIKE)

116

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

117

endif()

118

119

add_library_ex(armnnTfParser SHARED ${armnn_tf_parser_sources})

120

121

target_include_directories(armnnTfParser PRIVATE src/armnnUtils)

122

123

target_link_libraries(armnnTfParser armnn)

124

125

# Protobuf (use the specific version tensorflow wants)

126

target_link_libraries(armnnTfParser ${PROTOBUF_LIBRARIES})

127

endif()

128

telsoa01

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

[diff] [blame]

129

if(BUILD_TF_LITE_PARSER)

130

set(armnn_tf_lite_parser_sources)

131

list(APPEND armnn_tf_lite_parser_sources

132

include/armnnTfLiteParser/ITfLiteParser.hpp

133

src/armnnTfLiteParser/TfLiteParser.hpp

134

src/armnnTfLiteParser/TfLiteParser.cpp

135

)

136

137

add_library_ex(armnnTfLiteParser SHARED ${armnn_tf_lite_parser_sources})

138

139

target_include_directories(armnnTfLiteParser PRIVATE src/armnnUtils)

140

141

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

142

target_link_libraries(armnnTfLiteParser armnn ${FLATBUFFERS_LIBRARY})

143

endif()

144

telsoa01

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

[diff] [blame]

145

list(APPEND armnn_sources

146

include/armnn/ArmNN.hpp

147

include/armnn/Descriptors.hpp

148

include/armnn/DescriptorsFwd.hpp

149

include/armnn/IRuntime.hpp

150

include/armnn/INetwork.hpp

151

include/armnn/Tensor.hpp

152

include/armnn/TensorFwd.hpp

153

include/armnn/Types.hpp

154

include/armnn/TypesUtils.hpp

155

include/armnn/Utils.hpp

156

include/armnn/LayerSupport.hpp

157

include/armnn/Version.hpp

surmeh01

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

[diff] [blame]

158

src/armnn/layers/LayerCloneBase.hpp

159

src/armnn/layers/LayerWithParameters.hpp

160

src/armnn/layers/ActivationLayer.hpp

161

src/armnn/layers/ActivationLayer.cpp

162

src/armnn/layers/AdditionLayer.hpp

163

src/armnn/layers/AdditionLayer.cpp

Les Bell

033626d

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

[diff] [blame]

164

src/armnn/layers/ArithmeticBaseLayer.hpp

165

src/armnn/layers/ArithmeticBaseLayer.cpp

surmeh01

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

[diff] [blame]

166

src/armnn/layers/BatchNormalizationLayer.hpp

167

src/armnn/layers/BatchNormalizationLayer.cpp

168

src/armnn/layers/ConstantLayer.hpp

169

src/armnn/layers/ConstantLayer.cpp

170

src/armnn/layers/Convolution2dLayer.hpp

171

src/armnn/layers/Convolution2dLayer.cpp

telsoa01

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

[diff] [blame]

172

src/armnn/layers/ConvertFp16ToFp32Layer.hpp

173

src/armnn/layers/ConvertFp16ToFp32Layer.cpp

174

src/armnn/layers/ConvertFp32ToFp16Layer.hpp

175

src/armnn/layers/ConvertFp32ToFp16Layer.cpp

surmeh01

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

[diff] [blame]

176

src/armnn/layers/DepthwiseConvolution2dLayer.hpp

177

src/armnn/layers/DepthwiseConvolution2dLayer.cpp

178

src/armnn/layers/FakeQuantizationLayer.hpp

179

src/armnn/layers/FakeQuantizationLayer.cpp

180

src/armnn/layers/FloorLayer.hpp

181

src/armnn/layers/FloorLayer.cpp

182

src/armnn/layers/FullyConnectedLayer.hpp

183

src/armnn/layers/FullyConnectedLayer.cpp

184

src/armnn/layers/InputLayer.hpp

185

src/armnn/layers/InputLayer.cpp

186

src/armnn/layers/L2NormalizationLayer.hpp

187

src/armnn/layers/L2NormalizationLayer.cpp

telsoa01

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

[diff] [blame]

188

src/armnn/layers/LstmLayer.cpp

189

src/armnn/layers/LstmLayer.hpp

narpra01

32b9046

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

[diff] [blame]

190

src/armnn/layers/MeanLayer.hpp

191

src/armnn/layers/MeanLayer.cpp

surmeh01

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

[diff] [blame]

192

src/armnn/layers/MemCopyLayer.hpp

193

src/armnn/layers/MemCopyLayer.cpp

194

src/armnn/layers/MergerLayer.hpp

195

src/armnn/layers/MergerLayer.cpp

196

src/armnn/layers/MultiplicationLayer.hpp

197

src/armnn/layers/MultiplicationLayer.cpp

198

src/armnn/layers/NormalizationLayer.hpp

199

src/armnn/layers/NormalizationLayer.cpp

200

src/armnn/layers/OutputLayer.hpp

201

src/armnn/layers/OutputLayer.cpp

202

src/armnn/layers/PermuteLayer.hpp

203

src/armnn/layers/PermuteLayer.cpp

204

src/armnn/layers/Pooling2dLayer.hpp

205

src/armnn/layers/Pooling2dLayer.cpp

Francis Murtagh

e7a86a4

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

[diff] [blame]

206

src/armnn/layers/DivisionLayer.cpp

207

src/armnn/layers/DivisionLayer.hpp

surmeh01

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

[diff] [blame]

208

src/armnn/layers/ReshapeLayer.hpp

209

src/armnn/layers/ReshapeLayer.cpp

210

src/armnn/layers/ResizeBilinearLayer.hpp

211

src/armnn/layers/ResizeBilinearLayer.cpp

212

src/armnn/layers/SoftmaxLayer.hpp

213

src/armnn/layers/SoftmaxLayer.cpp

214

src/armnn/layers/SplitterLayer.hpp

215

src/armnn/layers/SplitterLayer.cpp

David Beck

c2044fe

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

[diff] [blame]

216

src/armnn/layers/SubtractionLayer.cpp

217

src/armnn/layers/SubtractionLayer.hpp

telsoa01

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

[diff] [blame]

218

src/armnn/Half.hpp

telsoa01

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

[diff] [blame]

219

src/armnn/InternalTypes.hpp

220

src/armnn/InternalTypes.cpp

telsoa01

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

[diff] [blame]

221

src/armnn/JsonPrinter.hpp

222

src/armnn/JsonPrinter.cpp

telsoa01

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

[diff] [blame]

223

src/armnn/LayerFwd.hpp

224

src/armnn/Layer.hpp

225

src/armnn/Layer.cpp

226

src/armnn/LayersFwd.hpp

telsoa01

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

[diff] [blame]

227

src/armnn/Runtime.hpp

228

src/armnn/Runtime.cpp

surmeh01

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

[diff] [blame]

229

src/armnn/SerializeLayerParameters.cpp

230

src/armnn/SerializeLayerParameters.hpp

telsoa01

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

[diff] [blame]

231

src/armnn/Descriptors.cpp

telsoa01

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

[diff] [blame]

232

src/armnn/DeviceSpec.hpp

telsoa01

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

[diff] [blame]

233

src/armnn/LoadedNetwork.hpp

234

src/armnn/LoadedNetwork.cpp

235

src/armnn/Exceptions.cpp

236

src/armnn/Graph.hpp

237

src/armnn/Graph.cpp

238

src/armnn/Network.hpp

239

src/armnn/Network.cpp

telsoa01

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

[diff] [blame]

240

src/armnn/NetworkUtils.hpp

telsoa01

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

[diff] [blame]

241

src/armnn/ProfilingEvent.cpp

242

src/armnn/ProfilingEvent.hpp

telsoa01

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

[diff] [blame]

243

src/armnn/Profiling.cpp

telsoa01

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

[diff] [blame]

244

src/armnn/Instrument.hpp

245

src/armnn/WallClockTimer.hpp

246

src/armnn/WallClockTimer.cpp

telsoa01

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

[diff] [blame]

247

src/armnn/Tensor.cpp

248

src/armnn/Utils.cpp

249

src/armnn/LayerSupport.cpp

250

src/armnn/LayerSupportCommon.hpp

251

src/armnn/optimizations/All.hpp

telsoa01

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

[diff] [blame]

252

src/armnn/optimizations/ConvertConstants.hpp

telsoa01

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

[diff] [blame]

253

src/armnn/optimizations/MovePermuteUp.hpp

254

src/armnn/optimizations/Optimization.hpp

255

src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp

256

src/armnn/optimizations/OptimizeInversePermutes.hpp

257

src/armnn/optimizations/PermuteAsReshape.hpp

258

src/armnn/optimizations/SquashEqualSiblings.hpp

telsoa01

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

[diff] [blame]

259

src/armnn/optimizations/OptimizeInverseConversions.hpp

260

src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp

telsoa01

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

[diff] [blame]

261

src/armnn/Optimizer.hpp

262

src/armnn/Optimizer.cpp

telsoa01

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

[diff] [blame]

263

third-party/half/half.hpp

264

src/armnn/IGraphObservable.hpp

265

src/armnn/Observable.hpp

266

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

271

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

272

src/armnn/NeonInterceptorScheduler.hpp

273

src/armnn/NeonInterceptorScheduler.cpp

274

src/armnn/NeonTimer.hpp

275

src/armnn/NeonTimer.cpp)

telsoa01

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

[diff] [blame]

276

endif()

277

if(ARMCOMPUTECL)

278

# Additionally include source files for ARM Compute OpenCL backend

279

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

280

src/armnn/OpenClTimer.cpp

281

src/armnn/OpenClTimer.hpp)

telsoa01

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

[diff] [blame]

282

endif()

283

# Files shared by all ARM Compute backends

284

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

285

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

286

src/armnn/memory/IMemoryPool.hpp

287

src/armnn/memory/BlobMemoryPool.cpp

288

src/armnn/memory/BlobMemoryPool.hpp

289

src/armnn/memory/BlobLifetimeManager.cpp

290

src/armnn/memory/BlobLifetimeManager.hpp

291

src/armnn/memory/PoolManager.cpp

292

src/armnn/memory/PoolManager.hpp

293

src/armnn/memory/BaseMemoryManager.hpp

294

src/armnn/memory/BaseMemoryManager.cpp

295

src/armnn/memory/OffsetMemoryPool.cpp

296

src/armnn/memory/OffsetMemoryPool.hpp

297

src/armnn/memory/OffsetLifetimeManager.cpp

298

src/armnn/memory/OffsetLifetimeManager.hpp)

telsoa01

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

[diff] [blame]

299

endif()

300

301

# Files used for Streamline-based profiling backend

302

if(PROFILING_BACKEND_STREAMLINE)

303

list(APPEND armnn_sources

304

${GATOR_ROOT}/annotate/streamline_annotate.h

305

${GATOR_ROOT}/annotate/streamline_annotate.c)

306

endif()

307

308

add_library_ex(armnn SHARED ${armnn_sources})

David Beck

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

[diff] [blame^]

309

310

# the backends under src/backends extend the list of

311

# static libs armnn to link against

312

list(APPEND armnnLibraries armnnUtils)

313

include(src/backends/backends.cmake)

314

315

target_include_directories(armnn PRIVATE src)

telsoa01

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

[diff] [blame]

316

target_include_directories(armnn PRIVATE src/armnn)

317

target_include_directories(armnn PRIVATE src/armnnUtils)

David Beck

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

[diff] [blame^]

318

foreach(lib ${armnnLibraries})

319

target_link_libraries(armnn ${lib})

320

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

321

endforeach()

telsoa01

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

[diff] [blame]

322

target_link_libraries(armnn ${CMAKE_DL_LIBS})

323

324

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

325

if(BUILD_CAFFE_PARSER)

326

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

327

endif()

telsoa01

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

[diff] [blame]

328

if(BUILD_ONNX_PARSER)

329

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

330

endif()

surmeh01

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

[diff] [blame]

331

if(BUILD_TF_PARSER)

332

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

333

endif()

telsoa01

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

[diff] [blame]

334

if(BUILD_TF_LITE_PARSER)

335

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

336

endif()

telsoa01

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

[diff] [blame]

337

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

338

339

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

340

341

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

342

target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})

343

endif()

344

if(ARMCOMPUTECL)

345

target_link_libraries(armnn ${OPENCL_LIBRARIES})

346

endif()

347

348

if(PROFILING_BACKEND_STREAMLINE)

349

target_link_libraries(armnn pthread)

350

endif()

351

telsoa01

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

[diff] [blame]

352

if(BUILD_UNIT_TESTS)

353

set(unittest_sources)

354

list(APPEND unittest_sources

355

src/armnn/test/UnitTests.cpp

356

src/armnn/test/UnitTests.hpp

357

src/armnn/test/EndToEndTest.cpp

358

src/armnn/test/UtilsTests.cpp

telsoa01

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

[diff] [blame]

359

src/armnn/test/JsonPrinterTests.cpp

telsoa01

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

[diff] [blame]

360

src/armnn/test/GraphTests.cpp

surmeh01

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

[diff] [blame]

361

src/armnn/test/OptimizerTests.cpp

telsoa01

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

[diff] [blame]

362

src/armnn/test/ProfilerTests.cpp

telsoa01

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

[diff] [blame]

363

src/armnn/test/RuntimeTests.cpp

364

src/armnn/test/CreateWorkload.hpp

365

src/armnn/test/TensorTest.cpp

366

src/armnn/test/TensorHelpers.hpp

telsoa01

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

[diff] [blame]

367

src/armnn/test/CsvReaderTest.cpp

368

src/armnn/test/NetworkTests.cpp

369

src/armnn/test/FloatingPointConverterTest.cpp

370

src/armnn/test/ProfilingEventTest.cpp

telsoa01

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

[diff] [blame]

371

src/armnn/test/GraphUtils.hpp

telsoa01

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

[diff] [blame]

372

src/armnn/test/InstrumentTests.cpp

373

src/armnn/test/ObservableTest.cpp

David Beck

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

[diff] [blame^]

374

src/backends/test/IsLayerSupportedTest.cpp

375

src/backends/test/IsLayerSupportedTestImpl.hpp

376

src/backends/test/Reference.cpp

377

src/backends/test/WorkloadDataValidation.cpp

378

src/backends/test/TensorCopyUtils.hpp

379

src/backends/test/TensorCopyUtils.cpp

380

src/backends/test/LayerTests.hpp

381

src/backends/test/LayerTests.cpp

382

src/backends/test/LayerReleaseConstantDataTest.cpp

383

src/backends/test/Conv2dTestImpl.hpp

384

src/backends/test/ActivationTestImpl.hpp

385

src/backends/test/ActivationFixture.hpp

386

src/backends/test/Pooling2dTestImpl.hpp

387

src/backends/test/ReshapeTestImpl.hpp

388

src/backends/test/PermuteTestImpl.hpp

389

src/backends/test/FullyConnectedTestImpl.hpp

390

src/backends/test/SplitterTestImpl.hpp

391

src/backends/test/NormTestImpl.hpp

392

src/backends/test/BatchNormTestImpl.hpp

393

src/backends/test/WorkloadTestUtils.hpp

394

src/backends/test/CreateWorkloadRef.cpp

395

src/backends/test/QuantizeHelper.hpp)

telsoa01

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

[diff] [blame]

396

397

if(ARMCOMPUTENEON)

398

list(APPEND unittest_sources

David Beck

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

[diff] [blame^]

399

src/backends/test/ArmComputeNeon.cpp

400

src/backends/test/CreateWorkloadNeon.cpp

telsoa01

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

[diff] [blame]

401

src/armnn/test/CreateWorkloadClNeon.hpp

402

src/armnn/test/NeonTimerTest.cpp)

telsoa01

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

[diff] [blame]

endif()

if(ARMCOMPUTECL)

list(APPEND unittest_sources

David Beck

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

[diff] [blame^]

407

src/backends/test/ArmComputeCl.cpp

408

src/backends/test/ClContextControlFixture.hpp

409

src/backends/test/CreateWorkloadCl.cpp

telsoa01

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

[diff] [blame]

410

src/armnn/test/CreateWorkloadClNeon.hpp

411

src/armnn/test/OpenClTimerTest.cpp

412

src/armnn/test/FP16SupportTest.cpp)

telsoa01

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

[diff] [blame]

413

endif()

414

415

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

416

list(APPEND unittest_sources

David Beck

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

[diff] [blame^]

417

src/backends/test/MemCopyTests.cpp)

telsoa01

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

[diff] [blame]

418

endif()

419

surmeh01

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

[diff] [blame]

420

if(BUILD_TF_PARSER)

421

list(APPEND unittest_sources

422

src/armnnTfParser/test/Activations.cpp

423

src/armnnTfParser/test/Addition.cpp

424

src/armnnTfParser/test/BiasAdd.cpp

425

src/armnnTfParser/test/BroadcastForAdd.cpp

426

src/armnnTfParser/test/Convolution2d.cpp

427

src/armnnTfParser/test/Concat.cpp

428

src/armnnTfParser/test/ConcatOfConcats.cpp

429

src/armnnTfParser/test/DepthwiseConvolution2d.cpp

430

src/armnnTfParser/test/FusedBatchNorm.cpp

431

src/armnnTfParser/test/Identity.cpp

432

src/armnnTfParser/test/LocalResponseNormalization.cpp

telsoa01

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

[diff] [blame]

433

src/armnnTfParser/test/MaximumForLeakyRelu.cpp

surmeh01

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

[diff] [blame]

434

src/armnnTfParser/test/Multiplication.cpp

435

src/armnnTfParser/test/MultiOutput.cpp

436

src/armnnTfParser/test/PassThru.cpp

437

src/armnnTfParser/test/Pooling.cpp

saoste01

bbd4061

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

[diff] [blame]

438

src/armnnTfParser/test/RealDiv.cpp

surmeh01

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

[diff] [blame]

439

src/armnnTfParser/test/Reshape.cpp

440

src/armnnTfParser/test/ResizeBilinear.cpp

441

src/armnnTfParser/test/Shape.cpp

442

src/armnnTfParser/test/Softmax.cpp

443

src/armnnTfParser/test/TestDependencies.cpp

444

src/armnnTfParser/test/FullyConnected.cpp

445

src/armnnTfParser/test/Constant.cpp

446

src/armnnTfParser/test/TestMultiInputsOutputs.cpp

447

src/armnnTfParser/test/Squeeze.cpp)

448

endif()

449

telsoa01

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

[diff] [blame]

450

if(BUILD_TF_LITE_PARSER)

451

list(APPEND unittest_sources

452

src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp

453

src/armnnTfLiteParser/test/AvgPool2D.cpp

454

src/armnnTfLiteParser/test/Conv2D.cpp

455

src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp

456

src/armnnTfLiteParser/test/Softmax.cpp

457

src/armnnTfLiteParser/test/Squeeze.cpp

458

src/armnnTfLiteParser/test/LoadModel.cpp

459

src/armnnTfLiteParser/test/GetBuffer.cpp

460

src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp

461

src/armnnTfLiteParser/test/InputOutputTensorNames.cpp

462

src/armnnTfLiteParser/test/GetTensorIds.cpp

463

src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp

464

src/armnnTfLiteParser/test/GetInputsOutputs.cpp

Sadik Armagan

58f3919

2018-09-17 14:14:39 +0100

[diff] [blame]

465

src/armnnTfLiteParser/test/Activations.cpp

Sadik

b94967b

2018-09-19 15:30:00 +0100

[diff] [blame]

466

src/armnnTfLiteParser/test/Reshape.cpp

telsoa01

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

[diff] [blame]

)

endif()

telsoa01

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

[diff] [blame]

470

if(BUILD_CAFFE_PARSER)

471

list(APPEND unittest_sources

472

src/armnnCaffeParser/test/TestAdd.cpp

473

src/armnnCaffeParser/test/TestConcat.cpp

telsoa01

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

[diff] [blame]

474

src/armnnCaffeParser/test/TestConvolution.cpp

telsoa01

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

[diff] [blame]

475

src/armnnCaffeParser/test/TestDropout.cpp

476

src/armnnCaffeParser/test/TestInputs.cpp

477

src/armnnCaffeParser/test/TestMul.cpp

478

src/armnnCaffeParser/test/TestPooling2d.cpp

479

src/armnnCaffeParser/test/TestInPlace.cpp

480

src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp

481

src/armnnCaffeParser/test/TestSplit.cpp

)

endif()

telsoa01

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

[diff] [blame]

485

if(BUILD_ONNX_PARSER)

486

list(APPEND unittest_sources

487

src/armnnOnnxParser/test/Constructor.cpp

488

src/armnnOnnxParser/test/CreateNetwork.cpp

489

src/armnnOnnxParser/test/ProtoxtFixture.cpp

490

src/armnnOnnxParser/test/Const.cpp

491

src/armnnOnnxParser/test/Pooling.cpp

492

src/armnnOnnxParser/test/Reshape.cpp

493

src/armnnOnnxParser/test/Relu.cpp

494

src/armnnOnnxParser/test/Conv2D.cpp

495

src/armnnOnnxParser/test/Addition.cpp

496

src/armnnOnnxParser/test/FullyConnected.cpp

497

src/armnnOnnxParser/test/GetInputsOutputs.cpp

498

src/armnnOnnxParser/test/BatchNorm.cpp

499

src/armnnOnnxParser/test/DepthConv.cpp

)

endif()

telsoa01

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

[diff] [blame]

503

add_executable_ex(UnitTests ${unittest_sources})

David Beck

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

[diff] [blame^]

504

target_include_directories(UnitTests PRIVATE src)

telsoa01

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

[diff] [blame]

505

target_include_directories(UnitTests PRIVATE src/armnn)

506

target_include_directories(UnitTests PRIVATE src/armnnUtils)

507

telsoa01

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

[diff] [blame]

508

if(VALGRIND_FOUND)

509

if(HEAP_PROFILING OR LEAK_CHECKING)

510

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

511

else()

512

# Valgrind works with gperftools version number <= 2.4

513

target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")

514

endif()

telsoa01

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

[diff] [blame]

515

endif()

516

517

target_link_libraries(UnitTests armnn)

surmeh01

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

[diff] [blame]

518

target_link_libraries(UnitTests armnnUtils)

telsoa01

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

[diff] [blame]

519

target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})

telsoa01

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

[diff] [blame]

520

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]

521

surmeh01

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

[diff] [blame]

522

if(BUILD_TF_PARSER)

523

target_link_libraries(UnitTests armnnTfParser)

524

endif()

525

telsoa01

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

[diff] [blame]

526

if(BUILD_CAFFE_PARSER)

527

target_link_libraries(UnitTests armnnCaffeParser)

528

endif()

529

telsoa01

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

[diff] [blame]

530

if(BUILD_TF_LITE_PARSER)

531

target_link_libraries(UnitTests armnnTfLiteParser)

532

endif()

533

534

if(BUILD_ONNX_PARSER)

535

target_link_libraries(UnitTests armnnOnnxParser)

536

endif()

537

telsoa01