alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 1 | #---------------------------------------------------------------------------- |
Isabella Gottardi | c64f506 | 2022-01-21 15:27:13 +0000 | [diff] [blame] | 2 | # Copyright (c) 2021 - 2022 Arm Limited. All rights reserved. |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 3 | # SPDX-License-Identifier: Apache-2.0 |
| 4 | # |
| 5 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | # you may not use this file except in compliance with the License. |
| 7 | # You may obtain a copy of the License at |
| 8 | # |
| 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | # |
| 11 | # Unless required by applicable law or agreed to in writing, software |
| 12 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | # See the License for the specific language governing permissions and |
| 15 | # limitations under the License. |
| 16 | #---------------------------------------------------------------------------- |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 17 | cmake_minimum_required(VERSION 3.15.6) |
| 18 | include(ExternalProject) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 19 | |
| 20 | # Build in release mode by default |
| 21 | if (NOT CMAKE_BUILD_TYPE STREQUAL Debug) |
| 22 | set(CMAKE_BUILD_TYPE Release CACHE INTERNAL "") |
| 23 | endif() |
| 24 | |
| 25 | message(STATUS "Build type is set to ${CMAKE_BUILD_TYPE}") |
| 26 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 27 | # Set language standards. |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 28 | set(CMAKE_C_STANDARD 99) |
ayamas01 | 15f8070 | 2021-11-18 14:22:23 +0000 | [diff] [blame] | 29 | set(CMAKE_CXX_STANDARD 14) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 30 | |
| 31 | # Make the standard a requirement => prevent fallback to previous |
| 32 | # supported standard |
| 33 | set(CMAKE_C_STANDARD_REQUIRED ON) |
| 34 | set(CMAKE_CXX_STANDARD_REQUIRED ON) |
| 35 | |
| 36 | # We want to pass standard C/C++ flags, without gnu extensions |
| 37 | set(CMAKE_C_EXTENSIONS OFF) |
| 38 | set(CMAKE_CXX_EXTENSIONS OFF) |
| 39 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 40 | set(SCRIPTS_DIR ${CMAKE_CURRENT_SOURCE_DIR}/scripts) |
| 41 | set(CMAKE_SCRIPTS_DIR ${SCRIPTS_DIR}/cmake) |
| 42 | set(CMAKE_TOOLCHAIN_DIR ${CMAKE_SCRIPTS_DIR}/toolchains) |
Kshitij Sisodia | f9c19ea | 2021-05-07 16:08:14 +0100 | [diff] [blame] | 43 | set(DOWNLOAD_DEP_DIR ${CMAKE_BINARY_DIR}/dependencies) |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 44 | set(DEPENDENCY_ROOT_DIR ${CMAKE_CURRENT_SOURCE_DIR}/dependencies) |
Kshitij Sisodia | c22e80e | 2022-03-14 09:26:48 +0000 | [diff] [blame] | 45 | set(CORE_PLATFORM_DIR ${DEPENDENCY_ROOT_DIR}/core-platform) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 46 | |
Kshitij Sisodia | ddcb56d | 2021-05-11 14:46:01 +0100 | [diff] [blame] | 47 | set(RESOURCES_DIR ${CMAKE_CURRENT_SOURCE_DIR}/resources_downloaded |
| 48 | CACHE PATH "Resources directory") |
Kshitij Sisodia | c22e80e | 2022-03-14 09:26:48 +0000 | [diff] [blame] | 49 | set(HAL_PLATFORM_DIR ${CMAKE_CURRENT_SOURCE_DIR}/source/hal/source/platform) |
Kshitij Sisodia | ddcb56d | 2021-05-11 14:46:01 +0100 | [diff] [blame] | 50 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 51 | include(${CMAKE_SCRIPTS_DIR}/source_gen_utils.cmake) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 52 | |
| 53 | if (${CMAKE_BINARY_DIR} STREQUAL ${CMAKE_SOURCE_DIR}) |
| 54 | message(FATAL_ERROR "Source and build are in the same directory") |
| 55 | else() |
| 56 | message(STATUS "Source directory: ${CMAKE_SOURCE_DIR}") |
| 57 | message(STATUS "Binary directory: ${CMAKE_BINARY_DIR}") |
| 58 | endif() |
| 59 | |
Kshitij Sisodia | b59ba68 | 2021-11-23 17:19:52 +0000 | [diff] [blame] | 60 | include(${CMAKE_SCRIPTS_DIR}/common_user_options.cmake) |
alexander | cb8a987 | 2022-02-11 13:23:22 +0000 | [diff] [blame] | 61 | |
Isabella Gottardi | ef2b9dd | 2022-02-16 14:24:03 +0000 | [diff] [blame] | 62 | # Check if the resources_downloaded needs update |
| 63 | check_update_public_resources(${RESOURCES_DIR}) |
| 64 | |
alexander | cb8a987 | 2022-02-11 13:23:22 +0000 | [diff] [blame] | 65 | add_platform_build_configuration(TARGET_PLATFORM ${TARGET_PLATFORM}) |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 66 | |
| 67 | set_platform_global_defaults() |
| 68 | |
| 69 | message(STATUS "Using CMAKE_TOOLCHAIN_FILE: ${CMAKE_TOOLCHAIN_FILE}") |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 70 | |
Kshitij Sisodia | b59ba68 | 2021-11-23 17:19:52 +0000 | [diff] [blame] | 71 | # Make sure the following options are defined before proceeding: |
| 72 | assert_defined(LOG_LEVEL) |
| 73 | assert_defined(TENSORFLOW_SRC_PATH) |
| 74 | assert_defined(TARGET_PLATFORM) |
| 75 | assert_defined(TARGET_SUBSYSTEM) |
| 76 | assert_defined(ETHOS_U_NPU_ENABLED) |
| 77 | assert_defined(USE_CASE_BUILD) |
| 78 | assert_defined(CPU_PROFILE_ENABLED) |
| 79 | assert_defined(CMAKE_TOOLCHAIN_FILE) |
Kshitij Sisodia | f9c19ea | 2021-05-07 16:08:14 +0100 | [diff] [blame] | 80 | |
Nina Drozd | 5916952 | 2022-02-10 13:33:20 +0000 | [diff] [blame] | 81 | if(POLICY CMP0123) |
| 82 | cmake_policy(SET CMP0123 NEW) |
| 83 | endif() |
| 84 | |
Kshitij Sisodia | f9c19ea | 2021-05-07 16:08:14 +0100 | [diff] [blame] | 85 | project(arm_ml_embedded_evaluation_kit |
Richard Burton | b4123fd | 2022-03-04 09:19:09 +0000 | [diff] [blame] | 86 | VERSION 22.02.0 |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 87 | DESCRIPTION "ARM ML Embedded Evaluation Kit" |
| 88 | LANGUAGES C CXX ASM) |
Kshitij Sisodia | f9c19ea | 2021-05-07 16:08:14 +0100 | [diff] [blame] | 89 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 90 | enforce_compiler_version() |
| 91 | setup_source_generator() |
| 92 | |
| 93 | set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib) |
| 94 | set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib) |
| 95 | set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin) |
| 96 | set(SRC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/source) |
Isabella Gottardi | 8520983 | 2021-04-20 14:08:52 +0100 | [diff] [blame] | 97 | set(TEST_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/tests) |
| 98 | list(APPEND USE_CASES_TESTS_SEARCH_DIR_LIST ${TEST_SRCS}/use_case) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 99 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 100 | # We include log target |
| 101 | add_subdirectory(${SRC_PATH}/log ${CMAKE_BINARY_DIR}/log) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 102 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 103 | # We include arm_math target |
| 104 | add_subdirectory(${SRC_PATH}/math ${CMAKE_BINARY_DIR}/math) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 105 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 106 | # We include the hal target |
| 107 | add_subdirectory(${SRC_PATH}/hal ${CMAKE_BINARY_DIR}/hal) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 108 | |
Kshitij Sisodia | e2da7ee | 2022-02-14 11:22:58 +0000 | [diff] [blame] | 109 | # Add the profiler target |
| 110 | if (NOT DEFINED PROFILER_DIR) |
| 111 | set(PROFILER_DIR ${SRC_PATH}/profiler) |
| 112 | endif () |
| 113 | add_subdirectory(${PROFILER_DIR} ${CMAKE_BINARY_DIR}/profiler) |
| 114 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 115 | # Include the tensorflow build target |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 116 | include(${CMAKE_SCRIPTS_DIR}/tensorflow.cmake) |
| 117 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 118 | # Include directories for application module: |
| 119 | set(APPLICATION_INCLUDE_DIRS |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 120 | ${SRC_PATH}/application/tensorflow-lite-micro/include |
Kshitij Sisodia | e2da7ee | 2022-02-14 11:22:58 +0000 | [diff] [blame] | 121 | ${SRC_PATH}/application/main/include) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 122 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 123 | # Source files for application module: |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 124 | file(GLOB_RECURSE SRC_APPLICATION |
| 125 | "${SRC_PATH}/application/main/*.cc" |
| 126 | "${SRC_PATH}/application/main/*.cpp" |
| 127 | "${SRC_PATH}/application/main/*.c" |
| 128 | "${SRC_PATH}/application/main/**/*.cc" |
| 129 | "${SRC_PATH}/application/main/**/*.cpp" |
| 130 | "${SRC_PATH}/application/main/**/*.c" |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 131 | "${SRC_PATH}/application/tensorflow-lite-micro/**/*.cc" |
| 132 | "${SRC_PATH}/application/tensorflow-lite-micro/*.cc" |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 133 | ) |
| 134 | list(FILTER SRC_APPLICATION EXCLUDE REGEX ".*main\\.c.*$") |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 135 | set(SRC_MAIN "${SRC_PATH}/application/main/Main.cc") |
| 136 | set_source_files_properties(${SRC_MAIN} |
| 137 | PROPERTIES COMPILE_DEFINITIONS |
| 138 | "PRJ_VER_STR=\"${PROJECT_VERSION}\";PRJ_DES_STR=\"${PROJECT_DESCRIPTION}\"") |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 139 | |
| 140 | list(JOIN USE_CASE_BUILD "" USE_CASE_BUILD_STR) |
Isabella Gottardi | 8520983 | 2021-04-20 14:08:52 +0100 | [diff] [blame] | 141 | list(APPEND USE_CASES_SEARCH_DIR_LIST ${SRC_PATH}/use_case) |
| 142 | message(STATUS "Use-cases source paths: ${USE_CASES_SEARCH_DIR_LIST}.") |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 143 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 144 | if (${USE_CASE_BUILD_STR} STREQUAL all) |
Isabella Gottardi | 8520983 | 2021-04-20 14:08:52 +0100 | [diff] [blame] | 145 | foreach(USE_CASES_SEARCH_DIR ${USE_CASES_SEARCH_DIR_LIST}) |
| 146 | SUBDIRLIST(USE_CASES_SUBDIRS ${USE_CASES_SEARCH_DIR}) |
| 147 | list(APPEND USE_CASES ${USE_CASES_SUBDIRS}) |
| 148 | endforeach() |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 149 | else() |
| 150 | set(USE_CASES ${USE_CASE_BUILD}) |
| 151 | endif() |
| 152 | |
alexander | cb8a987 | 2022-02-11 13:23:22 +0000 | [diff] [blame] | 153 | list(REMOVE_ITEM USE_CASES "" ${EXCLUDED_USE_CASES}) |
| 154 | message(STATUS "Use-cases excluded by platform configuration: ${EXCLUDED_USE_CASES}") |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 155 | message(STATUS "Building use-cases: ${USE_CASES}.") |
| 156 | foreach(use_case ${USE_CASES}) |
| 157 | |
Isabella Gottardi | 8520983 | 2021-04-20 14:08:52 +0100 | [diff] [blame] | 158 | set(SRC_USE_CASE "") |
| 159 | foreach(USE_CASES_SEARCH_DIR ${USE_CASES_SEARCH_DIR_LIST}) |
| 160 | if (EXISTS ${USE_CASES_SEARCH_DIR}/${use_case}) |
| 161 | message(STATUS "Found sources for use-case ${use_case}") |
| 162 | set(SRC_USE_CASE ${USE_CASES_SEARCH_DIR}) |
| 163 | break() |
| 164 | endif () |
| 165 | endforeach() |
| 166 | |
| 167 | if (${SRC_USE_CASE} STREQUAL "") |
| 168 | message(FATAL_ERROR "Failed to find sources for ${use_case}!") |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 169 | endif () |
Kshitij Sisodia | 661959c | 2021-11-24 10:39:52 +0000 | [diff] [blame] | 170 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 171 | # Executable application: |
| 172 | set(TARGET_NAME "ethos-u-${use_case}") |
| 173 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 174 | set(DEFAULT_MODEL_DIR ${RESOURCES_DIR}/${use_case}) |
| 175 | set(DEFAULT_TEST_DATA_DIR ${DEFAULT_MODEL_DIR}) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 176 | set(SRC_GEN_DIR ${CMAKE_BINARY_DIR}/generated/${use_case}/src) |
| 177 | set(INC_GEN_DIR ${CMAKE_BINARY_DIR}/generated/${use_case}/include) |
| 178 | |
| 179 | # Remove old files and recreate dirs |
| 180 | file(REMOVE_RECURSE ${SRC_GEN_DIR} ${INC_GEN_DIR}) |
| 181 | file(MAKE_DIRECTORY ${SRC_GEN_DIR} ${INC_GEN_DIR}) |
| 182 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 183 | file(GLOB UC_CMAKE_FILE |
| 184 | "${SRC_USE_CASE}/${use_case}/*.cmake") |
| 185 | |
| 186 | include(${UC_CMAKE_FILE}) |
| 187 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 188 | file(GLOB_RECURSE UC_SRC |
Isabella Gottardi | 8520983 | 2021-04-20 14:08:52 +0100 | [diff] [blame] | 189 | "${SRC_USE_CASE}/${use_case}/src/*.cpp" |
| 190 | "${SRC_USE_CASE}/${use_case}/src/*.cc" |
| 191 | "${SRC_USE_CASE}/${use_case}/src/*.c" |
| 192 | "${SRC_USE_CASE}/${use_case}/src/**/*.cpp" |
| 193 | "${SRC_USE_CASE}/${use_case}/src/**/*.cc" |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 194 | "${SRC_USE_CASE}/${use_case}/src/**/*.c") |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 195 | |
| 196 | file(GLOB_RECURSE SRC_GEN |
| 197 | "${SRC_GEN_DIR}/*.cc" |
| 198 | "${SRC_GEN_DIR}/*.cpp" |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 199 | "${SRC_GEN_DIR}/*.c") |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 200 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 201 | set(UC_INCLUDE ${SRC_USE_CASE}/${use_case}/include) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 202 | |
Kshitij Sisodia | aa5e1f6 | 2021-09-24 14:42:08 +0100 | [diff] [blame] | 203 | if (DEFINED ${use_case}_COMPILE_DEFS) |
| 204 | message(STATUS "Additional compilation flags for ${use_case}: ${${use_case}_COMPILE_DEFS}") |
| 205 | set_source_files_properties(${UC_SRC} |
| 206 | PROPERTIES COMPILE_DEFINITIONS |
| 207 | "${${use_case}_COMPILE_DEFS}") |
| 208 | endif() |
| 209 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 210 | set(UC_LIB_NAME lib${TARGET_NAME}) |
| 211 | |
| 212 | # Consolidated application static lib: |
| 213 | add_library(${UC_LIB_NAME} STATIC |
| 214 | ${SRC_APPLICATION} |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 215 | ${UC_SRC} |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 216 | ${SRC_GEN}) |
| 217 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 218 | target_include_directories(${UC_LIB_NAME} PUBLIC |
| 219 | ${APPLICATION_INCLUDE_DIRS} |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 220 | ${UC_INCLUDE} |
| 221 | ${INC_GEN_DIR} |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 222 | ${TENSORFLOW_SRC_PATH}/tensorflow/lite/micro/tools/make/downloads/flatbuffers/include) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 223 | |
| 224 | # Set the activation buffer size |
| 225 | target_compile_definitions(${UC_LIB_NAME} PUBLIC |
| 226 | "ACTIVATION_BUF_SZ=${${use_case}_ACTIVATION_BUF_SZ}") |
| 227 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 228 | target_link_libraries(${UC_LIB_NAME} PUBLIC |
| 229 | log |
| 230 | arm_math |
| 231 | hal |
Kshitij Sisodia | e2da7ee | 2022-02-14 11:22:58 +0000 | [diff] [blame] | 232 | profiler |
| 233 | tensorflow-lite-micro) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 234 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 235 | add_executable(${TARGET_NAME} ${SRC_MAIN}) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 236 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 237 | target_link_libraries(${TARGET_NAME} PUBLIC ${UC_LIB_NAME}) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 238 | |
alexander | 31ae9f0 | 2022-02-10 16:15:54 +0000 | [diff] [blame] | 239 | platform_custom_post_build(TARGET_NAME ${TARGET_NAME}) |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 240 | |
alexander | 3c79893 | 2021-03-26 21:42:19 +0000 | [diff] [blame] | 241 | endforeach() |
| 242 | |
Kshitij Sisodia | a1256e3 | 2022-02-23 14:40:45 +0000 | [diff] [blame] | 243 | print_useroptions() |