diff --git a/mlearning/tflite-micro/CMakeLists.txt b/mlearning/tflite-micro/CMakeLists.txt index 919dcd146..af44b7020 100644 --- a/mlearning/tflite-micro/CMakeLists.txt +++ b/mlearning/tflite-micro/CMakeLists.txt @@ -62,16 +62,12 @@ if(CONFIG_TFLITEMICRO) set(COMMON_FLAGS -DTF_LITE_STATIC_MEMORY + -DTFLITE_EMULATE_FLOAT -DTF_LITE_DISABLE_X86_NEON - -Wno-sign-compare - -Wno-unused-variable - -Wno-undef -Wno-shadow - -O3) - - if(CONFIG_MLEARNING_CMSIS_NN) - list(APPEND COMMON_FLAGS -DCMSIS_NN) - endif() + -Wno-sign-compare + -Wno-undef + -Wno-unused-variable) if(CONFIG_TFLITEMICRO_DEBUG) list(APPEND COMMON_FLAGS -DTF_LITE_SHOW_MEMORY_USE) @@ -132,7 +128,7 @@ if(CONFIG_TFLITEMICRO) endif() endforeach() - # 添加CMSIS_NN文件 + # Add CMSIS-NN kernel sources list(APPEND NEW_TFLITE_MICRO_SRCS ${CMSIS_NN_SRCS}) set(TFLITE_MICRO_SRCS ${NEW_TFLITE_MICRO_SRCS}) @@ -153,9 +149,8 @@ if(CONFIG_TFLITEMICRO) set(INCDIR ${NUTTX_APPS_DIR}/math/gemmlowp/gemmlowp + ${NUTTX_APPS_DIR}/math/kissfft/kissfft ${NUTTX_APPS_DIR}/math/ruy/ruy - ${NUTTX_APPS_DIR}/math/kissfft/kissfft - ${NUTTX_APPS_DIR}/math/kissfft/kissfft ${NUTTX_APPS_DIR}/mlearning/tflite-micro/tflite-micro ${NUTTX_APPS_DIR}/system/flatbuffers/flatbuffers/include) diff --git a/mlearning/tflite-micro/Makefile b/mlearning/tflite-micro/Makefile index 179d0e9c3..7d23bc5d1 100644 --- a/mlearning/tflite-micro/Makefile +++ b/mlearning/tflite-micro/Makefile @@ -56,6 +56,7 @@ COMMON_FLAGS += -Wno-shadow COMMON_FLAGS += -Wno-sign-compare COMMON_FLAGS += -Wno-undef COMMON_FLAGS += -Wno-unused-variable +COMMON_FLAGS += -DTF_LITE_STATIC_MEMORY COMMON_FLAGS += -DTFLITE_EMULATE_FLOAT COMMON_FLAGS += -DTF_LITE_DISABLE_X86_NEON ifneq ($(CONFIG_TFLITEMICRO_DEBUG),)