aboutsummaryrefslogtreecommitdiff
path: root/applications/message_handler/test/CMakeLists.txt
diff options
context:
space:
mode:
authorDavide Grohmann <davide.grohmann@arm.com>2022-06-15 11:23:25 +0200
committerKristofer Jonsson <kristofer.jonsson@arm.com>2022-08-25 09:13:36 +0000
commitb35f0c681a1153e935bb8e40cd2cca2b04c7b5c0 (patch)
tree3970df6a0ef9f3af0a31ff3e911b318ab3e8f8ac /applications/message_handler/test/CMakeLists.txt
parentf4379e99431e5b027b0d5942ccff56af6bfe82b1 (diff)
downloadethos-u-core-platform-b35f0c681a1153e935bb8e40cd2cca2b04c7b5c0.tar.gz
Add message_handler tests for inference cancellation
Change-Id: Ifdacc47024250e34549d45377795501c371c69f5
Diffstat (limited to 'applications/message_handler/test/CMakeLists.txt')
-rw-r--r--applications/message_handler/test/CMakeLists.txt53
1 files changed, 29 insertions, 24 deletions
diff --git a/applications/message_handler/test/CMakeLists.txt b/applications/message_handler/test/CMakeLists.txt
index 44fd471..7cba4c3 100644
--- a/applications/message_handler/test/CMakeLists.txt
+++ b/applications/message_handler/test/CMakeLists.txt
@@ -21,30 +21,35 @@ set(TEST_MESSAGE_HANDLER_MODEL_1 "" CACHE STRING "Path to built in model 1")
set(TEST_MESSAGE_HANDLER_MODEL_2 "" CACHE STRING "Path to built in model 2")
set(TEST_MESSAGE_HANDLER_MODEL_3 "" CACHE STRING "Path to built in model 3")
-if(TARGET ethosu_core_driver)
- file(GLOB models LIST_DIRECTORIES true "${CMAKE_CURRENT_SOURCE_DIR}/../../baremetal/models/${ETHOSU_TARGET_NPU_CONFIG}/*")
-endif()
+function(ethosu_add_message_handler_test testname)
+ if(TARGET ethosu_core_driver)
+ file(GLOB models LIST_DIRECTORIES true "${CMAKE_CURRENT_SOURCE_DIR}/../../baremetal/models/${ETHOSU_TARGET_NPU_CONFIG}/*")
+ endif()
-foreach(model ${models})
- get_filename_component(modelname ${model} NAME)
- ethosu_add_executable_test(message_handler_test_${modelname}
- SOURCES
- main.cpp
- message_client.cpp
- LIBRARIES
- message_handler_lib
- freertos_kernel
- ethosu_mhu_dummy)
+ foreach(model ${models})
+ get_filename_component(modelname ${model} NAME)
+ ethosu_add_executable_test(mh_${testname}_${modelname}
+ SOURCES
+ ${testname}.cpp
+ message_client.cpp
+ LIBRARIES
+ message_handler_lib
+ freertos_kernel
+ ethosu_mhu_dummy)
- target_include_directories(message_handler_test_${modelname} PRIVATE
- ../indexed_networks
- ${model}
- ${LINUX_DRIVER_STACK_PATH}/kernel)
+ target_include_directories(mh_${testname}_${modelname} PRIVATE
+ ../indexed_networks
+ ${model}
+ ${LINUX_DRIVER_STACK_PATH}/kernel)
- target_compile_definitions(message_handler_test_${modelname} PRIVATE
- TENSOR_ARENA_SIZE=${MESSAGE_HANDLER_ARENA_SIZE}
- $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_0}>:MODEL_0=${TEST_MESSAGE_HANDLER_MODEL_0}>
- $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_1}>:MODEL_1=${TEST_MESSAGE_HANDLER_MODEL_1}>
- $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_2}>:MODEL_2=${TEST_MESSAGE_HANDLER_MODEL_2}>
- $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_3}>:MODEL_3=${TEST_MESSAGE_HANDLER_MODEL_3}>)
-endforeach()
+ target_compile_definitions(mh_${testname}_${modelname} PRIVATE
+ TENSOR_ARENA_SIZE=${MESSAGE_HANDLER_ARENA_SIZE}
+ $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_0}>:MODEL_0=${TEST_MESSAGE_HANDLER_MODEL_0}>
+ $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_1}>:MODEL_1=${TEST_MESSAGE_HANDLER_MODEL_1}>
+ $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_2}>:MODEL_2=${TEST_MESSAGE_HANDLER_MODEL_2}>
+ $<$<BOOL:${TEST_MESSAGE_HANDLER_MODEL_3}>:MODEL_3=${TEST_MESSAGE_HANDLER_MODEL_3}>)
+ endforeach()
+endfunction()
+
+ethosu_add_message_handler_test(run_inference_test)
+ethosu_add_message_handler_test(cancel_reject_inference_test)