TFLu reference kernels
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
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 @@
# 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