blob: 284c70a6246250a037edfb63cf4f575402da0e03 [file] [log] [blame]
alexander3c798932021-03-26 21:42:19 +00001#----------------------------------------------------------------------------
alexandercb8a9872022-02-11 13:23:22 +00002# Copyright (c) 2021 - 2022 Arm Limited. All rights reserved.
alexander3c798932021-03-26 21:42:19 +00003# 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#----------------------------------------------------------------------------
17
18include(ProcessorCount)
19ProcessorCount(J)
20
21if (CMAKE_BUILD_TYPE STREQUAL Debug)
Richard Burton0d110592021-08-12 17:26:30 +010022 set(TENSORFLOW_LITE_MICRO_CORE_OPTIMIZATION_LEVEL "-O0")
23 set(TENSORFLOW_LITE_MICRO_KERNEL_OPTIMIZATION_LEVEL "-O0")
alexander3c798932021-03-26 21:42:19 +000024elseif (CMAKE_BUILD_TYPE STREQUAL Release)
Richard Burton0d110592021-08-12 17:26:30 +010025 set(TENSORFLOW_LITE_MICRO_CORE_OPTIMIZATION_LEVEL "-O3")
26 set(TENSORFLOW_LITE_MICRO_KERNEL_OPTIMIZATION_LEVEL "-O3")
alexander3c798932021-03-26 21:42:19 +000027endif()
28
Kshitij Sisodiab59ba682021-11-23 17:19:52 +000029assert_defined(TENSORFLOW_LITE_MICRO_BUILD_TYPE)
30assert_defined(TENSORFLOW_LITE_MICRO_CLEAN_DOWNLOADS)
31assert_defined(TENSORFLOW_LITE_MICRO_CLEAN_BUILD)
alexander3c798932021-03-26 21:42:19 +000032
33if (CMAKE_CXX_COMPILER_ID STREQUAL "ARMClang")
34 set(TENSORFLOW_LITE_MICRO_TOOLCHAIN "armclang")
35elseif (CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
36 set(TENSORFLOW_LITE_MICRO_TOOLCHAIN "gcc")
37else ()
38 message(FATAL_ERROR "No compiler ID is set")
39endif()
40
41get_filename_component(TENSORFLOW_LITE_MICRO_TARGET_TOOLCHAIN_ROOT ${CMAKE_C_COMPILER} DIRECTORY)
42set(TENSORFLOW_LITE_MICRO_TARGET_TOOLCHAIN_ROOT "${TENSORFLOW_LITE_MICRO_TARGET_TOOLCHAIN_ROOT}/")
43
44set(TENSORFLOW_LITE_MICRO_PATH "${TENSORFLOW_SRC_PATH}/tensorflow/lite/micro")
45set(TENSORFLOW_LITE_MICRO_GENDIR ${CMAKE_CURRENT_BINARY_DIR}/tensorflow/)
alexander31ae9f02022-02-10 16:15:54 +000046set(TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME "libtensorflow-microlite.a")
ayamas0115f80702021-11-18 14:22:23 +000047
alexander3c798932021-03-26 21:42:19 +000048if (TARGET_PLATFORM STREQUAL native)
49 set(TENSORFLOW_LITE_MICRO_TARGET "linux")
50 set(TENSORFLOW_LITE_MICRO_TARGET_ARCH x86_64)
51else()
Nina Drozd68549b42022-02-15 14:32:57 +000052 set(TENSORFLOW_LITE_MICRO_TARGET "cortex_m_generic")
alexander3c798932021-03-26 21:42:19 +000053 set(TENSORFLOW_LITE_MICRO_TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES})
Cisco Cervelleraf085fa52021-08-02 09:32:07 +010054 if(ETHOS_U_NPU_ENABLED)
alexander3c798932021-03-26 21:42:19 +000055 # Arm Ethos-U55 NPU is the co-processor for ML workload:
56 set(TENSORFLOW_LITE_MICRO_CO_PROCESSOR "ethos_u")
Nina Drozd68549b42022-02-15 14:32:57 +000057 string(TOLOWER ${ETHOS_U_NPU_ID} ETHOSU_ARCH)
alexander3c798932021-03-26 21:42:19 +000058 endif()
59
60 set(TENSORFLOW_LITE_MICRO_OPTIMIZED_KERNEL "cmsis_nn")
alexander3c798932021-03-26 21:42:19 +000061endif()
62
63if (TENSORFLOW_LITE_MICRO_CLEAN_DOWNLOADS)
64 list(APPEND MAKE_TARGETS_LIST "clean_downloads")
65endif()
66
67if (TENSORFLOW_LITE_MICRO_CLEAN_BUILD)
68 list(APPEND MAKE_TARGETS_LIST "clean")
69endif()
70
Richard Burton527b4742021-11-26 12:17:19 +000071if (ETHOS_U_NPU_ID)
72 string(TOLOWER ${ETHOS_U_NPU_ID} ETHOSU_ARCH)
73endif()
74
alexander3c798932021-03-26 21:42:19 +000075# Primary target
76list(APPEND MAKE_TARGETS_LIST "microlite")
77message(STATUS "TensorFlow Lite Micro build to be called for these targets: ${MAKE_TARGETS_LIST}")
78
79# Commands and targets
80add_custom_target(tensorflow_build ALL
81
82 # Command to build the TensorFlow Lite Micro library
83 COMMAND make -j${J} -f ${TENSORFLOW_LITE_MICRO_PATH}/tools/make/Makefile ${MAKE_TARGETS_LIST}
84 TARGET_TOOLCHAIN_ROOT=${TENSORFLOW_LITE_MICRO_TARGET_TOOLCHAIN_ROOT}
85 TOOLCHAIN=${TENSORFLOW_LITE_MICRO_TOOLCHAIN}
86 GENDIR=${TENSORFLOW_LITE_MICRO_GENDIR}
87 TARGET=${TENSORFLOW_LITE_MICRO_TARGET}
88 TARGET_ARCH=${TENSORFLOW_LITE_MICRO_TARGET_ARCH}
89 BUILD_TYPE=${TENSORFLOW_LITE_MICRO_BUILD_TYPE}
alexander31ae9f02022-02-10 16:15:54 +000090 CMSIS_PATH=${CMSIS_SRC_PATH}
alexander3c798932021-03-26 21:42:19 +000091 # Conditional arguments
Nina Drozd68549b42022-02-15 14:32:57 +000092 $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_ARCH=${ETHOSU_ARCH}>
93 $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_DRIVER_PATH=${ETHOS_U_NPU_DRIVER_SRC_PATH}>
94 $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_DRIVER_LIBS=$<TARGET_FILE:ethosu_core_driver>>
95
Richard Burton0d110592021-08-12 17:26:30 +010096 $<$<BOOL:${TENSORFLOW_LITE_MICRO_CORE_OPTIMIZATION_LEVEL}>:CORE_OPTIMIZATION_LEVEL=${TENSORFLOW_LITE_MICRO_CORE_OPTIMIZATION_LEVEL}>
97 $<$<BOOL:${TENSORFLOW_LITE_MICRO_KERNEL_OPTIMIZATION_LEVEL}>:KERNEL_OPTIMIZATION_LEVEL=${TENSORFLOW_LITE_MICRO_KERNEL_OPTIMIZATION_LEVEL}>
alexander3c798932021-03-26 21:42:19 +000098 $<$<BOOL:${TENSORFLOW_LITE_MICRO_OPTIMIZED_KERNEL}>:OPTIMIZED_KERNEL_DIR=${TENSORFLOW_LITE_MICRO_OPTIMIZED_KERNEL}>
99 $<$<BOOL:${TENSORFLOW_LITE_MICRO_CO_PROCESSOR}>:CO_PROCESSOR=${TENSORFLOW_LITE_MICRO_CO_PROCESSOR}>
100
101 # Command to copy over the generated library to the local build tree.
102 COMMAND ${CMAKE_COMMAND} -E copy ${TENSORFLOW_LITE_MICRO_GENDIR}/lib/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
103 ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
104
105 COMMENT "Building TensorFlow Lite Micro library..."
106
107 BYPRODUCTS ${TENSORFLOW_SRC_PATH}/tensorflow/tensorflow/lite/micro/tools/make/downloads
108 ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
109 ${TENSORFLOW_LITE_MICRO_GENDIR}/lib/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
110
111 WORKING_DIRECTORY ${TENSORFLOW_SRC_PATH})
112
113# Create library
alexander31ae9f02022-02-10 16:15:54 +0000114
Nina Drozd68549b42022-02-15 14:32:57 +0000115
116set(TENSORFLOW_LITE_MICRO_TARGET tensorflow-lite-micro)
117add_library(${TENSORFLOW_LITE_MICRO_TARGET} STATIC IMPORTED)
118
119if(ETHOS_U_NPU_ENABLED)
120 add_dependencies(tensorflow_build ethosu_core_driver)
121endif()
alexander31ae9f02022-02-10 16:15:54 +0000122
alexander3c798932021-03-26 21:42:19 +0000123add_dependencies(tensorflow-lite-micro tensorflow_build)
alexander31ae9f02022-02-10 16:15:54 +0000124
ayamas0115f80702021-11-18 14:22:23 +0000125set_property(TARGET tensorflow-lite-micro PROPERTY IMPORTED_LOCATION
126 "${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}")
alexander31ae9f02022-02-10 16:15:54 +0000127
128target_include_directories(tensorflow-lite-micro
129 INTERFACE
130 ${TENSORFLOW_SRC_PATH})
131
132target_compile_definitions(tensorflow-lite-micro
133 INTERFACE
Nina Drozd68549b42022-02-15 14:32:57 +0000134 TF_LITE_STATIC_MEMORY)