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

David Beck

e97c6e0

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

[diff] [blame]

153

include/armnn/ILayerSupport.hpp

telsoa01

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

[diff] [blame]

154

include/armnn/INetwork.hpp

David Beck

5eec11d

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

[diff] [blame]

155

include/armnn/Optional.hpp

telsoa01

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

[diff] [blame]

156

include/armnn/Tensor.hpp

157

include/armnn/TensorFwd.hpp

158

include/armnn/Types.hpp

159

include/armnn/TypesUtils.hpp

160

include/armnn/Utils.hpp

161

include/armnn/LayerSupport.hpp

162

include/armnn/Version.hpp

surmeh01

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

[diff] [blame]

163

src/armnn/layers/LayerCloneBase.hpp

164

src/armnn/layers/LayerWithParameters.hpp

165

src/armnn/layers/ActivationLayer.hpp

166

src/armnn/layers/ActivationLayer.cpp

167

src/armnn/layers/AdditionLayer.hpp

168

src/armnn/layers/AdditionLayer.cpp

Les Bell

033626d

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

[diff] [blame]

169

src/armnn/layers/ArithmeticBaseLayer.hpp

170

src/armnn/layers/ArithmeticBaseLayer.cpp

surmeh01

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

[diff] [blame]

171

src/armnn/layers/BatchNormalizationLayer.hpp

172

src/armnn/layers/BatchNormalizationLayer.cpp

173

src/armnn/layers/ConstantLayer.hpp

174

src/armnn/layers/ConstantLayer.cpp

175

src/armnn/layers/Convolution2dLayer.hpp

176

src/armnn/layers/Convolution2dLayer.cpp

telsoa01

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

[diff] [blame]

177

src/armnn/layers/ConvertFp16ToFp32Layer.hpp

178

src/armnn/layers/ConvertFp16ToFp32Layer.cpp

179

src/armnn/layers/ConvertFp32ToFp16Layer.hpp

180

src/armnn/layers/ConvertFp32ToFp16Layer.cpp

surmeh01

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

[diff] [blame]

181

src/armnn/layers/DepthwiseConvolution2dLayer.hpp

182

src/armnn/layers/DepthwiseConvolution2dLayer.cpp

183

src/armnn/layers/FakeQuantizationLayer.hpp

184

src/armnn/layers/FakeQuantizationLayer.cpp

185

src/armnn/layers/FloorLayer.hpp

186

src/armnn/layers/FloorLayer.cpp

187

src/armnn/layers/FullyConnectedLayer.hpp

188

src/armnn/layers/FullyConnectedLayer.cpp

189

src/armnn/layers/InputLayer.hpp

190

src/armnn/layers/InputLayer.cpp

191

src/armnn/layers/L2NormalizationLayer.hpp

192

src/armnn/layers/L2NormalizationLayer.cpp

telsoa01

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

[diff] [blame]

193

src/armnn/layers/LstmLayer.cpp

194

src/armnn/layers/LstmLayer.hpp

narpra01

32b9046

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

[diff] [blame]

195

src/armnn/layers/MeanLayer.hpp

196

src/armnn/layers/MeanLayer.cpp

surmeh01

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

[diff] [blame]

197

src/armnn/layers/MemCopyLayer.hpp

198

src/armnn/layers/MemCopyLayer.cpp

199

src/armnn/layers/MergerLayer.hpp

200

src/armnn/layers/MergerLayer.cpp

201

src/armnn/layers/MultiplicationLayer.hpp

202

src/armnn/layers/MultiplicationLayer.cpp

203

src/armnn/layers/NormalizationLayer.hpp

204

src/armnn/layers/NormalizationLayer.cpp

205

src/armnn/layers/OutputLayer.hpp

206

src/armnn/layers/OutputLayer.cpp

Mohamed Nour Abouelseoud

5662c20

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

[diff] [blame]

207

src/armnn/layers/PadLayer.hpp

208

src/armnn/layers/PadLayer.cpp

surmeh01

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

[diff] [blame]

209

src/armnn/layers/PermuteLayer.hpp

210

src/armnn/layers/PermuteLayer.cpp

211

src/armnn/layers/Pooling2dLayer.hpp

212

src/armnn/layers/Pooling2dLayer.cpp

Francis Murtagh

e7a86a4

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

[diff] [blame]

213

src/armnn/layers/DivisionLayer.cpp

214

src/armnn/layers/DivisionLayer.hpp

surmeh01

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

[diff] [blame]

215

src/armnn/layers/ReshapeLayer.hpp

216

src/armnn/layers/ReshapeLayer.cpp

217

src/armnn/layers/ResizeBilinearLayer.hpp

