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

Mohamed Nour Abouelseoud

5662c20

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

[diff] [blame^]

202

src/armnn/layers/PadLayer.hpp

203

src/armnn/layers/PadLayer.cpp

surmeh01

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

[diff] [blame]

204

src/armnn/layers/PermuteLayer.hpp

205

src/armnn/layers/PermuteLayer.cpp

206

src/armnn/layers/Pooling2dLayer.hpp

207

src/armnn/layers/Pooling2dLayer.cpp

Francis Murtagh

e7a86a4

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

[diff] [blame]

208

src/armnn/layers/DivisionLayer.cpp

209

src/armnn/layers/DivisionLayer.hpp

surmeh01

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

[diff] [blame]

210

src/armnn/layers/ReshapeLayer.hpp

211

src/armnn/layers/ReshapeLayer.cpp

212

src/armnn/layers/ResizeBilinearLayer.hpp

213

src/armnn/layers/ResizeBilinearLayer.cpp

214

src/armnn/layers/SoftmaxLayer.hpp

215

src/armnn/layers/SoftmaxLayer.cpp

216

src/armnn/layers/SplitterLayer.hpp

217

src/armnn/layers/SplitterLayer.cpp

David Beck

c2044fe

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

[diff] [blame]

218

src/armnn/layers/SubtractionLayer.cpp

219

src/armnn/layers/SubtractionLayer.hpp

telsoa01

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

[diff] [blame]

220

src/armnn/Half.hpp

telsoa01

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

[diff] [blame]

221

src/armnn/InternalTypes.hpp

222

src/armnn/InternalTypes.cpp

telsoa01

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

[diff] [blame]

223

src/armnn/JsonPrinter.hpp

224

src/armnn/JsonPrinter.cpp

telsoa01

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

[diff] [blame]

225

src/armnn/LayerFwd.hpp

226

src/armnn/Layer.hpp

227

src/armnn/Layer.cpp

228

src/armnn/LayersFwd.hpp

telsoa01

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

[diff] [blame]

229

src/armnn/Runtime.hpp

230

src/armnn/Runtime.cpp

surmeh01

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

[diff] [blame]

231

src/armnn/SerializeLayerParameters.cpp

232

src/armnn/SerializeLayerParameters.hpp

telsoa01

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

[diff] [blame]

233

src/armnn/Descriptors.cpp

telsoa01

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

[diff] [blame]

234

src/armnn/DeviceSpec.hpp

telsoa01

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

[diff] [blame]

235

src/armnn/LoadedNetwork.hpp

236

src/armnn/LoadedNetwork.cpp

237

src/armnn/Exceptions.cpp

238

src/armnn/Graph.hpp

239

src/armnn/Graph.cpp

240

src/armnn/Network.hpp

241

src/armnn/Network.cpp

telsoa01

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

[diff] [blame]

242

src/armnn/NetworkUtils.hpp

telsoa01

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

[diff] [blame]

243

src/armnn/ProfilingEvent.cpp

244

src/armnn/ProfilingEvent.hpp

telsoa01

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

[diff] [blame]

245

src/armnn/Profiling.cpp

telsoa01

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

[diff] [blame]

246

src/armnn/Instrument.hpp

247

src/armnn/WallClockTimer.hpp

248

src/armnn/WallClockTimer.cpp

telsoa01

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

[diff] [blame]

249

src/armnn/Tensor.cpp

250

src/armnn/Utils.cpp

251

src/armnn/LayerSupport.cpp

252

src/armnn/LayerSupportCommon.hpp

253

src/armnn/optimizations/All.hpp

telsoa01

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

[diff] [blame]

254

src/armnn/optimizations/ConvertConstants.hpp

telsoa01

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

[diff] [blame]

255

src/armnn/optimizations/MovePermuteUp.hpp

256

src/armnn/optimizations/Optimization.hpp

257

src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp

258

src/armnn/optimizations/OptimizeInversePermutes.hpp

259

src/armnn/optimizations/PermuteAsReshape.hpp

260

src/armnn/optimizations/SquashEqualSiblings.hpp

telsoa01

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

[diff] [blame]

261

src/armnn/optimizations/OptimizeInverseConversions.hpp

262

src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp

telsoa01

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

[diff] [blame]

263

src/armnn/Optimizer.hpp

264

src/armnn/Optimizer.cpp

telsoa01

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

[diff] [blame]

265

third-party/half/half.hpp

266

src/armnn/IGraphObservable.hpp

267

src/armnn/Observable.hpp

268

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

273

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

274

src/armnn/NeonInterceptorScheduler.hpp

275

