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

150

include/armnn/Descriptors.hpp

151

include/armnn/DescriptorsFwd.hpp

152

include/armnn/IRuntime.hpp

153

include/armnn/INetwork.hpp

154

include/armnn/Tensor.hpp

155

include/armnn/TensorFwd.hpp

156

include/armnn/Types.hpp

157

include/armnn/TypesUtils.hpp

158

include/armnn/Utils.hpp

159

include/armnn/LayerSupport.hpp

160

include/armnn/Version.hpp

surmeh01

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

[diff] [blame]

161

src/armnn/layers/LayerCloneBase.hpp

162

src/armnn/layers/LayerWithParameters.hpp

163

src/armnn/layers/ActivationLayer.hpp

164

src/armnn/layers/ActivationLayer.cpp

165

src/armnn/layers/AdditionLayer.hpp

166

src/armnn/layers/AdditionLayer.cpp

Les Bell

033626d

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

[diff] [blame]

167

src/armnn/layers/ArithmeticBaseLayer.hpp

168

src/armnn/layers/ArithmeticBaseLayer.cpp

surmeh01

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

[diff] [blame]

169

src/armnn/layers/BatchNormalizationLayer.hpp

170

src/armnn/layers/BatchNormalizationLayer.cpp

171

src/armnn/layers/ConstantLayer.hpp

172

src/armnn/layers/ConstantLayer.cpp

173

src/armnn/layers/Convolution2dLayer.hpp

174

src/armnn/layers/Convolution2dLayer.cpp

telsoa01

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

[diff] [blame]

175

src/armnn/layers/ConvertFp16ToFp32Layer.hpp

176

src/armnn/layers/ConvertFp16ToFp32Layer.cpp

177

src/armnn/layers/ConvertFp32ToFp16Layer.hpp

178

src/armnn/layers/ConvertFp32ToFp16Layer.cpp

surmeh01

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

[diff] [blame]

179

src/armnn/layers/DepthwiseConvolution2dLayer.hpp

180

src/armnn/layers/DepthwiseConvolution2dLayer.cpp

181

src/armnn/layers/FakeQuantizationLayer.hpp

182

src/armnn/layers/FakeQuantizationLayer.cpp

183

src/armnn/layers/FloorLayer.hpp

184

src/armnn/layers/FloorLayer.cpp

185

src/armnn/layers/FullyConnectedLayer.hpp

186

src/armnn/layers/FullyConnectedLayer.cpp

187

src/armnn/layers/InputLayer.hpp

188

src/armnn/layers/InputLayer.cpp

189

src/armnn/layers/L2NormalizationLayer.hpp

190

src/armnn/layers/L2NormalizationLayer.cpp

telsoa01

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

[diff] [blame]

191

src/armnn/layers/LstmLayer.cpp

192

src/armnn/layers/LstmLayer.hpp

narpra01

32b9046

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

[diff] [blame]

193

src/armnn/layers/MeanLayer.hpp

194

src/armnn/layers/MeanLayer.cpp

surmeh01

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

[diff] [blame]

195

src/armnn/layers/MemCopyLayer.hpp

196

src/armnn/layers/MemCopyLayer.cpp

197

src/armnn/layers/MergerLayer.hpp

198

src/armnn/layers/MergerLayer.cpp

199

src/armnn/layers/MultiplicationLayer.hpp

200

src/armnn/layers/MultiplicationLayer.cpp

201

src/armnn/layers/NormalizationLayer.hpp

202

src/armnn/layers/NormalizationLayer.cpp

203

src/armnn/layers/OutputLayer.hpp

204

src/armnn/layers/OutputLayer.cpp

Mohamed Nour Abouelseoud

5662c20

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

[diff] [blame]

205

src/armnn/layers/PadLayer.hpp

206

src/armnn/layers/PadLayer.cpp

surmeh01

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

[diff] [blame]

207

src/armnn/layers/PermuteLayer.hpp

208

src/armnn/layers/PermuteLayer.cpp

209

src/armnn/layers/Pooling2dLayer.hpp

210

src/armnn/layers/Pooling2dLayer.cpp

Francis Murtagh

e7a86a4

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

[diff] [blame]

211

src/armnn/layers/DivisionLayer.cpp

212

src/armnn/layers/DivisionLayer.hpp

surmeh01

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

[diff] [blame]

213

src/armnn/layers/ReshapeLayer.hpp

214

src/armnn/layers/ReshapeLayer.cpp

215