218

src/armnn/layers/ResizeBilinearLayer.cpp

219

src/armnn/layers/SoftmaxLayer.hpp

220

src/armnn/layers/SoftmaxLayer.cpp

221

src/armnn/layers/SplitterLayer.hpp

222

src/armnn/layers/SplitterLayer.cpp

David Beck

c2044fe

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

[diff] [blame]

223

src/armnn/layers/SubtractionLayer.cpp

224

src/armnn/layers/SubtractionLayer.hpp

telsoa01

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

[diff] [blame]

225

src/armnn/InternalTypes.hpp

226

src/armnn/InternalTypes.cpp

telsoa01

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

[diff] [blame]

227

src/armnn/JsonPrinter.hpp

228

src/armnn/JsonPrinter.cpp

telsoa01

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

[diff] [blame]

229

src/armnn/LayerFwd.hpp

230

src/armnn/Layer.hpp

231

src/armnn/Layer.cpp

232

src/armnn/LayersFwd.hpp

telsoa01

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

[diff] [blame]

233

src/armnn/Runtime.hpp

234

src/armnn/Runtime.cpp

surmeh01

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

[diff] [blame]

235

src/armnn/SerializeLayerParameters.cpp

236

src/armnn/SerializeLayerParameters.hpp

telsoa01

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

[diff] [blame]

237

src/armnn/Descriptors.cpp

telsoa01

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

[diff] [blame]

238

src/armnn/DeviceSpec.hpp

telsoa01

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

[diff] [blame]

239

src/armnn/LoadedNetwork.hpp

240

src/armnn/LoadedNetwork.cpp

241

src/armnn/Exceptions.cpp

242

src/armnn/Graph.hpp

243

src/armnn/Graph.cpp

244

src/armnn/Network.hpp

245

src/armnn/Network.cpp

telsoa01

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

[diff] [blame]

246

src/armnn/NetworkUtils.hpp

telsoa01

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

[diff] [blame]

247

src/armnn/ProfilingEvent.cpp

248

src/armnn/ProfilingEvent.hpp

telsoa01

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

[diff] [blame]

249

src/armnn/Profiling.cpp

telsoa01

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

[diff] [blame]

250

src/armnn/Instrument.hpp

251

src/armnn/WallClockTimer.hpp

252

src/armnn/WallClockTimer.cpp

telsoa01

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

[diff] [blame]

253

src/armnn/Tensor.cpp

254

src/armnn/Utils.cpp

255

src/armnn/LayerSupport.cpp

256

src/armnn/LayerSupportCommon.hpp

257

src/armnn/optimizations/All.hpp

telsoa01

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

[diff] [blame]

258

src/armnn/optimizations/ConvertConstants.hpp

telsoa01

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

[diff] [blame]

259

src/armnn/optimizations/MovePermuteUp.hpp

260

src/armnn/optimizations/Optimization.hpp

261

src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp

262

src/armnn/optimizations/OptimizeInversePermutes.hpp

263

src/armnn/optimizations/PermuteAsReshape.hpp

264

src/armnn/optimizations/SquashEqualSiblings.hpp

telsoa01

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

[diff] [blame]

265

src/armnn/optimizations/OptimizeInverseConversions.hpp

266

src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp

telsoa01

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

[diff] [blame]

267

src/armnn/Optimizer.hpp

268

src/armnn/Optimizer.cpp

telsoa01

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

[diff] [blame]

269

third-party/half/half.hpp

270

src/armnn/IGraphObservable.hpp

271

src/armnn/Observable.hpp

272

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

277

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

278

src/armnn/NeonInterceptorScheduler.hpp

279

src/armnn/NeonInterceptorScheduler.cpp

280

src/armnn/NeonTimer.hpp

281

src/armnn/NeonTimer.cpp)

telsoa01

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

[diff] [blame]

282

endif()

283

if(ARMCOMPUTECL)

284

# Additionally include source files for ARM Compute OpenCL backend

285

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

286

src/armnn/OpenClTimer.cpp

287

src/armnn/OpenClTimer.hpp)

telsoa01

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

[diff] [blame]

288

endif()

289

# Files shared by all ARM Compute backends

290

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

291