src/armnn/NeonInterceptorScheduler.cpp

276

src/armnn/NeonTimer.hpp

277

src/armnn/NeonTimer.cpp)

telsoa01

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

[diff] [blame]

278

endif()

279

if(ARMCOMPUTECL)

280

# Additionally include source files for ARM Compute OpenCL backend

281

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

282

src/armnn/OpenClTimer.cpp

283

src/armnn/OpenClTimer.hpp)

telsoa01

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

[diff] [blame]

284

endif()

285

# Files shared by all ARM Compute backends

286

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

287

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

288

src/armnn/memory/IMemoryPool.hpp

289

src/armnn/memory/BlobMemoryPool.cpp

290

src/armnn/memory/BlobMemoryPool.hpp

291

src/armnn/memory/BlobLifetimeManager.cpp

292

src/armnn/memory/BlobLifetimeManager.hpp

293

src/armnn/memory/PoolManager.cpp

294

src/armnn/memory/PoolManager.hpp

295

src/armnn/memory/BaseMemoryManager.hpp

296

src/armnn/memory/BaseMemoryManager.cpp

297

src/armnn/memory/OffsetMemoryPool.cpp

298

src/armnn/memory/OffsetMemoryPool.hpp

299

src/armnn/memory/OffsetLifetimeManager.cpp

300

src/armnn/memory/OffsetLifetimeManager.hpp)

telsoa01

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

[diff] [blame]

301

endif()

302

303

# Files used for Streamline-based profiling backend

304

if(PROFILING_BACKEND_STREAMLINE)

305

list(APPEND armnn_sources

306

${GATOR_ROOT}/annotate/streamline_annotate.h

307

${GATOR_ROOT}/annotate/streamline_annotate.c)

308

endif()

309

310

add_library_ex(armnn SHARED ${armnn_sources})

David Beck

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

[diff] [blame]

311

312

# the backends under src/backends extend the list of

313

# static libs armnn to link against

314

list(APPEND armnnLibraries armnnUtils)

315

include(src/backends/backends.cmake)

316

317

target_include_directories(armnn PRIVATE src)

telsoa01

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

[diff] [blame]

318

target_include_directories(armnn PRIVATE src/armnn)

319

target_include_directories(armnn PRIVATE src/armnnUtils)

David Beck

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

[diff] [blame]

320

foreach(lib ${armnnLibraries})

321

target_link_libraries(armnn ${lib})

322

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

323

endforeach()

telsoa01

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

[diff] [blame]

324

target_link_libraries(armnn ${CMAKE_DL_LIBS})

325

326

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

327

if(BUILD_CAFFE_PARSER)

328

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

329

endif()

telsoa01

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

[diff] [blame]

330

if(BUILD_ONNX_PARSER)

331

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

332

endif()

surmeh01

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

[diff] [blame]

333

if(BUILD_TF_PARSER)

334

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

335

endif()

telsoa01

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

[diff] [blame]

336

if(BUILD_TF_LITE_PARSER)

337

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

338

endif()

telsoa01

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

[diff] [blame]

339

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

340

341

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

342

343

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

344

target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})

345

endif()

346

if(ARMCOMPUTECL)

347

target_link_libraries(armnn ${OPENCL_LIBRARIES})

348

endif()

349

350

if(PROFILING_BACKEND_STREAMLINE)

351

target_link_libraries(armnn pthread)

352

endif()

353

telsoa01

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

[diff] [blame]

354

if(BUILD_UNIT_TESTS)

355

set(unittest_sources)

356

