diff options
Diffstat (limited to 'applications/message_handler/CMakeLists.txt')
| -rw-r--r-- | applications/message_handler/CMakeLists.txt | 38 |
1 files changed, 18 insertions, 20 deletions
diff --git a/applications/message_handler/CMakeLists.txt b/applications/message_handler/CMakeLists.txt index 040b27d..8d4ef54 100644 --- a/applications/message_handler/CMakeLists.txt +++ b/applications/message_handler/CMakeLists.txt @@ -21,36 +21,34 @@ if (NOT TARGET freertos_kernel) return() endif() +# Split total tensor arena equally for each NPU +if (TARGET ethosu_core_driver AND ETHOSU_TARGET_NPU_COUNT GREATER 0) + set(NUM_ARENAS ${ETHOSU_TARGET_NPU_COUNT}) +else() + set(NUM_ARENAS 1) +endif() + set(MESSAGE_HANDLER_ARENA_SIZE 2000000 CACHE STRING "Total size of all message handler tensor arenas") -set(MESSAGE_HANDLER_MODEL_0 FALSE CACHE STRING "Path to built in model 0") -set(MESSAGE_HANDLER_MODEL_1 FALSE CACHE STRING "Path to built in model 1") -set(MESSAGE_HANDLER_MODEL_2 FALSE CACHE STRING "Path to built in model 2") -set(MESSAGE_HANDLER_MODEL_3 FALSE CACHE STRING "Path to built in model 3") +math(EXPR TENSOR_ARENA_SIZE "${MESSAGE_HANDLER_ARENA_SIZE} / ${NUM_ARENAS}") + +add_subdirectory(lib) + +set(MESSAGE_HANDLER_MODEL_0 "" CACHE STRING "Path to built in model 0") +set(MESSAGE_HANDLER_MODEL_1 "" CACHE STRING "Path to built in model 1") +set(MESSAGE_HANDLER_MODEL_2 "" CACHE STRING "Path to built in model 2") +set(MESSAGE_HANDLER_MODEL_3 "" CACHE STRING "Path to built in model 3") ethosu_add_executable(message_handler SOURCES main.cpp - message_handler.cpp - message_queue.cpp LIBRARIES - cmsis_device - $<$<TARGET_EXISTS:ethosu_core_driver>:ethosu_core_driver> - ethosu_mailbox - freertos_kernel - inference_process) + message_handler_lib + freertos_kernel) target_include_directories(message_handler PRIVATE + indexed_networks ${LINUX_DRIVER_STACK_PATH}/kernel) -# Split total tensor arena equally for each NPU -if (TARGET ethosu_core_driver AND ETHOSU_TARGET_NPU_COUNT GREATER 0) - set(NUM_ARENAS ${ETHOSU_TARGET_NPU_COUNT}) -else() - set(NUM_ARENAS 1) -endif() - -math(EXPR TENSOR_ARENA_SIZE "${MESSAGE_HANDLER_ARENA_SIZE} / ${NUM_ARENAS}") - target_compile_definitions(message_handler PRIVATE TENSOR_ARENA_SIZE=${TENSOR_ARENA_SIZE} $<$<BOOL:${MESSAGE_HANDLER_MODEL_0}>:MODEL_0=${MESSAGE_HANDLER_MODEL_0}> |