mirror of
https://github.com/apache/nuttx-apps.git
synced 2026-09-12 13:40:16 +00:00
a81fe2f4d9
Build tflm_hello from Makefile, allocate tensors in tflm, keep syslog strings without debug, and honor CONFIG_DARKNET_YOLO_VER. Assisted-by: Cursor:Grok-4.6 Signed-off-by: Abhishek Mishra <mishra.abhishek2808@gmail.com>
153 lines
5.8 KiB
Makefile
153 lines
5.8 KiB
Makefile
############################################################################
|
|
# apps/mlearning/tflite-micro/Makefile
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
# contributor license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright ownership. The
|
|
# ASF licenses this file to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance with the
|
|
# License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
#
|
|
############################################################################
|
|
|
|
include $(APPDIR)/Make.defs
|
|
|
|
CSRCS :=
|
|
CXXEXT := .cc
|
|
CXXSRCS :=
|
|
COMMON_FLAGS :=
|
|
TFLM_UNPACK := tflite-micro
|
|
TFLM_VER := cfa4c91d1b36c37c7c104b9c664615e59f1abfe3
|
|
TFLM_DIR := $(APPDIR)/mlearning/tflite-micro/$(TFLM_UNPACK)
|
|
|
|
tflite-micro.zip:
|
|
$(Q) curl -L https://github.com/tensorflow/tflite-micro/archive/$(TFLM_VER).zip -o tflite-micro.zip
|
|
$(Q) unzip -o tflite-micro.zip
|
|
$(Q) mv tflite-micro-$(TFLM_VER) $(TFLM_UNPACK)
|
|
$(Q) patch -d $(TFLM_UNPACK) -p1 < tflite-micro.patch
|
|
$(Q) patch -d $(TFLM_UNPACK) -p1 < 0001-dequantize-int8.patch
|
|
$(Q) patch -d $(TFLM_UNPACK) -p1 < 0002-quantize-int8.patch
|
|
$(Q) patch -d $(TFLM_UNPACK) -p1 < 0003-mean-int8.patch
|
|
$(Q) patch -d $(TFLM_UNPACK) -p1 < 0004-tflite-add-extern-C-to-main-function-to-avoid-c-mang.patch
|
|
|
|
# Download and unpack tarball if no git repo found
|
|
ifeq ($(wildcard $(TFLM_UNPACK)/.git),)
|
|
context:: tflite-micro.zip
|
|
distclean::
|
|
$(call DELDIR, $(TFLM_UNPACK))
|
|
$(call DELFILE, tflite-micro.zip)
|
|
endif
|
|
|
|
COMMON_FLAGS += ${INCDIR_PREFIX}$(TFLM_DIR)
|
|
COMMON_FLAGS += ${INCDIR_PREFIX}$(APPDIR)/math/gemmlowp/gemmlowp
|
|
COMMON_FLAGS += ${INCDIR_PREFIX}$(APPDIR)/math/kissfft/kissfft
|
|
COMMON_FLAGS += ${INCDIR_PREFIX}$(APPDIR)/math/ruy/ruy
|
|
COMMON_FLAGS += -Wno-shadow
|
|
COMMON_FLAGS += -Wno-sign-compare
|
|
COMMON_FLAGS += -Wno-undef
|
|
COMMON_FLAGS += -Wno-unused-variable
|
|
COMMON_FLAGS += -DTFLITE_EMULATE_FLOAT
|
|
COMMON_FLAGS += -DTF_LITE_DISABLE_X86_NEON
|
|
ifneq ($(CONFIG_TFLITEMICRO_DEBUG),)
|
|
COMMON_FLAGS += -DTF_LITE_SHOW_MEMORY_USE
|
|
COMMON_FLAGS += -DTF_LITE_USE_CTIME
|
|
endif
|
|
|
|
ifeq ($(CONFIG_TFLITEMICRO_DEBUG)$(CONFIG_TFLITEMICRO_SYSLOG),)
|
|
COMMON_FLAGS += -DTF_LITE_STRIP_ERROR_STRINGS
|
|
endif
|
|
|
|
# tflite micro
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/c/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/core/c/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/core/api/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/kernels/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/kernels/internal/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/kernels/internal/optimized/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/kernels/internal/reference/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/micro/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/micro/arena_allocator/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/micro/kernels/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/micro/memory_planner/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/micro/tflite_bridge/*.cc)
|
|
CXXSRCS += $(wildcard $(TFLM_DIR)/tensorflow/lite/schema/*.cc)
|
|
CXXSRCS := $(filter-out %test.cc, $(CXXSRCS))
|
|
|
|
ifneq ($(CONFIG_TFLITEMICRO_SYSLOG),)
|
|
CXXSRCS := $(filter-out %micro_log.cc, $(CXXSRCS))
|
|
CXXSRCS += $(wildcard $(CURDIR)/tflm_syslog.cc)
|
|
endif
|
|
|
|
# cmsis
|
|
ifneq ($(CONFIG_MLEARNING_CMSIS_NN),)
|
|
COMMON_FLAGS += -DCMSIS_NN
|
|
COMMON_FLAGS += ${INCDIR_PREFIX}$(APPDIR)/mlearning/cmsis-nn/cmsis-nn
|
|
|
|
CMSIS_NN_SRCS := $(wildcard $(TFLM_DIR)/tensorflow/lite/micro/kernels/cmsis_nn/*.cc)
|
|
UNNEEDED_SRCS := $(addprefix $(TFLM_DIR)/tensorflow/lite/micro/kernels/, $(notdir $(CMSIS_NN_SRCS)))
|
|
CXXSRCS := $(filter-out $(UNNEEDED_SRCS), $(CXXSRCS))
|
|
CXXSRCS += $(CMSIS_NN_SRCS)
|
|
|
|
ifneq ($(CONFIG_ARM_NEON),)
|
|
CSRCS += operators/neon/arm_convolve_s8.c
|
|
CSRCS += operators/neon/arm_nn_mat_mult_kernel_s8_s16.c
|
|
CSRCS += operators/neon/arm_q7_to_q15_with_offset.c
|
|
CSRCS += operators/neon/arm_elementwise_add_s8.c
|
|
endif
|
|
endif
|
|
|
|
# extra hardware support.
|
|
-include $(TFLM_DIR)/tensorflow/lite/micro/nuttx/Makefile
|
|
|
|
PROGNAME :=
|
|
PRIORITY :=
|
|
STACKSIZE :=
|
|
MAINSRC :=
|
|
|
|
ifneq ($(CONFIG_TFLITEMICRO_TOOL),)
|
|
PROGNAME += tflm
|
|
PRIORITY += $(CONFIG_TFLITEMICRO_TOOL_PRIORITY)
|
|
STACKSIZE += $(CONFIG_TFLITEMICRO_TOOL_STACKSIZE)
|
|
MAINSRC += tflm_tool.cc
|
|
endif
|
|
|
|
ifneq ($(CONFIG_TFLITEMICRO_HELLOWORLD),)
|
|
TFLM_HW_DIR := $(TFLM_UNPACK)/tensorflow/lite/micro/examples/hello_world
|
|
TFLM_HW_MODELS := $(TFLM_HW_DIR)/models
|
|
|
|
ifeq ($(wildcard $(TFLM_UNPACK)/.git),)
|
|
$(TFLM_HW_MODELS)/hello_world_float_model_data.h \
|
|
$(TFLM_HW_MODELS)/hello_world_int8_model_data.h: tflite-micro.zip
|
|
endif
|
|
|
|
$(TFLM_HW_MODELS)/hello_world_float_model_data.h:
|
|
$(Q) ( cd $(TFLM_HW_MODELS) && xxd -i hello_world_float.tflite ) | \
|
|
sed -e 's/hello_world_float_tflite/g_hello_world_float_model_data/g' > $@
|
|
|
|
$(TFLM_HW_MODELS)/hello_world_int8_model_data.h:
|
|
$(Q) ( cd $(TFLM_HW_MODELS) && xxd -i hello_world_int8.tflite ) | \
|
|
sed -e 's/hello_world_int8_tflite/g_hello_world_int8_model_data/g' > $@
|
|
|
|
context:: $(TFLM_HW_MODELS)/hello_world_float_model_data.h
|
|
context:: $(TFLM_HW_MODELS)/hello_world_int8_model_data.h
|
|
|
|
PROGNAME += tflm_hello
|
|
PRIORITY += $(CONFIG_TFLITEMICRO_HELLOWORLD_PRIORITY)
|
|
STACKSIZE += $(CONFIG_TFLITEMICRO_HELLOWORLD_STACKSIZE)
|
|
MAINSRC += $(TFLM_HW_DIR)/hello_world_test.cc
|
|
endif
|
|
|
|
CFLAGS += ${COMMON_FLAGS}
|
|
CXXFLAGS += ${COMMON_FLAGS}
|
|
|
|
include $(APPDIR)/Application.mk
|