diff --git a/mlearning/cmsis-nn/CMakeLists.txt b/mlearning/cmsis-nn/CMakeLists.txt index c5944620d..43cbcfe89 100644 --- a/mlearning/cmsis-nn/CMakeLists.txt +++ b/mlearning/cmsis-nn/CMakeLists.txt @@ -69,6 +69,8 @@ if(CONFIG_MLEARNING_CMSIS_NN) ${CMSIS_NN_DIR}/Source/LSTMFunctions/*.c ${CMSIS_NN_DIR}/Source/ReshapeFunctions/*.c) + set(CMSIS_NN_INCDIR ${CMAKE_CURRENT_LIST_DIR}/cmsis-nn/Include) + if(CONFIG_ARM_NEON) list( REMOVE_ITEM @@ -86,5 +88,6 @@ if(CONFIG_MLEARNING_CMSIS_NN) nuttx_add_library(cmsis_nn STATIC) target_compile_options(cmsis_nn PRIVATE ${CFLAGS}) target_sources(cmsis_nn PRIVATE ${CMSIS_NN_SRCS}) + target_include_directories(cmsis_nn PUBLIC ${CMSIS_NN_INCDIR}) endif() diff --git a/mlearning/tflite-micro/CMakeLists.txt b/mlearning/tflite-micro/CMakeLists.txt index 482e920b7..abc9b9fa5 100644 --- a/mlearning/tflite-micro/CMakeLists.txt +++ b/mlearning/tflite-micro/CMakeLists.txt @@ -152,7 +152,7 @@ if(CONFIG_TFLITEMICRO) ${NUTTX_APPS_DIR}/system/flatbuffers/flatbuffers/include) if(CONFIG_MLEARNING_CMSIS_NN) - list(APPEND INCDIR ${NUTTX_APPS_DIR}/apps/mlearning/cmsis-nn/cmsis-nn) + list(APPEND INCDIR ${NUTTX_APPS_DIR}/mlearning/cmsis-nn/cmsis-nn) endif() # ############################################################################