list(APPEND armnn_sources

telsoa01

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

[diff] [blame]

292

src/armnn/memory/IMemoryPool.hpp

293

src/armnn/memory/BlobMemoryPool.cpp

294

src/armnn/memory/BlobMemoryPool.hpp

295

src/armnn/memory/BlobLifetimeManager.cpp

296

src/armnn/memory/BlobLifetimeManager.hpp

297

src/armnn/memory/PoolManager.cpp

298

src/armnn/memory/PoolManager.hpp

299

src/armnn/memory/BaseMemoryManager.hpp

300

src/armnn/memory/BaseMemoryManager.cpp

301

src/armnn/memory/OffsetMemoryPool.cpp

302

src/armnn/memory/OffsetMemoryPool.hpp

303

src/armnn/memory/OffsetLifetimeManager.cpp

304

src/armnn/memory/OffsetLifetimeManager.hpp)

telsoa01

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

[diff] [blame]

305

endif()

306

307

# Files used for Streamline-based profiling backend

308

if(PROFILING_BACKEND_STREAMLINE)

309

list(APPEND armnn_sources

310

${GATOR_ROOT}/annotate/streamline_annotate.h

311

${GATOR_ROOT}/annotate/streamline_annotate.c)

312

endif()

313

314

add_library_ex(armnn SHARED ${armnn_sources})

David Beck

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

[diff] [blame]

315

316

# the backends under src/backends extend the list of

317

# static libs armnn to link against

318

list(APPEND armnnLibraries armnnUtils)

319

include(src/backends/backends.cmake)

320

321

target_include_directories(armnn PRIVATE src)

telsoa01

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

[diff] [blame]

322

target_include_directories(armnn PRIVATE src/armnn)

323

target_include_directories(armnn PRIVATE src/armnnUtils)

David Beck

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

[diff] [blame]

324

foreach(lib ${armnnLibraries})

325

target_link_libraries(armnn ${lib})

326

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

327

endforeach()

telsoa01

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

[diff] [blame]

328

target_link_libraries(armnn ${CMAKE_DL_LIBS})

329

330

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

331

if(BUILD_CAFFE_PARSER)

332

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

333

endif()

telsoa01

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

[diff] [blame]

334

if(BUILD_ONNX_PARSER)

335

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

336

endif()

surmeh01

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

[diff] [blame]

337

if(BUILD_TF_PARSER)

338

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

339

endif()

telsoa01

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

[diff] [blame]

340

if(BUILD_TF_LITE_PARSER)

341

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

342

endif()

telsoa01

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

[diff] [blame]

343

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

344

345

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

346

347

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

348

target_link_libraries(armnn ${ARMCOMPUTE_LIBRARIES})

349

endif()

350

if(ARMCOMPUTECL)

351

target_link_libraries(armnn ${OPENCL_LIBRARIES})

352

endif()

353

354

if(PROFILING_BACKEND_STREAMLINE)

355

target_link_libraries(armnn pthread)

356

endif()

357

telsoa01

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

[diff] [blame]

358

if(BUILD_UNIT_TESTS)

359

set(unittest_sources)

360

list(APPEND unittest_sources

361

src/armnn/test/UnitTests.cpp

362

src/armnn/test/UnitTests.hpp

363

src/armnn/test/EndToEndTest.cpp

364

src/armnn/test/UtilsTests.cpp

telsoa01

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

[diff] [blame]

365

src/armnn/test/JsonPrinterTests.cpp

telsoa01

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

[diff] [blame]

366

src/armnn/test/GraphTests.cpp

surmeh01

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

[diff] [blame]

367

src/armnn/test/OptimizerTests.cpp

telsoa01

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

[diff] [blame]

368

src/armnn/test/ProfilerTests.cpp

telsoa01

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

[diff] [blame]

369

src/armnn/test/RuntimeTests.cpp

370

src/armnn/test/CreateWorkload.hpp

371

src/armnn/test/TensorTest.cpp

372

src/armnn/test/TensorHelpers.hpp

telsoa01

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

[diff] [blame]

373

src/armnn/test/CsvReaderTest.cpp

374

src/armnn/test/NetworkTests.cpp

375

src/armnn/test/FloatingPointConverterTest.cpp

376

src/armnn/test/ProfilingEventTest.cpp

telsoa01

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

[diff] [blame]

377

src/armnn/test/GraphUtils.hpp

telsoa01

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

[diff] [blame]

378

src/armnn/test/InstrumentTests.cpp

379

src/armnn/test/ObservableTest.cpp

David Beck

5eec11d

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

[diff] [blame]

380

src/armnn/test/OptionalTest.cpp

David Beck

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

[diff] [blame]

381

src/backends/test/IsLayerSupportedTest.cpp

382

src/backends/test/IsLayerSupportedTestImpl.hpp

383

src/backends/test/Reference.cpp

384

src/backends/test/WorkloadDataValidation.cpp

385

src/backends/test/TensorCopyUtils.hpp

386

src/backends/test/TensorCopyUtils.cpp

387

src/backends/test/LayerTests.hpp

388

src/backends/test/LayerTests.cpp

389

src/backends/test/LayerReleaseConstantDataTest.cpp

390

src/backends/test/Conv2dTestImpl.hpp

391

src/backends/test/ActivationTestImpl.hpp

392

src/backends/test/ActivationFixture.hpp

393

src/backends/test/Pooling2dTestImpl.hpp

394

src/backends/test/ReshapeTestImpl.hpp

395

src/backends/test/PermuteTestImpl.hpp

396

src/backends/test/FullyConnectedTestImpl.hpp

397

src/backends/test/SplitterTestImpl.hpp

398

src/backends/test/NormTestImpl.hpp

399

src/backends/test/BatchNormTestImpl.hpp

400

src/backends/test/WorkloadTestUtils.hpp

401

src/backends/test/CreateWorkloadRef.cpp

402

src/backends/test/QuantizeHelper.hpp)

