diff options
author | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2021-11-24 14:08:10 +0100 |
---|---|---|
committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2021-11-24 14:08:10 +0100 |
commit | 02856bdd80f0ed52d09dd2fea450e61eb02372d6 (patch) | |
tree | f3ac89c62771de7c4caea18a1e94224191445655 /tflite_micro.cmake | |
parent | eb912395c25fbdeef4f322d7aea9226578228184 (diff) | |
download | ethos-u-core-software-02856bdd80f0ed52d09dd2fea450e61eb02372d6.tar.gz |
When CORE_SOFTWARE_ACCELERATOR is set to CPU, then the TFLu reference
kernels should be used rather than the CMSIS-NN optimized kernels.
Change-Id: I09418b2e32d8c1654892bc5caec9e5b80ccd6c9f
Diffstat (limited to 'tflite_micro.cmake')
-rw-r--r-- | tflite_micro.cmake | 24 |
1 files changed, 13 insertions, 11 deletions
diff --git a/tflite_micro.cmake b/tflite_micro.cmake index 7d5291c..e01a423 100644 --- a/tflite_micro.cmake +++ b/tflite_micro.cmake @@ -120,22 +120,24 @@ target_include_directories(tflu PUBLIC # CMSIS-NN ############################################################################# -add_subdirectory(${CMSIS_PATH}/CMSIS/NN cmsis_nn) +if (NOT ${CORE_SOFTWARE_ACCELERATOR} STREQUAL "CPU") + add_subdirectory(${CMSIS_PATH}/CMSIS/NN cmsis_nn) -target_compile_options(cmsis-nn PRIVATE - ${TFLU_OPTIMIZATION_LEVEL}) + target_compile_options(cmsis-nn PRIVATE + ${TFLU_OPTIMIZATION_LEVEL}) -tensorflow_target_sources_glob(tflu GLOB TRUE - ${TFLU_PATH}/kernels/cmsis_nn/*.cc) + tensorflow_target_sources_glob(tflu GLOB TRUE + ${TFLU_PATH}/kernels/cmsis_nn/*.cc) -target_include_directories(tflu PUBLIC - ${CMSIS_PATH}) + target_include_directories(tflu PUBLIC + ${CMSIS_PATH}) -target_compile_definitions(tflu PUBLIC - CMSIS_NN) + target_compile_definitions(tflu PUBLIC + CMSIS_NN) -target_link_libraries(tflu PUBLIC - cmsis-nn) + target_link_libraries(tflu PUBLIC + cmsis-nn) +endif() ############################################################################# # Ethos-U |