diff options
author | Alex Tawse <alex.tawse@arm.com> | 2024-06-25 09:39:36 +0100 |
---|---|---|
committer | Alex Tawse <alex.tawse@arm.com> | 2024-06-26 16:28:52 +0100 |
commit | 4b10eef739b40c87d373b816d22fa847cf5c6fdf (patch) | |
tree | 8e7fb6bcc447c9b3014e02b0b78a468b01e19627 | |
parent | 865ce919755f8a7c3410f8d6a044aad33fbd1bce (diff) | |
download | ml-embedded-evaluation-kit-4b10eef739b40c87d373b816d22fa847cf5c6fdf.tar.gz |
MLECO-5087: Set tflite-micro default NPU to U55
* Use 'u55' as the default NPU ID for tflite-micro
if other NPU values don't match
Change-Id: I6826acf7f33379e8c1edc591cd6d621234a87099
Signed-off-by: Alex Tawse <alex.tawse@arm.com>
-rw-r--r-- | scripts/cmake/tensorflow_lite_micro.cmake | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/scripts/cmake/tensorflow_lite_micro.cmake b/scripts/cmake/tensorflow_lite_micro.cmake index 79e3e07..4a988aa 100644 --- a/scripts/cmake/tensorflow_lite_micro.cmake +++ b/scripts/cmake/tensorflow_lite_micro.cmake @@ -75,7 +75,11 @@ else() if(ETHOS_U_NPU_ENABLED) # Arm Ethos-U55 NPU is the co-processor for ML workload: set(TENSORFLOW_LITE_MICRO_CO_PROCESSOR "ethos_u") - string(TOLOWER ${ETHOS_U_NPU_ID} TENSORFLOW_LITE_MICRO_CO_PROCESSOR_ARCH) + if(${ETHOS_U_NPU_ID} STREQUAL "U65") + set(TENSORFLOW_LITE_MICRO_CO_PROCESSOR_ARCH "u65") + else() + set(TENSORFLOW_LITE_MICRO_CO_PROCESSOR_ARCH "u55") + endif () endif() set(TENSORFLOW_LITE_MICRO_OPTIMIZED_KERNEL "cmsis_nn") |