telsoa01

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

[diff] [blame]

403

404

if(ARMCOMPUTENEON)

405

list(APPEND unittest_sources

David Beck

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

[diff] [blame]

406

src/backends/test/ArmComputeNeon.cpp

407

src/backends/test/CreateWorkloadNeon.cpp

telsoa01

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

[diff] [blame]

408

src/armnn/test/CreateWorkloadClNeon.hpp

409

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]

414

src/backends/test/ArmComputeCl.cpp

415

src/backends/test/ClContextControlFixture.hpp

416

src/backends/test/CreateWorkloadCl.cpp

telsoa01

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

[diff] [blame]

417

src/armnn/test/CreateWorkloadClNeon.hpp

418

src/armnn/test/OpenClTimerTest.cpp

419

src/armnn/test/FP16SupportTest.cpp)

telsoa01

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

[diff] [blame]

420

endif()

421

422

if(ARMCOMPUTENEON OR ARMCOMPUTECL)

423

list(APPEND unittest_sources

David Beck

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

[diff] [blame]

424

src/backends/test/MemCopyTests.cpp)

telsoa01

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

[diff] [blame]

425

endif()

426

surmeh01

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

[diff] [blame]

427

if(BUILD_TF_PARSER)

428

list(APPEND unittest_sources

429

src/armnnTfParser/test/Activations.cpp

430

src/armnnTfParser/test/Addition.cpp

431

src/armnnTfParser/test/BiasAdd.cpp

432

src/armnnTfParser/test/BroadcastForAdd.cpp

433

src/armnnTfParser/test/Convolution2d.cpp

434

src/armnnTfParser/test/Concat.cpp

435

src/armnnTfParser/test/ConcatOfConcats.cpp

436

src/armnnTfParser/test/DepthwiseConvolution2d.cpp

437

src/armnnTfParser/test/FusedBatchNorm.cpp

438

src/armnnTfParser/test/Identity.cpp

439

src/armnnTfParser/test/LocalResponseNormalization.cpp

telsoa01

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

[diff] [blame]

440

src/armnnTfParser/test/MaximumForLeakyRelu.cpp

surmeh01

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

[diff] [blame]

441

src/armnnTfParser/test/Multiplication.cpp

442

src/armnnTfParser/test/MultiOutput.cpp

443

src/armnnTfParser/test/PassThru.cpp

444

src/armnnTfParser/test/Pooling.cpp

saoste01

bbd4061

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

[diff] [blame]

445

src/armnnTfParser/test/RealDiv.cpp

surmeh01

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

[diff] [blame]

446

src/armnnTfParser/test/Reshape.cpp

447

src/armnnTfParser/test/ResizeBilinear.cpp

448

src/armnnTfParser/test/Shape.cpp

449

src/armnnTfParser/test/Softmax.cpp

450

src/armnnTfParser/test/TestDependencies.cpp

451

src/armnnTfParser/test/FullyConnected.cpp

452

src/armnnTfParser/test/Constant.cpp

453

src/armnnTfParser/test/TestMultiInputsOutputs.cpp

454

src/armnnTfParser/test/Squeeze.cpp)

455

endif()

456

telsoa01

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

[diff] [blame]

457

if(BUILD_TF_LITE_PARSER)

458