src/armnn/layers/ResizeBilinearLayer.hpp

216

src/armnn/layers/ResizeBilinearLayer.cpp

217

src/armnn/layers/SoftmaxLayer.hpp

218

src/armnn/layers/SoftmaxLayer.cpp

219

src/armnn/layers/SplitterLayer.hpp

220

src/armnn/layers/SplitterLayer.cpp

David Beck

c2044fe

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

[diff] [blame]

221

src/armnn/layers/SubtractionLayer.cpp

222

src/armnn/layers/SubtractionLayer.hpp

telsoa01

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

[diff] [blame]

223

src/armnn/InternalTypes.hpp

224

src/armnn/InternalTypes.cpp

telsoa01

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

[diff] [blame]

225

src/armnn/JsonPrinter.hpp

226

src/armnn/JsonPrinter.cpp

telsoa01

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

[diff] [blame]

227

src/armnn/LayerFwd.hpp

228

src/armnn/Layer.hpp

229

src/armnn/Layer.cpp

230

src/armnn/LayersFwd.hpp

telsoa01

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

[diff] [blame]

231

src/armnn/Runtime.hpp

232

src/armnn/Runtime.cpp

surmeh01

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

[diff] [blame]

233

src/armnn/SerializeLayerParameters.cpp

234

src/armnn/SerializeLayerParameters.hpp

telsoa01

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

[diff] [blame]

235

src/armnn/Descriptors.cpp

telsoa01

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

[diff] [blame]

236

src/armnn/DeviceSpec.hpp

telsoa01

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

[diff] [blame]

237

src/armnn/LoadedNetwork.hpp

238

src/armnn/LoadedNetwork.cpp

239

src/armnn/Exceptions.cpp

240

src/armnn/Graph.hpp

241

src/armnn/Graph.cpp

242

src/armnn/Network.hpp

243

src/armnn/Network.cpp

telsoa01

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

[diff] [blame]

244

src/armnn/NetworkUtils.hpp

telsoa01

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

[diff] [blame]

245

src/armnn/ProfilingEvent.cpp

246

src/armnn/ProfilingEvent.hpp

telsoa01

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

[diff] [blame]

247

src/armnn/Profiling.cpp

telsoa01

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

[diff] [blame]

248

src/armnn/Instrument.hpp

249

src/armnn/WallClockTimer.hpp

250

src/armnn/WallClockTimer.cpp

telsoa01

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

[diff] [blame]

251

src/armnn/Tensor.cpp

252

src/armnn/Utils.cpp

253

src/armnn/LayerSupport.cpp

254

src/armnn/LayerSupportCommon.hpp

255

src/armnn/optimizations/All.hpp

telsoa01

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

[diff] [blame]

256

src/armnn/optimizations/ConvertConstants.hpp

telsoa01

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

[diff] [blame]

257

src/armnn/optimizations/MovePermuteUp.hpp

258

src/armnn/optimizations/Optimization.hpp

259

src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp

260

src/armnn/optimizations/OptimizeInversePermutes.hpp

261

src/armnn/optimizations/PermuteAsReshape.hpp

262

src/armnn/optimizations/SquashEqualSiblings.hpp

telsoa01

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

[diff] [blame]

263

src/armnn/optimizations/OptimizeInverseConversions.hpp

264

src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp

telsoa01

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

[diff] [blame]

265

src/armnn/Optimizer.hpp

266

src/armnn/Optimizer.cpp

telsoa01

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

[diff] [blame]

267

third-party/half/half.hpp

268

src/armnn/IGraphObservable.hpp

269

src/armnn/Observable.hpp

270

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

275

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

276

src/armnn/NeonInterceptorScheduler.hpp

277

src/armnn/NeonInterceptorScheduler.cpp

278

src/armnn/NeonTimer.hpp

279

src/armnn/NeonTimer.cpp)

telsoa01

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

[diff] [blame]

280

endif()

281

if(ARMCOMPUTECL)

282

# Additionally include source files for ARM Compute OpenCL backend

283

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

284

src/armnn/OpenClTimer.cpp

285

src/armnn/OpenClTimer.hpp)

telsoa01

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

[diff] [blame]

286

endif()

287

# Files shared by all ARM Compute backends

288

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

289

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

290

src/armnn/memory/IMemoryPool.hpp

291

src/armnn/memory/BlobMemoryPool.cpp

292

src/armnn/memory/BlobMemoryPool.hpp

293

src/armnn/memory/BlobLifetimeManager.cpp

