From 4c94b30eb19f168a5e81e27a26c8e25bdab005b1 Mon Sep 17 00:00:00 2001 From: Kristofer Jonsson Date: Fri, 6 Nov 2020 10:33:21 +0100 Subject: Fast memory error handling Return error if the fast memory area would be smaller than the spilling tensor. Change-Id: Ica89a3a933a3317a346e3479f5209ddc735d1c91 --- src/ethosu_driver.c | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/src/ethosu_driver.c b/src/ethosu_driver.c index 77b8727..d21e0a7 100644 --- a/src/ethosu_driver.c +++ b/src/ethosu_driver.c @@ -337,14 +337,13 @@ int ethosu_invoke_v2(const void *custom_data_ptr, if (base_addr_size != NULL && base_addr_size[FAST_MEMORY_BASE_ADDR_INDEX] > ethosu_drv.fast_memory_size) { - LOG_WARN("Fast memory area too small. fast_memory_size=%u, base_addr_size=%u\n", - ethosu_drv.fast_memory_size, - base_addr_size[FAST_MEMORY_BASE_ADDR_INDEX]); - } - else - { - *fast_memory = ethosu_drv.fast_memory; + LOG_ERR("Fast memory area too small. fast_memory_size=%u, base_addr_size=%u\n", + ethosu_drv.fast_memory_size, + base_addr_size[FAST_MEMORY_BASE_ADDR_INDEX]); + return -1; } + + *fast_memory = ethosu_drv.fast_memory; } if (ethosu_drv.dev.reset != ethosu_read_reg(ðosu_drv.dev, NPU_REG_PROT)) -- cgit v1.2.1