commit 68013163b2633114de71c76fd2b1c5c9fc13571c
author Kristofer Jonsson <kristofer.jonsson@arm.com> Fri May 29 13:04:08 2020 +0200
committer Kristofer Jonsson <kristofer.jonsson@arm.com> Fri Aug 21 17:04:27 2020 +0200
tree 8fc9d2b9f78ce49caed66f026f0b98a155c0bafe
parent 6f4cb2a73d97cb4fae399d72a563af92e1b1a534

Vela fast memory scratch buffer

Change-Id: I42f7a2e122ddd5826d4e0d83fccacc3865045b03

tensorflow.cmake

diff --git a/tensorflow.cmake b/tensorflow.cmake
index 691ee53..9396f1d 100644
--- a/tensorflow.cmake
+++ b/tensorflow.cmake
@@ -42,6 +42,7 @@
 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")
 
 if(CORE_SOFTWARE_ACCELERATOR STREQUAL NPU)
     set(TFLU_ETHOSU_LIBS $<TARGET_FILE:ethosu_core_driver>)
@@ -56,7 +57,7 @@
 
 # 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} 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}
                   WORKING_DIRECTORY ${TENSORFLOW_PATH})
 
 # Create library and link library to custom target