blob: 8757723c107afd847f410e4b15177e6676dd23ef [file] [log] [blame]
Pavel Macenauer59e057f2020-04-15 14:17:26 +00001#
Jim Flynnbbfe6032020-07-20 16:57:44 +01002# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
Pavel Macenauer59e057f2020-04-15 14:17:26 +00003# Copyright 2020 NXP
4# SPDX-License-Identifier: MIT
5#
telsoa01c577f2c2018-08-31 09:22:23 +01006option(BUILD_ONNX_PARSER "Build Onnx parser" OFF)
telsoa014fcda012018-03-09 14:13:49 +00007option(BUILD_UNIT_TESTS "Build unit tests" ON)
8option(BUILD_TESTS "Build test applications" OFF)
9option(BUILD_FOR_COVERAGE "Use no optimization and output .gcno and .gcda files" OFF)
10option(ARMCOMPUTENEON "Build with ARM Compute NEON support" OFF)
11option(ARMCOMPUTECL "Build with ARM Compute OpenCL support" OFF)
Matteo Martincighf88663c2019-08-28 16:38:53 +010012option(ARMNNREF "Build with ArmNN reference support" ON)
telsoa014fcda012018-03-09 14:13:49 +000013option(PROFILING_BACKEND_STREAMLINE "Forward the armNN profiling events to DS-5/Streamline as annotations" OFF)
telsoa01c577f2c2018-08-31 09:22:23 +010014# options used for heap profiling and leak checking
surmeh013537c2c2018-05-18 16:31:43 +010015option(HEAP_PROFILING "Build with heap profiling enabled" OFF)
telsoa01c577f2c2018-08-31 09:22:23 +010016option(LEAK_CHECKING "Build with leak checking enabled" OFF)
surmeh013537c2c2018-05-18 16:31:43 +010017option(GPERFTOOLS_ROOT "Location where the gperftools 'include' and 'lib' folders to be found" Off)
telsoa01c577f2c2018-08-31 09:22:23 +010018# options used for tensorflow lite support
19option(BUILD_TF_LITE_PARSER "Build Tensorflow Lite parser" OFF)
Nattapat Chaimanowong949f1252019-01-31 15:36:39 +000020option(BUILD_ARMNN_SERIALIZER "Build Armnn Serializer" OFF)
Éanna Ó Catháina563b922019-05-09 11:34:06 +010021option(BUILD_ACCURACY_TOOL "Build Accuracy Tool" OFF)
Nattapat Chaimanowong949f1252019-01-31 15:36:39 +000022option(FLATC_DIR "Path to Flatbuffers compiler" OFF)
telsoa01c577f2c2018-08-31 09:22:23 +010023option(TF_LITE_GENERATED_PATH "Tensorflow lite generated C++ schema location" OFF)
24option(FLATBUFFERS_ROOT "Location where the flatbuffers 'include' and 'lib' folders to be found" Off)
Matteo Martincighe7d44982019-08-05 12:16:47 +010025option(DYNAMIC_BACKEND_PATHS "Colon seperated list of paths where to load the dynamic backends from" "")
Colm Donelan0dbe00a2020-06-03 08:00:28 +010026option(SAMPLE_DYNAMIC_BACKEND "Include the sample dynamic backend and its tests in the build" OFF)
Colm Donelana21620d2019-10-11 13:09:49 +010027option(BUILD_GATORD_MOCK "Build the Gatord simulator for external profiling testing." ON)
Sadik Armagan6f86b692020-03-09 11:04:32 +000028option(BUILD_TIMELINE_DECODER "Build the Timeline Decoder for external profiling." ON)
Matthew Benthamb4e67202019-10-31 09:55:01 +000029option(SHARED_BOOST "Use dynamic linking for boost libraries" OFF)
Finn Williams2ed809c2020-04-20 21:21:07 +010030option(BUILD_BASE_PIPE_SERVER "Build the server to handle external profiling pipe traffic" ON)
Pavel Macenauer59e057f2020-04-15 14:17:26 +000031option(BUILD_PYTHON_WHL "Build Python wheel package" OFF)
32option(BUILD_PYTHON_SRC "Build Python source package" OFF)
Jim Flynnbbfe6032020-07-20 16:57:44 +010033option(BUILD_STATIC_PIPE_LIBS "Build Static PIPE libraries" OFF)
34option(BUILD_PIPE_ONLY "Build the PIPE libraries only" OFF)
Sadik Armagan5d03e312020-11-17 16:43:56 +000035option(BUILD_ARMNN_TFLITE_DELEGATE "Build the Arm NN TfLite delegate" OFF)
telsoa014fcda012018-03-09 14:13:49 +000036
37include(SelectLibraryConfigurations)
38
39set(COMPILER_IS_GNU_LIKE 0)
Jim Flynn82262f52020-07-16 11:23:45 +010040if(${CMAKE_CXX_COMPILER_ID} STREQUAL GNU OR
41 ${CMAKE_CXX_COMPILER_ID} STREQUAL Clang OR
42 ${CMAKE_CXX_COMPILER_ID} STREQUAL AppleClang)
telsoa014fcda012018-03-09 14:13:49 +000043 set(COMPILER_IS_GNU_LIKE 1)
44endif()
45
46# Enable CCache if available and not disabled
47option(USE_CCACHE "USE_CCACHE" ON)
48find_program(CCACHE_FOUND ccache)
49if(CCACHE_FOUND AND USE_CCACHE)
50 get_property(rule_launch_compile DIRECTORY PROPERTY RULE_LAUNCH_COMPILE)
51 set_property(DIRECTORY PROPERTY RULE_LAUNCH_COMPILE "CCACHE_CPP2=yes ${rule_launch_compile} ccache")
52endif()
53
54# Enable distcc if available and not disabled
55option(USE_DISTCC "USE_DISTCC" OFF)
56find_program(DISTCC_FOUND distcc)
57if(DISTCC_FOUND AND USE_DISTCC)
58 get_property(rule_launch_compile DIRECTORY PROPERTY RULE_LAUNCH_COMPILE)
59 set_property(DIRECTORY PROPERTY RULE_LAUNCH_COMPILE "${rule_launch_compile} distcc")
60endif()
61
62# Set to release configuration by default
63if(NOT CMAKE_BUILD_TYPE)
64 set(CMAKE_BUILD_TYPE "Release")
65endif()
66
67# Compiler flags that are always set
68set(CMAKE_POSITION_INDEPENDENT_CODE ON)
69if(COMPILER_IS_GNU_LIKE)
Derek Lambertiba25aee2019-12-10 22:20:54 +000070 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14 -Wall -Wextra -Werror -Wold-style-cast -Wno-missing-braces -Wconversion -Wsign-conversion")
telsoa014fcda012018-03-09 14:13:49 +000071elseif(${CMAKE_CXX_COMPILER_ID} STREQUAL MSVC)
Rob Hughesfc6bf052019-12-16 17:10:51 +000072 # Disable C4996 (use of deprecated identifier) due to https://developercommunity.visualstudio.com/content/problem/252574/deprecated-compilation-warning-for-virtual-overrid.html
73 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /EHsc /MP /wd4996")
Rob Hughesbc873d22020-05-20 13:11:37 +010074 add_definitions(-DNO_STRICT=1)
telsoa014fcda012018-03-09 14:13:49 +000075endif()
76if("${CMAKE_SYSTEM_NAME}" STREQUAL Android)
Keith Davisb8942bf2021-05-31 11:22:35 +010077 # -lz is necessary for when building with ACL set with compressed kernels
78 set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -llog -lz")
79 set(CMAKE_SHARED_LINKER_FLAGS "${CMAKE_SHARED_LINKER_FLAGS} -llog -lz")
telsoa014fcda012018-03-09 14:13:49 +000080endif()
81
82# Compiler flags for Release builds
Matthew Benthame30054f2019-06-24 13:10:54 +010083set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} -DNDEBUG")
telsoa014fcda012018-03-09 14:13:49 +000084if(COMPILER_IS_GNU_LIKE)
85 set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} -O3")
86elseif(${CMAKE_CXX_COMPILER_ID} STREQUAL MSVC)
87 set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /MD /O2")
88endif()
89
90# Compiler flags for Debug builds
91if(COMPILER_IS_GNU_LIKE)
Matthew Benthame30054f2019-06-24 13:10:54 +010092 set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -g -O0")
telsoa014fcda012018-03-09 14:13:49 +000093elseif(${CMAKE_CXX_COMPILER_ID} STREQUAL MSVC)
Matthew Benthame30054f2019-06-24 13:10:54 +010094 set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /MDd /ZI /Od")
telsoa014fcda012018-03-09 14:13:49 +000095 # Disable SAFESEH which is necessary for Edit and Continue to work
96 set(CMAKE_EXE_LINKER_FLAGS_DEBUG "${CMAKE_EXE_LINKER_FLAGS_DEBUG} /SAFESEH:NO")
97 set(CMAKE_SHARED_LINKER_FLAGS_DEBUG "${CMAKE_EXE_LINKER_FLAGS_DEBUG} /SAFESEH:NO")
98endif()
99
100# Modify RelWithDebInfo so that NDEBUG isn't defined.
101# This enables asserts.
102if (COMPILER_IS_GNU_LIKE)
103 string(REPLACE "-DNDEBUG" "" CMAKE_CXX_FLAGS_RELWITHDEBINFO "${CMAKE_CXX_FLAGS_RELWITHDEBINFO}")
104elseif (${CMAKE_CXX_COMPILER_ID} STREQUAL MSVC)
105 string(REPLACE "/DNDEBUG" "" CMAKE_CXX_FLAGS_RELWITHDEBINFO "${CMAKE_CXX_FLAGS_RELWITHDEBINFO}")
106endif()
107
108# Compiler flags for code coverage measurements
109if(BUILD_FOR_COVERAGE)
110 if(NOT CMAKE_BUILD_TYPE EQUAL "Debug")
111 message(WARNING "BUILD_FOR_COVERAGE set so forcing to Debug build")
112 set(CMAKE_BUILD_TYPE "Debug")
113 endif()
114
115 set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} --coverage")
116 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} --coverage")
117endif()
118
119if(BUILD_FOR_COVERAGE AND NOT BUILD_UNIT_TESTS)
120 message(WARNING "BUILD_FOR_COVERAGE set but not BUILD_UNIT_TESTS, so code coverage will not be able to run")
121endif()
122
123set(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake/modules ${CMAKE_MODULE_PATH})
124
Derek Lambertieac4adb2020-08-25 13:05:59 +0100125include(CMakeFindDependencyMacro)
126
Jim Flynnbbfe6032020-07-20 16:57:44 +0100127if (NOT BUILD_PIPE_ONLY)
128 # Boost
129 message(STATUS "Finding Boost")
130 if(SHARED_BOOST)
Matthew Benthamb4e67202019-10-31 09:55:01 +0000131 add_definitions(-DBOOST_ALL_DYN_LINK)
132 set(Boost_USE_STATIC_LIBS OFF)
Jim Flynnbbfe6032020-07-20 16:57:44 +0100133 else()
Matthew Benthamb4e67202019-10-31 09:55:01 +0000134 set(Boost_USE_STATIC_LIBS ON)
Jim Flynnbbfe6032020-07-20 16:57:44 +0100135 endif()
Matthew Benthame688d612020-11-13 12:04:01 +0000136 if (BUILD_UNIT_TESTS)
137 add_definitions("-DBOOST_ALL_NO_LIB") # Turn off auto-linking as we specify the libs manually
138 find_package(Boost 1.59 REQUIRED COMPONENTS unit_test_framework)
139 include_directories(SYSTEM "${Boost_INCLUDE_DIRS}")
140 link_directories(${Boost_LIBRARY_DIRS})
141 endif()
Matthew Benthamb4e67202019-10-31 09:55:01 +0000142endif()
telsoa014fcda012018-03-09 14:13:49 +0000143
Jim Flynnbbfe6032020-07-20 16:57:44 +0100144if (NOT BUILD_PIPE_ONLY)
145 # cxxopts (Alternative to boost::program_options)
Vincent ABRIOU54940192020-09-03 13:02:43 +0200146 find_path(CXXOPTS_INCLUDE cxxopts/cxxopts.hpp PATHS third-party NO_CMAKE_FIND_ROOT_PATH)
Jim Flynnbbfe6032020-07-20 16:57:44 +0100147 include_directories(SYSTEM "${CXXOPTS_INCLUDE}")
148endif()
Jan Eilers7e989832020-06-19 11:47:21 +0100149
Jim Flynnbbfe6032020-07-20 16:57:44 +0100150if (NOT BUILD_PIPE_ONLY)
151 # ghc (Alternative to boost::filesystem)
Vincent ABRIOU54940192020-09-03 13:02:43 +0200152 find_path(GHC_INCLUDE ghc/filesystem.hpp PATHS third-party NO_CMAKE_FIND_ROOT_PATH)
Jim Flynnbbfe6032020-07-20 16:57:44 +0100153 include_directories(SYSTEM "${GHC_INCLUDE}")
154endif()
Jan Eilers307fd342020-06-23 14:16:04 +0100155
telsoa014fcda012018-03-09 14:13:49 +0000156# pthread
Derek Lambertieac4adb2020-08-25 13:05:59 +0100157find_dependency(Threads)
telsoa014fcda012018-03-09 14:13:49 +0000158
159# Favour the protobuf passed on command line
Nikhil Raj5d955cf2021-04-19 16:59:48 +0100160if(BUILD_ONNX_PARSER)
telsoa014fcda012018-03-09 14:13:49 +0000161 find_library(PROTOBUF_LIBRARY_DEBUG NAMES "protobufd"
162 PATHS ${PROTOBUF_ROOT}/lib
163 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
164 find_library(PROTOBUF_LIBRARY_DEBUG NAMES "protobufd")
165
166 find_library(PROTOBUF_LIBRARY_RELEASE NAMES "protobuf"
167 PATHS ${PROTOBUF_ROOT}/lib
168 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
169 find_library(PROTOBUF_LIBRARY_RELEASE NAMES "protobuf")
170
171 select_library_configurations(PROTOBUF)
172
173 find_path(PROTOBUF_INCLUDE_DIRS "google/protobuf/message.h"
174 PATHS ${PROTOBUF_ROOT}/include
175 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
176 find_path(PROTOBUF_INCLUDE_DIRS "google/protobuf/message.h")
177
178 include_directories(SYSTEM "${PROTOBUF_INCLUDE_DIRS}")
179 add_definitions(-DPROTOBUF_USE_DLLS)
telsoa014fcda012018-03-09 14:13:49 +0000180
telsoa01c577f2c2018-08-31 09:22:23 +0100181 add_definitions(-DARMNN_ONNX_PARSER)
182
183 find_path(ONNX_GENERATED_SOURCES "onnx/onnx.pb.cc")
184
185 # C++ headers generated for onnx protobufs
186 include_directories(SYSTEM "${ONNX_GENERATED_SOURCES}")
187endif()
188
Sadik Armagan5d03e312020-11-17 16:43:56 +0000189if(BUILD_ARMNN_TFLITE_DELEGATE)
190 add_definitions(-DARMNN_TFLITE_DELEGATE)
191endif()
Nattapat Chaimanowong949f1252019-01-31 15:36:39 +0000192# Flatbuffers support for TF Lite and Armnn Serializer
193if(BUILD_TF_LITE_PARSER OR BUILD_ARMNN_SERIALIZER)
telsoa01c577f2c2018-08-31 09:22:23 +0100194 # verify we have a valid flatbuffers include path
195 find_path(FLATBUFFERS_INCLUDE_PATH flatbuffers/flatbuffers.h
196 HINTS ${FLATBUFFERS_ROOT}/include /usr/local/include /usr/include)
197
Matthew Benthamd1ae3a62019-02-22 17:30:32 +0000198 message(STATUS "Flatbuffers headers are located at: ${FLATBUFFERS_INCLUDE_PATH}")
telsoa01c577f2c2018-08-31 09:22:23 +0100199
200 find_library(FLATBUFFERS_LIBRARY
201 NAMES libflatbuffers.a flatbuffers
202 HINTS ${FLATBUFFERS_ROOT}/lib /usr/local/lib /usr/lib)
203
Matthew Benthamd1ae3a62019-02-22 17:30:32 +0000204 message(STATUS "Flatbuffers library located at: ${FLATBUFFERS_LIBRARY}")
Nattapat Chaimanowong949f1252019-01-31 15:36:39 +0000205endif()
206
207# Flatbuffers schema support for TF Lite
208if(BUILD_TF_LITE_PARSER)
209 find_path(TF_LITE_SCHEMA_INCLUDE_PATH
210 schema_generated.h
211 HINTS ${TF_LITE_GENERATED_PATH})
212
Matthew Benthamd1ae3a62019-02-22 17:30:32 +0000213 message(STATUS "Tf Lite generated header found at: ${TF_LITE_SCHEMA_INCLUDE_PATH}")
Nattapat Chaimanowong949f1252019-01-31 15:36:39 +0000214
Matthew Benthamd1ae3a62019-02-22 17:30:32 +0000215 add_definitions(-DARMNN_TF_LITE_PARSER)
telsoa01c577f2c2018-08-31 09:22:23 +0100216endif()
217
Kevin May43a799c2019-02-08 16:31:42 +0000218if(BUILD_ARMNN_SERIALIZER)
Kevin May43a799c2019-02-08 16:31:42 +0000219 add_definitions(-DARMNN_SERIALIZER)
Matthew Bentham268509a2019-02-25 13:58:24 +0000220 add_definitions(-DARMNN_SERIALIZER_SCHEMA_PATH="${CMAKE_CURRENT_SOURCE_DIR}/src/armnnSerializer/ArmnnSchema.fbs")
Kevin May43a799c2019-02-08 16:31:42 +0000221endif()
222
telsoa014fcda012018-03-09 14:13:49 +0000223include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include)
Sadik Armagana97a0be2020-03-03 10:44:56 +0000224include_directories(${CMAKE_CURRENT_SOURCE_DIR}/profiling)
telsoa014fcda012018-03-09 14:13:49 +0000225
226# ARM Compute
227# Note that ARM Compute has a different folder layout depending on the branch but also on
228# whether it comes from a prepackaged archive (this is why we add several hints below)
229if(ARMCOMPUTENEON OR ARMCOMPUTECL)
Francis Murtagh82490ee2020-11-04 13:33:41 +0000230 find_path(ARMCOMPUTE_INCLUDE arm_compute/core/CL/OpenCL.h
telsoa014fcda012018-03-09 14:13:49 +0000231 PATHS ${ARMCOMPUTE_ROOT}/include
232 PATHS ${ARMCOMPUTE_ROOT}/applications/arm_compute
233 PATHS ${ARMCOMPUTE_ROOT}
234 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
Francis Murtagh82490ee2020-11-04 13:33:41 +0000235 find_path(ARMCOMPUTE_INCLUDE arm_compute/core/CL/OpenCL.h)
telsoa014fcda012018-03-09 14:13:49 +0000236 include_directories(SYSTEM "${ARMCOMPUTE_INCLUDE}")
237
238 # Find the Arm Compute libraries if not already specified (the user may have already defined this in advance,
239 # e.g. if building clframework as a dependent cmake project)
240 if (NOT DEFINED ARMCOMPUTE_LIBRARIES)
241 # We link to the static variant so that customers don't need to find and build a compatible version of clframework.
242 # First try the folders specified ARMCOMPUTE_BUILD_DIR (with PATH_SUFFIXES for
243 # Windows builds)
Matthew Bentham2624dd72020-05-20 10:44:18 +0100244 if ((NOT DEFINED ARMCOMPUTE_BUILD_DIR) AND (DEFINED ARMCOMPUTE_ROOT))
245 # Default build directory for ComputeLibrary is under the root
246 set(ARMCOMPUTE_BUILD_DIR ${ARMCOMPUTE_ROOT}/build)
247 endif()
248
telsoa014fcda012018-03-09 14:13:49 +0000249 find_library(ARMCOMPUTE_LIBRARY_DEBUG NAMES arm_compute-static
250 PATHS ${ARMCOMPUTE_BUILD_DIR}
251 PATH_SUFFIXES "Debug"
252 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
253 find_library(ARMCOMPUTE_LIBRARY_RELEASE NAMES arm_compute-static
254 PATHS ${ARMCOMPUTE_BUILD_DIR}
255 PATH_SUFFIXES "Release"
256 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
257 find_library(ARMCOMPUTE_CORE_LIBRARY_DEBUG NAMES arm_compute_core-static
258 PATHS ${ARMCOMPUTE_BUILD_DIR}
259 PATH_SUFFIXES "Debug"
260 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
261 find_library(ARMCOMPUTE_CORE_LIBRARY_RELEASE NAMES arm_compute_core-static
262 PATHS ${ARMCOMPUTE_BUILD_DIR}
263 PATH_SUFFIXES "Release"
264 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
265
266 # In case it wasn't there, try a default search (will work in cases where
267 # the library has been installed into a standard location)
268 find_library(ARMCOMPUTE_LIBRARY_DEBUG NAMES arm_compute-static)
269 find_library(ARMCOMPUTE_LIBRARY_RELEASE NAMES arm_compute-static)
270 find_library(ARMCOMPUTE_CORE_LIBRARY_DEBUG NAMES arm_compute_core-static)
271 find_library(ARMCOMPUTE_CORE_LIBRARY_RELEASE NAMES arm_compute_core-static)
272
Matthew Bentham6445cff2020-03-10 11:13:17 +0000273 # In case it wasn't there, try the dynamic libraries
274 # This case will get used in a linux setup where the Compute Library
275 # has been installed in a standard system library path as a dynamic library
276 find_library(ARMCOMPUTE_LIBRARY_DEBUG NAMES arm_compute)
277 find_library(ARMCOMPUTE_LIBRARY_RELEASE NAMES arm_compute)
278 find_library(ARMCOMPUTE_CORE_LIBRARY_DEBUG NAMES arm_compute_core)
279 find_library(ARMCOMPUTE_CORE_LIBRARY_RELEASE NAMES arm_compute_core)
280
telsoa014fcda012018-03-09 14:13:49 +0000281 set(ARMCOMPUTE_LIBRARIES
282 debug ${ARMCOMPUTE_LIBRARY_DEBUG} ${ARMCOMPUTE_CORE_LIBRARY_DEBUG}
283 optimized ${ARMCOMPUTE_LIBRARY_RELEASE} ${ARMCOMPUTE_CORE_LIBRARY_RELEASE} )
284 endif()
285endif()
286
287# ARM Compute NEON backend
288if(ARMCOMPUTENEON)
289 # Add preprocessor definition for ARM Compute NEON
290 add_definitions(-DARMCOMPUTENEON_ENABLED)
291 # The ARM Compute headers contain some NEON intrinsics, so we need to build armnn with NEON support on armv7
292 if(${CMAKE_SYSTEM_PROCESSOR} MATCHES armv7 AND COMPILER_IS_GNU_LIKE)
293 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -mfpu=neon")
294 endif()
295endif()
296
297# ARM Compute OpenCL backend
298if(ARMCOMPUTECL)
Nikhil Rajb55d3662021-02-11 14:49:19 +0000299 # verify we have a valid flatbuffers include path
300 find_path(FLATBUFFERS_INCLUDE_PATH flatbuffers/flatbuffers.h
301 HINTS ${FLATBUFFERS_ROOT}/include /usr/local/include /usr/include)
302
303 message(STATUS "Flatbuffers headers are located at: ${FLATBUFFERS_INCLUDE_PATH}")
304
305 find_library(FLATBUFFERS_LIBRARY
306 NAMES libflatbuffers.a flatbuffers
307 HINTS ${FLATBUFFERS_ROOT}/lib /usr/local/lib /usr/lib)
308
309 message(STATUS "Flatbuffers library located at: ${FLATBUFFERS_LIBRARY}")
310
telsoa014fcda012018-03-09 14:13:49 +0000311 # Always use Arm compute library OpenCL headers
312 find_path(OPENCL_INCLUDE CL/cl2.hpp
313 PATHS ${ARMCOMPUTE_ROOT}/include
314 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
315
Matthew Bentham3b72db02018-10-11 09:47:01 +0100316 # Link against libOpenCL in opencl-1.2-stubs, but don't search there at runtime
317 link_libraries(-L${ARMCOMPUTE_BUILD_DIR}/opencl-1.2-stubs)
318 set(OPENCL_LIBRARIES OpenCL)
telsoa014fcda012018-03-09 14:13:49 +0000319
Matthew Bentham416b41d2020-02-05 22:15:26 +0000320 include_directories(SYSTEM ${OPENCL_INCLUDE})
telsoa014fcda012018-03-09 14:13:49 +0000321
322 # Add preprocessor definition for ARM Compute OpenCL
323 add_definitions(-DARMCOMPUTECL_ENABLED)
324
325 set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -DARM_COMPUTE_DEBUG_ENABLED")
326endif()
327
328# Used by both Arm Compute backends, but should be added
329# to the search path after the system directories if necessary
330if(ARMCOMPUTENEON OR ARMCOMPUTECL)
331 find_path(HALF_INCLUDE half/half.hpp)
332 find_path(HALF_INCLUDE half/half.hpp
333 PATHS ${ARMCOMPUTE_ROOT}/include
334 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
telsoa01c577f2c2018-08-31 09:22:23 +0100335 include_directories(SYSTEM ${HALF_INCLUDE})
telsoa014fcda012018-03-09 14:13:49 +0000336endif()
337
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100338# ArmNN reference backend
339if(ARMNNREF)
340 add_definitions(-DARMNNREF_ENABLED)
Matteo Martincighe67edb22019-08-14 14:05:46 +0100341endif()
342
Colm Donelanaa93d982020-06-28 08:16:46 +0100343# This is the root for the dynamic backend tests to search for dynamic
344# backends. By default it will be the project build directory.
Rob Hughes35c31c02020-08-17 15:47:20 +0100345add_definitions(-DDYNAMIC_BACKEND_BUILD_DIR="${PROJECT_BINARY_DIR}")
Colm Donelanaa93d982020-06-28 08:16:46 +0100346
Narumol Prangnawarat60a20fb2019-12-09 17:24:41 +0000347# ArmNN dynamic backend
348if(DYNAMIC_BACKEND_PATHS)
349 add_definitions(-DARMNN_DYNAMIC_BACKEND_ENABLED)
350endif()
351
Narumol Prangnawarat867eba52020-02-03 12:29:56 +0000352if(SAMPLE_DYNAMIC_BACKEND)
353 add_definitions(-DSAMPLE_DYNAMIC_BACKEND_ENABLED)
354endif()
355
telsoa014fcda012018-03-09 14:13:49 +0000356# Streamline annotate
357if(PROFILING_BACKEND_STREAMLINE)
358 include_directories("${GATOR_ROOT}/annotate")
359 add_definitions(-DARMNN_STREAMLINE_ENABLED)
360endif()
361
telsoa01c577f2c2018-08-31 09:22:23 +0100362if(HEAP_PROFILING OR LEAK_CHECKING)
surmeh013537c2c2018-05-18 16:31:43 +0100363 # enable heap profiling for everything except for referencetests
364 if(NOT ${PROJECT_NAME} STREQUAL "referencetests")
365 find_path(HEAP_PROFILER_INCLUDE gperftools/heap-profiler.h
366 PATHS ${GPERFTOOLS_ROOT}/include
367 NO_DEFAULT_PATH NO_CMAKE_FIND_ROOT_PATH)
368 include_directories(SYSTEM "${HEAP_PROFILER_INCLUDE}")
369 find_library(GPERF_TOOLS_LIBRARY
370 NAMES tcmalloc_debug
371 HINTS ${GPERFTOOLS_ROOT}/lib)
372 link_directories(${GPERFTOOLS_ROOT}/lib)
373
374 link_libraries(${GPERF_TOOLS_LIBRARY})
telsoa01c577f2c2018-08-31 09:22:23 +0100375 if (HEAP_PROFILING)
376 add_definitions("-DARMNN_HEAP_PROFILING_ENABLED=1")
377 endif()
378 if (LEAK_CHECKING)
379 add_definitions("-DARMNN_LEAK_CHECKING_ENABLED=1")
380 endif()
surmeh013537c2c2018-05-18 16:31:43 +0100381 else()
Rob Hughes721b82f2019-11-15 09:04:17 +0000382 message(STATUS "Heap profiling and leak checking are disabled for referencetests")
surmeh013537c2c2018-05-18 16:31:43 +0100383 endif()
384else()
385 # Valgrind only works with gperftools version number <= 2.4
386 CHECK_INCLUDE_FILE(valgrind/memcheck.h VALGRIND_FOUND)
387endif()
388
389
telsoa01c577f2c2018-08-31 09:22:23 +0100390if(NOT BUILD_TF_LITE_PARSER)
391 message(STATUS "Tensorflow Lite parser support is disabled")
392endif()
David Beck10b4dfd2018-09-19 12:03:20 +0100393
Nattapat Chaimanowong949f1252019-01-31 15:36:39 +0000394if(NOT BUILD_ARMNN_SERIALIZER)
395 message(STATUS "Armnn Serializer support is disabled")
396endif()
397
Pavel Macenauer59e057f2020-04-15 14:17:26 +0000398if(NOT BUILD_PYTHON_WHL)
399 message(STATUS "PyArmNN wheel package is disabled")
400endif()
401
402if(NOT BUILD_PYTHON_SRC)
403 message(STATUS "PyArmNN source package is disabled")
404endif()
405
406if(BUILD_PYTHON_WHL OR BUILD_PYTHON_SRC)
407 find_package(PythonInterp 3 REQUIRED)
408 if(NOT ${PYTHONINTERP_FOUND})
409 message(FATAL_ERROR "Python 3.x required to build PyArmNN, but not found")
410 endif()
411
412 find_package(PythonLibs 3 REQUIRED)
413 if(NOT ${PYTHONLIBS_FOUND})
414 message(FATAL_ERROR "Python 3.x development package required to build PyArmNN, but not found")
415 endif()
416
417 find_package(SWIG 4 REQUIRED)
418 if(NOT ${SWIG_FOUND})
419 message(FATAL_ERROR "SWIG 4.x requried to build PyArmNN, but not found")
420 endif()
421endif()
422
David Beck10b4dfd2018-09-19 12:03:20 +0100423# ArmNN source files required for all build options
424include_directories(SYSTEM third-party)