294

src/armnn/memory/BlobLifetimeManager.hpp

295

src/armnn/memory/PoolManager.cpp

296

src/armnn/memory/PoolManager.hpp

297

src/armnn/memory/BaseMemoryManager.hpp

298

src/armnn/memory/BaseMemoryManager.cpp

299

src/armnn/memory/OffsetMemoryPool.cpp

300

src/armnn/memory/OffsetMemoryPool.hpp

301

src/armnn/memory/OffsetLifetimeManager.cpp

302

src/armnn/memory/OffsetLifetimeManager.hpp)

telsoa01

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

[diff] [blame]

303

endif()

304

305

# Files used for Streamline-based profiling backend

306

if(PROFILING_BACKEND_STREAMLINE)

307

list(APPEND armnn_sources

308

${GATOR_ROOT}/annotate/streamline_annotate.h

309

${GATOR_ROOT}/annotate/streamline_annotate.c)

310

endif()

311

312

add_library_ex(armnn SHARED ${armnn_sources})

David Beck

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

[diff] [blame]

313

314

# the backends under src/backends extend the list of

315

# static libs armnn to link against

316

list(APPEND armnnLibraries armnnUtils)

317

include(src/backends/backends.cmake)

318

319

target_include_directories(armnn PRIVATE src)

telsoa01

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

[diff] [blame]

320

target_include_directories(armnn PRIVATE src/armnn)

321

target_include_directories(armnn PRIVATE src/armnnUtils)

David Beck

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

[diff] [blame]

322

foreach(lib ${armnnLibraries})

323

target_link_libraries(armnn ${lib})

324

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

325

endforeach()

telsoa01

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

[diff] [blame]

326

target_link_libraries(armnn ${CMAKE_DL_LIBS})

327

328

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

329

if(BUILD_CAFFE_PARSER)

330

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

331

endif()

telsoa01

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

[diff] [blame]

332

if(BUILD_ONNX_PARSER)

333

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

334

endif()

surmeh01

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

[diff] [blame]

335

if(BUILD_TF_PARSER)

336

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

337

endif()

telsoa01

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

[diff] [blame]

338

if(BUILD_TF_LITE_PARSER)

339

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

340

endif()

telsoa01

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

[diff] [blame]

341

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

342

343

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

344

345

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

346

target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})

347

endif()

348

if(ARMCOMPUTECL)

349

target_link_libraries(armnn ${OPENCL_LIBRARIES})

350

endif()

351

352

if(PROFILING_BACKEND_STREAMLINE)

353

target_link_libraries(armnn pthread)

354

endif()

355

telsoa01

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

[diff] [blame]

356

if(BUILD_UNIT_TESTS)

357

set(unittest_sources)

358

list(APPEND unittest_sources

359

src/armnn/test/UnitTests.cpp

360

src/armnn/test/UnitTests.hpp

361

src/armnn/test/EndToEndTest.cpp

362

src/armnn/test/UtilsTests.cpp

telsoa01

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

[diff] [blame]

363

src/armnn/test/JsonPrinterTests.cpp

telsoa01

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

[diff] [blame]

364

src/armnn/test/GraphTests.cpp

surmeh01

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

[diff] [blame]

365

src/armnn/test/OptimizerTests.cpp

telsoa01

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

[diff] [blame]

366

src/armnn/test/ProfilerTests.cpp

telsoa01

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

[diff] [blame]

367

src/armnn/test/RuntimeTests.cpp

368

src/armnn/test/CreateWorkload.hpp

369

src/armnn/test/TensorTest.cpp

370

src/armnn/test/TensorHelpers.hpp

telsoa01

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

[diff] [blame]

371

src/armnn/test/CsvReaderTest.cpp

372

src/armnn/test/NetworkTests.cpp

373

src/armnn/test/FloatingPointConverterTest.cpp

374

src/armnn/test/ProfilingEventTest.cpp

telsoa01

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

[diff] [blame]

375

src/armnn/test/GraphUtils.hpp

telsoa01

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

[diff] [blame]

376

src/armnn/test/InstrumentTests.cpp

377

src/armnn/test/ObservableTest.cpp

David Beck

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

[diff] [blame]

378

src/backends/test/IsLayerSupportedTest.cpp

379

src/backends/test/IsLayerSupportedTestImpl.hpp

380

src/backends/test/Reference.cpp

381

src/backends/test/WorkloadDataValidation.cpp

382

src/backends/test/TensorCopyUtils.hpp

383

src/backends/test/TensorCopyUtils.cpp

384

src/backends/test/LayerTests.hpp

385

src/backends/test/LayerTests.cpp

386

src/backends/test/LayerReleaseConstantDataTest.cpp

387

src/backends/test/Conv2dTestImpl.hpp

388

src/backends/test/ActivationTestImpl.hpp

389

src/backends/test/ActivationFixture.hpp

390

src/backends/test/Pooling2dTestImpl.hpp

391

src/backends/test/ReshapeTestImpl.hpp

392

src/backends/test/PermuteTestImpl.hpp

393

src/backends/test/FullyConnectedTestImpl.hpp

394

src/backends/test/SplitterTestImpl.hpp

395

src/backends/test/NormTestImpl.hpp

396

src/backends/test/BatchNormTestImpl.hpp

397

src/backends/test/WorkloadTestUtils.hpp

398

src/backends/test/CreateWorkloadRef.cpp

399

src/backends/test/QuantizeHelper.hpp)

