From 02856bdd80f0ed52d09dd2fea450e61eb02372d6 Mon Sep 17 00:00:00 2001
From: Kristofer Jonsson <kristofer.jonsson@arm.com>
Date: Wed, 24 Nov 2021 14:08:10 +0100
Subject: 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
---
 tflite_micro.cmake | 24 +++++++++++++-----------
 1 file 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
-- 
cgit v1.2.1