list(APPEND unittest_sources

357

src/armnn/test/UnitTests.cpp

358

src/armnn/test/UnitTests.hpp

359

src/armnn/test/EndToEndTest.cpp

360

src/armnn/test/UtilsTests.cpp

telsoa01

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

[diff] [blame]

361

src/armnn/test/JsonPrinterTests.cpp

telsoa01

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

[diff] [blame]

362

src/armnn/test/GraphTests.cpp

surmeh01

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

[diff] [blame]

363

src/armnn/test/OptimizerTests.cpp

telsoa01

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

[diff] [blame]

364

src/armnn/test/ProfilerTests.cpp

telsoa01

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

[diff] [blame]

365

src/armnn/test/RuntimeTests.cpp

366

src/armnn/test/CreateWorkload.hpp

367

src/armnn/test/TensorTest.cpp

368

src/armnn/test/TensorHelpers.hpp

telsoa01

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

[diff] [blame]

369

src/armnn/test/CsvReaderTest.cpp

370

src/armnn/test/NetworkTests.cpp

371

src/armnn/test/FloatingPointConverterTest.cpp

372

src/armnn/test/ProfilingEventTest.cpp

telsoa01

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

[diff] [blame]

373

src/armnn/test/GraphUtils.hpp

telsoa01

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

[diff] [blame]

374

src/armnn/test/InstrumentTests.cpp

375

src/armnn/test/ObservableTest.cpp

David Beck

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

[diff] [blame]

376

src/backends/test/IsLayerSupportedTest.cpp

377

src/backends/test/IsLayerSupportedTestImpl.hpp

378

src/backends/test/Reference.cpp

379

src/backends/test/WorkloadDataValidation.cpp

380

src/backends/test/TensorCopyUtils.hpp

381

src/backends/test/TensorCopyUtils.cpp

382

src/backends/test/LayerTests.hpp

383

src/backends/test/LayerTests.cpp

384

src/backends/test/LayerReleaseConstantDataTest.cpp

385

src/backends/test/Conv2dTestImpl.hpp

386

src/backends/test/ActivationTestImpl.hpp

387

src/backends/test/ActivationFixture.hpp

388

src/backends/test/Pooling2dTestImpl.hpp

389

src/backends/test/ReshapeTestImpl.hpp

390

src/backends/test/PermuteTestImpl.hpp

391

src/backends/test/FullyConnectedTestImpl.hpp

392

src/backends/test/SplitterTestImpl.hpp

393

src/backends/test/NormTestImpl.hpp

394

src/backends/test/BatchNormTestImpl.hpp

395

src/backends/test/WorkloadTestUtils.hpp

396

src/backends/test/CreateWorkloadRef.cpp

397

src/backends/test/QuantizeHelper.hpp)

telsoa01

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

[diff] [blame]

398

399

if(ARMCOMPUTENEON)

400

list(APPEND unittest_sources

David Beck

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

[diff] [blame]

401

src/backends/test/ArmComputeNeon.cpp

402

src/backends/test/CreateWorkloadNeon.cpp

telsoa01

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

[diff] [blame]

403

src/armnn/test/CreateWorkloadClNeon.hpp

404

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]

409

src/backends/test/ArmComputeCl.cpp

410

src/backends/test/ClContextControlFixture.hpp

411

src/backends/test/CreateWorkloadCl.cpp

telsoa01

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

[diff] [blame]

412

src/armnn/test/CreateWorkloadClNeon.hpp

413

src/armnn/test/OpenClTimerTest.cpp

414

src/armnn/test/FP16SupportTest.cpp)

telsoa01

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

[diff] [blame]

415

endif()

416

417

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

418

list(APPEND unittest_sources

David Beck

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

[diff] [blame]

419

src/backends/test/MemCopyTests.cpp)

telsoa01

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

[diff] [blame]

420

endif()

421

surmeh01

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

[diff] [blame]

422

if(BUILD_TF_PARSER)

423

list(APPEND unittest_sources

424

src/armnnTfParser/test/Activations.cpp

425

src/armnnTfParser/test/Addition.cpp

426

src/armnnTfParser/test/BiasAdd.cpp

427

src/armnnTfParser/test/BroadcastForAdd.cpp

428

src/armnnTfParser/test/Convolution2d.cpp

429

src/armnnTfParser/test/Concat.cpp

430

src/armnnTfParser/test/ConcatOfConcats.cpp

431

src/armnnTfParser/test/DepthwiseConvolution2d.cpp

432

src/armnnTfParser/test/FusedBatchNorm.cpp

433

src/armnnTfParser/test/Identity.cpp

434

src/armnnTfParser/test/LocalResponseNormalization.cpp

telsoa01

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

[diff] [blame]

435

src/armnnTfParser/test/MaximumForLeakyRelu.cpp

surmeh01

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

[diff] [blame]

436

src/armnnTfParser/test/Multiplication.cpp

437

src/armnnTfParser/test/MultiOutput.cpp

438

src/armnnTfParser/test/PassThru.cpp

439

src/armnnTfParser/test/Pooling.cpp

saoste01

bbd4061

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

[diff] [blame]

440

src/armnnTfParser/test/RealDiv.cpp

surmeh01

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

[diff] [blame]

441

src/armnnTfParser/test/Reshape.cpp

442

src/armnnTfParser/test/ResizeBilinear.cpp

443

src/armnnTfParser/test/Shape.cpp

444

src/armnnTfParser/test/Softmax.cpp

445

src/armnnTfParser/test/TestDependencies.cpp

446

src/armnnTfParser/test/FullyConnected.cpp

447

src/armnnTfParser/test/Constant.cpp

448

src/armnnTfParser/test/TestMultiInputsOutputs.cpp

449

src/armnnTfParser/test/Squeeze.cpp)