telsoa01

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

[diff] [blame]

400

401

if(ARMCOMPUTENEON)

402

list(APPEND unittest_sources

David Beck

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

[diff] [blame]

403

src/backends/test/ArmComputeNeon.cpp

404

src/backends/test/CreateWorkloadNeon.cpp

telsoa01

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

[diff] [blame]

405

src/armnn/test/CreateWorkloadClNeon.hpp

406

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]

411

src/backends/test/ArmComputeCl.cpp

412

src/backends/test/ClContextControlFixture.hpp

413

src/backends/test/CreateWorkloadCl.cpp

telsoa01

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

[diff] [blame]

414

src/armnn/test/CreateWorkloadClNeon.hpp

415

src/armnn/test/OpenClTimerTest.cpp

416

src/armnn/test/FP16SupportTest.cpp)

telsoa01

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

[diff] [blame]

417

endif()

418

419

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

420

list(APPEND unittest_sources

David Beck

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

[diff] [blame]

421

src/backends/test/MemCopyTests.cpp)

telsoa01

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

[diff] [blame]

422

endif()

423

surmeh01

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

[diff] [blame]

424

if(BUILD_TF_PARSER)

425

list(APPEND unittest_sources

426

src/armnnTfParser/test/Activations.cpp

427

src/armnnTfParser/test/Addition.cpp

428

src/armnnTfParser/test/BiasAdd.cpp

429

src/armnnTfParser/test/BroadcastForAdd.cpp

430

src/armnnTfParser/test/Convolution2d.cpp

431

src/armnnTfParser/test/Concat.cpp

432

src/armnnTfParser/test/ConcatOfConcats.cpp

433

src/armnnTfParser/test/DepthwiseConvolution2d.cpp

434

src/armnnTfParser/test/FusedBatchNorm.cpp

435

src/armnnTfParser/test/Identity.cpp

436

src/armnnTfParser/test/LocalResponseNormalization.cpp

telsoa01

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

[diff] [blame]

437

src/armnnTfParser/test/MaximumForLeakyRelu.cpp

surmeh01

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

[diff] [blame]

438

src/armnnTfParser/test/Multiplication.cpp

439

src/armnnTfParser/test/MultiOutput.cpp

440

src/armnnTfParser/test/PassThru.cpp

441

src/armnnTfParser/test/Pooling.cpp

saoste01

bbd4061

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

[diff] [blame]

442

src/armnnTfParser/test/RealDiv.cpp

surmeh01

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

[diff] [blame]

443

src/armnnTfParser/test/Reshape.cpp

444

src/armnnTfParser/test/ResizeBilinear.cpp

445

src/armnnTfParser/test/Shape.cpp

446

src/armnnTfParser/test/Softmax.cpp

447

src/armnnTfParser/test/TestDependencies.cpp

448

src/armnnTfParser/test/FullyConnected.cpp

449

src/armnnTfParser/test/Constant.cpp

450

src/armnnTfParser/test/TestMultiInputsOutputs.cpp

451

src/armnnTfParser/test/Squeeze.cpp)

452

endif()

453

telsoa01

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

[diff] [blame]

454

if(BUILD_TF_LITE_PARSER)

455

