From 7e0775474aa06ecd264c5cec299f73e66c37af9e Mon Sep 17 00:00:00 2001
From: Henrik Hoglind <henrik.hoglind@arm.com>
Date: Fri, 21 Aug 2020 15:19:11 +0200
Subject: Add options to Tensorflow for compiler optimization level

Change-Id: I2c9b685b69dc3068b92d07a1c7840b0906954582
---
 tensorflow.cmake | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tensorflow.cmake b/tensorflow.cmake
index 9396f1d..f806ee4 100644
--- a/tensorflow.cmake
+++ b/tensorflow.cmake
@@ -43,6 +43,8 @@ set(TFLU_TARGET "lib")
 set(TFLU_TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES})
 set(TFLU_BUILD_TYPE "release" CACHE STRING "Tensorflow Lite Mirco build type, can be release or debug")
 set(TFLU_FAST_MEMORY_SIZE 0 CACHE STRING "Size of relocated fast memory scratch tensor")
+set(TFLU_DEBUG_OPTIMIZATION CACHE STRING "Tensorflow Lite Micro default optimization level for debug builds")
+set(TFLU_RELEASE_OPTIMIZATION CACHE STRING "Tensorflow Lite Micro default optimization level for debug builds")
 
 if(CORE_SOFTWARE_ACCELERATOR STREQUAL NPU)
     set(TFLU_ETHOSU_LIBS $<TARGET_FILE:ethosu_core_driver>)
@@ -57,7 +59,7 @@ string(JOIN TFLU_TAGS " " TFLU_TAGS)
 
 # Command and target
 add_custom_target(tflu_gen ALL
-                  COMMAND make -j${J} -f ${TFLU_PATH}/tools/make/Makefile microlite TARGET=${TFLU_TARGET} TARGET_ARCH=${TFLU_TARGET_ARCH} CC_TOOL=${TFLU_CC} CXX_TOOL=${TFLU_CXX} AR_TOOL=${TFLU_AR} GENDIR=${TFLU_GENDIR} CMSIS_PATH=${CMSIS_PATH} ETHOSU_DRIVER_PATH=${CORE_DRIVER_PATH} ETHOSU_DRIVER_LIBS=${TFLU_ETHOSU_LIBS} ETHOSU_FAST_MEMORY_SIZE=${TFLU_FAST_MEMORY_SIZE} TAGS="${TFLU_TAGS}" BUILD_TYPE=${TFLU_BUILD_TYPE}
+                  COMMAND make -j${J} -f ${TFLU_PATH}/tools/make/Makefile microlite TARGET=${TFLU_TARGET} TARGET_ARCH=${TFLU_TARGET_ARCH} CC_TOOL=${TFLU_CC} CXX_TOOL=${TFLU_CXX} AR_TOOL=${TFLU_AR} GENDIR=${TFLU_GENDIR} CMSIS_PATH=${CMSIS_PATH} ETHOSU_DRIVER_PATH=${CORE_DRIVER_PATH} ETHOSU_DRIVER_LIBS=${TFLU_ETHOSU_LIBS} ETHOSU_FAST_MEMORY_SIZE=${TFLU_FAST_MEMORY_SIZE} TAGS="${TFLU_TAGS}" BUILD_TYPE=${TFLU_BUILD_TYPE}  $<$<BOOL:${TFLU_DEBUG_OPTIMIZATION}>:DEBUG_DEFAULT_OPTIMIZATION=${TFLU_DEBUG_OPTIMIZATION}> $<$<BOOL:${TFLU_RELEASE_OPTIMIZATION}>:RELEASE_DEFAULT_OPTIMIZATION=${TFLU_RELEASE_OPTIMIZATION}>
                   WORKING_DIRECTORY ${TENSORFLOW_PATH})
 
 # Create library and link library to custom target
-- 
cgit v1.2.1