450

endif()

451

telsoa01

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

[diff] [blame]

452

if(BUILD_TF_LITE_PARSER)

453

list(APPEND unittest_sources

454

src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp

455

src/armnnTfLiteParser/test/AvgPool2D.cpp

456

src/armnnTfLiteParser/test/Conv2D.cpp

457

src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp

458

src/armnnTfLiteParser/test/Softmax.cpp

459

src/armnnTfLiteParser/test/Squeeze.cpp

460

src/armnnTfLiteParser/test/LoadModel.cpp

461

src/armnnTfLiteParser/test/GetBuffer.cpp

462

src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp

463

src/armnnTfLiteParser/test/InputOutputTensorNames.cpp

464

src/armnnTfLiteParser/test/GetTensorIds.cpp

465

src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp

466

src/armnnTfLiteParser/test/GetInputsOutputs.cpp

Sadik Armagan

58f3919

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

[diff] [blame]

467

src/armnnTfLiteParser/test/Activations.cpp

Sadik

b94967b

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

[diff] [blame]

468

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]

472

if(BUILD_CAFFE_PARSER)

473

list(APPEND unittest_sources

474

src/armnnCaffeParser/test/TestAdd.cpp

475

src/armnnCaffeParser/test/TestConcat.cpp

telsoa01

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

[diff] [blame]

476

src/armnnCaffeParser/test/TestConvolution.cpp

telsoa01

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

[diff] [blame]

477

src/armnnCaffeParser/test/TestDropout.cpp

478

src/armnnCaffeParser/test/TestInputs.cpp

479

src/armnnCaffeParser/test/TestMul.cpp

480

src/armnnCaffeParser/test/TestPooling2d.cpp

481

src/armnnCaffeParser/test/TestInPlace.cpp

482

src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp

483

src/armnnCaffeParser/test/TestSplit.cpp

)

endif()

telsoa01

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

[diff] [blame]

487

if(BUILD_ONNX_PARSER)

488

list(APPEND unittest_sources

489

src/armnnOnnxParser/test/Constructor.cpp

490

src/armnnOnnxParser/test/CreateNetwork.cpp

491

src/armnnOnnxParser/test/ProtoxtFixture.cpp

492

src/armnnOnnxParser/test/Const.cpp

493

src/armnnOnnxParser/test/Pooling.cpp

494

src/armnnOnnxParser/test/Reshape.cpp

495

src/armnnOnnxParser/test/Relu.cpp

496

src/armnnOnnxParser/test/Conv2D.cpp

497

src/armnnOnnxParser/test/Addition.cpp

498

src/armnnOnnxParser/test/FullyConnected.cpp

499

src/armnnOnnxParser/test/GetInputsOutputs.cpp

500

src/armnnOnnxParser/test/BatchNorm.cpp

501

src/armnnOnnxParser/test/DepthConv.cpp

)

endif()

telsoa01

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

[diff] [blame]

505

add_executable_ex(UnitTests ${unittest_sources})

David Beck

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

[diff] [blame]

506

target_include_directories(UnitTests PRIVATE src)

telsoa01

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

[diff] [blame]

507

target_include_directories(UnitTests PRIVATE src/armnn)

508

target_include_directories(UnitTests PRIVATE src/armnnUtils)

509

telsoa01

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

[diff] [blame]

510

if(VALGRIND_FOUND)

511

if(HEAP_PROFILING OR LEAK_CHECKING)

512

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

513

else()

514

# Valgrind works with gperftools version number <= 2.4

515

target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")

516

endif()

telsoa01

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

[diff] [blame]

517

endif()

518

519

target_link_libraries(UnitTests armnn)

surmeh01

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

[diff] [blame]

520

target_link_libraries(UnitTests armnnUtils)

telsoa01

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

[diff] [blame]

521

target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})

telsoa01

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

[diff] [blame]

522

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]

523

surmeh01

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

[diff] [blame]

524

if(BUILD_TF_PARSER)

525

target_link_libraries(UnitTests armnnTfParser)

526

endif()

527

telsoa01

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

[diff] [blame]

528

if(BUILD_CAFFE_PARSER)

529

target_link_libraries(UnitTests armnnCaffeParser)

530

endif()

531

telsoa01

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

[diff] [blame]

532

if(BUILD_TF_LITE_PARSER)

533

target_link_libraries(UnitTests armnnTfLiteParser)

534

endif()

535

536

if(BUILD_ONNX_PARSER)

537

target_link_libraries(UnitTests armnnOnnxParser)

538

endif()

539

telsoa01