list(APPEND unittest_sources

456

src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp

Sadik Armagan

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

[diff] [blame]

457

src/armnnTfLiteParser/test/Activations.cpp

telsoa01

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

[diff] [blame]

458

src/armnnTfLiteParser/test/AvgPool2D.cpp

Sadik Armagan

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

[diff] [blame]

459

src/armnnTfLiteParser/test/Concatenation.cpp

telsoa01

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

[diff] [blame]

460

src/armnnTfLiteParser/test/Conv2D.cpp

461

src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp

Sadik Armagan

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

[diff] [blame]

462

src/armnnTfLiteParser/test/Reshape.cpp

telsoa01

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

[diff] [blame]

463

src/armnnTfLiteParser/test/Softmax.cpp

464

src/armnnTfLiteParser/test/Squeeze.cpp

465

src/armnnTfLiteParser/test/LoadModel.cpp

466

src/armnnTfLiteParser/test/GetBuffer.cpp

467

src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp

468

src/armnnTfLiteParser/test/InputOutputTensorNames.cpp

469

src/armnnTfLiteParser/test/GetTensorIds.cpp

470

src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp

471

src/armnnTfLiteParser/test/GetInputsOutputs.cpp

)

endif()

telsoa01

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

[diff] [blame]

475

if(BUILD_CAFFE_PARSER)

476

list(APPEND unittest_sources

477

src/armnnCaffeParser/test/TestAdd.cpp

478

src/armnnCaffeParser/test/TestConcat.cpp

telsoa01

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

[diff] [blame]

479

src/armnnCaffeParser/test/TestConvolution.cpp

telsoa01

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

[diff] [blame]

480

src/armnnCaffeParser/test/TestDropout.cpp

481

src/armnnCaffeParser/test/TestInputs.cpp

482

src/armnnCaffeParser/test/TestMul.cpp

483

src/armnnCaffeParser/test/TestPooling2d.cpp

484

src/armnnCaffeParser/test/TestInPlace.cpp

485

src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp

486

src/armnnCaffeParser/test/TestSplit.cpp

)

endif()

telsoa01

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

[diff] [blame]

490

if(BUILD_ONNX_PARSER)

491

list(APPEND unittest_sources

492

src/armnnOnnxParser/test/Constructor.cpp

493

src/armnnOnnxParser/test/CreateNetwork.cpp

494

src/armnnOnnxParser/test/ProtoxtFixture.cpp

495

src/armnnOnnxParser/test/Const.cpp

496

src/armnnOnnxParser/test/Pooling.cpp

497

src/armnnOnnxParser/test/Reshape.cpp

498

src/armnnOnnxParser/test/Relu.cpp

499

src/armnnOnnxParser/test/Conv2D.cpp

500

src/armnnOnnxParser/test/Addition.cpp

501

src/armnnOnnxParser/test/FullyConnected.cpp

502

src/armnnOnnxParser/test/GetInputsOutputs.cpp

503

src/armnnOnnxParser/test/BatchNorm.cpp

504

src/armnnOnnxParser/test/DepthConv.cpp

)

endif()

telsoa01

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

[diff] [blame]

508

add_executable_ex(UnitTests ${unittest_sources})

David Beck

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

[diff] [blame]

509

target_include_directories(UnitTests PRIVATE src)

telsoa01

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

[diff] [blame]

510

target_include_directories(UnitTests PRIVATE src/armnn)

511

target_include_directories(UnitTests PRIVATE src/armnnUtils)

512

telsoa01

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

[diff] [blame]

513

if(VALGRIND_FOUND)

514

if(HEAP_PROFILING OR LEAK_CHECKING)

515

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

516

else()

517

# Valgrind works with gperftools version number <= 2.4

518

target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")

519

endif()

telsoa01

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

[diff] [blame]

520

endif()

521

522

target_link_libraries(UnitTests armnn)

surmeh01

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

[diff] [blame]

523

target_link_libraries(UnitTests armnnUtils)

telsoa01

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

[diff] [blame]

524

target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})

telsoa01

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

[diff] [blame]

525

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]

526

surmeh01

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

[diff] [blame]

527

if(BUILD_TF_PARSER)

528

target_link_libraries(UnitTests armnnTfParser)

529

endif()

530

telsoa01

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

[diff] [blame]

531

if(BUILD_CAFFE_PARSER)

532

target_link_libraries(UnitTests armnnCaffeParser)

533

endif()

534

telsoa01

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

[diff] [blame]

535

if(BUILD_TF_LITE_PARSER)

536

target_link_libraries(UnitTests armnnTfLiteParser)

537

endif()

538

539

if(BUILD_ONNX_PARSER)

540

target_link_libraries(UnitTests armnnOnnxParser)

541

endif()

542

telsoa01