diff options
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) |