diff options
author | Davide Grohmann <davide.grohmann@arm.com> | 2022-06-15 11:23:25 +0200 |
---|---|---|
committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2022-08-25 09:13:36 +0000 |
commit | b35f0c681a1153e935bb8e40cd2cca2b04c7b5c0 (patch) | |
tree | 3970df6a0ef9f3af0a31ff3e911b318ab3e8f8ac /applications/message_handler/test/CMakeLists.txt | |
parent | f4379e99431e5b027b0d5942ccff56af6bfe82b1 (diff) | |
download | ethos-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.txt | 53 |
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) |