list(APPEND unittest_sources

459

src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp

Sadik Armagan

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

[diff] [blame]

460

src/armnnTfLiteParser/test/Activations.cpp

telsoa01

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

[diff] [blame]

461

src/armnnTfLiteParser/test/AvgPool2D.cpp

Sadik Armagan

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

[diff] [blame]

462

src/armnnTfLiteParser/test/Concatenation.cpp

telsoa01

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

[diff] [blame]

463

src/armnnTfLiteParser/test/Conv2D.cpp

464

src/armnnTfLiteParser/test/DepthwiseConvolution2D.cpp

Sadik Armagan

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

[diff] [blame]

465

src/armnnTfLiteParser/test/Reshape.cpp

telsoa01

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

[diff] [blame]

466

src/armnnTfLiteParser/test/Softmax.cpp

467

src/armnnTfLiteParser/test/Squeeze.cpp

468

src/armnnTfLiteParser/test/LoadModel.cpp

469

src/armnnTfLiteParser/test/GetBuffer.cpp

470

src/armnnTfLiteParser/test/OutputShapeOfSqueeze.cpp

471

src/armnnTfLiteParser/test/InputOutputTensorNames.cpp

472

src/armnnTfLiteParser/test/GetTensorIds.cpp

473

src/armnnTfLiteParser/test/GetSubgraphInputsOutputs.cpp

474

src/armnnTfLiteParser/test/GetInputsOutputs.cpp

)

endif()

telsoa01

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

[diff] [blame]

478

if(BUILD_CAFFE_PARSER)

479

list(APPEND unittest_sources

480

src/armnnCaffeParser/test/TestAdd.cpp

481

src/armnnCaffeParser/test/TestConcat.cpp

telsoa01

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

[diff] [blame]

482

src/armnnCaffeParser/test/TestConvolution.cpp

telsoa01

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

[diff] [blame]

483

src/armnnCaffeParser/test/TestDropout.cpp

484

src/armnnCaffeParser/test/TestInputs.cpp

485

src/armnnCaffeParser/test/TestMul.cpp

486

src/armnnCaffeParser/test/TestPooling2d.cpp

487

src/armnnCaffeParser/test/TestInPlace.cpp

488

src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp

489

src/armnnCaffeParser/test/TestSplit.cpp

)

endif()

telsoa01

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

[diff] [blame]

493

if(BUILD_ONNX_PARSER)

494

list(APPEND unittest_sources

495

src/armnnOnnxParser/test/Constructor.cpp

496

src/armnnOnnxParser/test/CreateNetwork.cpp

497

src/armnnOnnxParser/test/ProtoxtFixture.cpp

498

src/armnnOnnxParser/test/Const.cpp

499

src/armnnOnnxParser/test/Pooling.cpp

500

src/armnnOnnxParser/test/Reshape.cpp

501

src/armnnOnnxParser/test/Relu.cpp

502

src/armnnOnnxParser/test/Conv2D.cpp

503

src/armnnOnnxParser/test/Addition.cpp

504

src/armnnOnnxParser/test/FullyConnected.cpp

505

src/armnnOnnxParser/test/GetInputsOutputs.cpp

506

src/armnnOnnxParser/test/BatchNorm.cpp

507

src/armnnOnnxParser/test/DepthConv.cpp

)

endif()

telsoa01

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

[diff] [blame]

511

add_executable_ex(UnitTests ${unittest_sources})

David Beck

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

[diff] [blame]

512

target_include_directories(UnitTests PRIVATE src)

telsoa01

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

[diff] [blame]

513

target_include_directories(UnitTests PRIVATE src/armnn)

514

target_include_directories(UnitTests PRIVATE src/armnnUtils)

515

telsoa01

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

[diff] [blame]

516

if(VALGRIND_FOUND)

517

if(HEAP_PROFILING OR LEAK_CHECKING)

518

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

519

else()

520

# Valgrind works with gperftools version number <= 2.4

521

target_compile_definitions(UnitTests PRIVATE "WITH_VALGRIND=1")

522

endif()

telsoa01

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

[diff] [blame]

523

endif()

524

525

target_link_libraries(UnitTests armnn)

surmeh01

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

[diff] [blame]

526

target_link_libraries(UnitTests armnnUtils)

telsoa01

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

[diff] [blame]

527

target_link_libraries(UnitTests ${CMAKE_THREAD_LIBS_INIT})

telsoa01

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

[diff] [blame]

528

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]

529

surmeh01

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

[diff] [blame]

530

if(BUILD_TF_PARSER)

531

target_link_libraries(UnitTests armnnTfParser)

532

endif()

533

telsoa01

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

[diff] [blame]

534

if(BUILD_CAFFE_PARSER)

535

target_link_libraries(UnitTests armnnCaffeParser)

536

endif()

537

telsoa01

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

[diff] [blame]

538

if(BUILD_TF_LITE_PARSER)

539

target_link_libraries(UnitTests armnnTfLiteParser)

540

endif()

541

542

if(BUILD_ONNX_PARSER)

543

target_link_libraries(UnitTests armnnOnnxParser)

544

endif()

545

telsoa01