blob: 127724b65a882a321b130bed9c2c597e7a93b791 [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")
Richard Burtonb4123fd2022-03-04 09:19:09 +000057 set(ETHOS_U_NPU_ID "u55") # Currently only u55 is supported by TFLite Micro.
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
71# Primary target
72list(APPEND MAKE_TARGETS_LIST "microlite")
73message(STATUS "TensorFlow Lite Micro build to be called for these targets: ${MAKE_TARGETS_LIST}")
74
75# Commands and targets
76add_custom_target(tensorflow_build ALL
77
78 # Command to build the TensorFlow Lite Micro library
79 COMMAND make -j${J} -f ${TENSORFLOW_LITE_MICRO_PATH}/tools/make/Makefile ${MAKE_TARGETS_LIST}
80 TARGET_TOOLCHAIN_ROOT=${TENSORFLOW_LITE_MICRO_TARGET_TOOLCHAIN_ROOT}
81 TOOLCHAIN=${TENSORFLOW_LITE_MICRO_TOOLCHAIN}
82 GENDIR=${TENSORFLOW_LITE_MICRO_GENDIR}
83 TARGET=${TENSORFLOW_LITE_MICRO_TARGET}
84 TARGET_ARCH=${TENSORFLOW_LITE_MICRO_TARGET_ARCH}
85 BUILD_TYPE=${TENSORFLOW_LITE_MICRO_BUILD_TYPE}
alexander31ae9f02022-02-10 16:15:54 +000086 CMSIS_PATH=${CMSIS_SRC_PATH}
alexander3c798932021-03-26 21:42:19 +000087 # Conditional arguments
Richard Burtonb4123fd2022-03-04 09:19:09 +000088 $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_ARCH=${ETHOS_U_NPU_ID}>
Nina Drozd68549b42022-02-15 14:32:57 +000089 $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_DRIVER_PATH=${ETHOS_U_NPU_DRIVER_SRC_PATH}>
90 $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_DRIVER_LIBS=$<TARGET_FILE:ethosu_core_driver>>
91
Richard Burton0d110592021-08-12 17:26:30 +010092 $<$<BOOL:${TENSORFLOW_LITE_MICRO_CORE_OPTIMIZATION_LEVEL}>:CORE_OPTIMIZATION_LEVEL=${TENSORFLOW_LITE_MICRO_CORE_OPTIMIZATION_LEVEL}>
93 $<$<BOOL:${TENSORFLOW_LITE_MICRO_KERNEL_OPTIMIZATION_LEVEL}>:KERNEL_OPTIMIZATION_LEVEL=${TENSORFLOW_LITE_MICRO_KERNEL_OPTIMIZATION_LEVEL}>
alexander3c798932021-03-26 21:42:19 +000094 $<$<BOOL:${TENSORFLOW_LITE_MICRO_OPTIMIZED_KERNEL}>:OPTIMIZED_KERNEL_DIR=${TENSORFLOW_LITE_MICRO_OPTIMIZED_KERNEL}>
95 $<$<BOOL:${TENSORFLOW_LITE_MICRO_CO_PROCESSOR}>:CO_PROCESSOR=${TENSORFLOW_LITE_MICRO_CO_PROCESSOR}>
96
97 # Command to copy over the generated library to the local build tree.
98 COMMAND ${CMAKE_COMMAND} -E copy ${TENSORFLOW_LITE_MICRO_GENDIR}/lib/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
99 ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
100
101 COMMENT "Building TensorFlow Lite Micro library..."
102
103 BYPRODUCTS ${TENSORFLOW_SRC_PATH}/tensorflow/tensorflow/lite/micro/tools/make/downloads
104 ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
105 ${TENSORFLOW_LITE_MICRO_GENDIR}/lib/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}
106
107 WORKING_DIRECTORY ${TENSORFLOW_SRC_PATH})
108
109# Create library
alexander31ae9f02022-02-10 16:15:54 +0000110
Nina Drozd68549b42022-02-15 14:32:57 +0000111
112set(TENSORFLOW_LITE_MICRO_TARGET tensorflow-lite-micro)
113add_library(${TENSORFLOW_LITE_MICRO_TARGET} STATIC IMPORTED)
114
115if(ETHOS_U_NPU_ENABLED)
116 add_dependencies(tensorflow_build ethosu_core_driver)
117endif()
alexander31ae9f02022-02-10 16:15:54 +0000118
alexander3c798932021-03-26 21:42:19 +0000119add_dependencies(tensorflow-lite-micro tensorflow_build)
alexander31ae9f02022-02-10 16:15:54 +0000120
ayamas0115f80702021-11-18 14:22:23 +0000121set_property(TARGET tensorflow-lite-micro PROPERTY IMPORTED_LOCATION
122 "${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/${TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME}")
alexander31ae9f02022-02-10 16:15:54 +0000123
124target_include_directories(tensorflow-lite-micro
125 INTERFACE
126 ${TENSORFLOW_SRC_PATH})
127
128target_compile_definitions(tensorflow-lite-micro
129 INTERFACE
Nina Drozd68549b42022-02-15 14:32:57 +0000130 TF_LITE_STATIC_MEMORY)