diff options
author | Henrik Hoglind <henrik.hoglind@arm.com> | 2020-05-28 12:09:04 +0200 |
---|---|---|
committer | Bhavik Patel <bhavik.patel@arm.com> | 2020-07-09 08:24:04 +0200 |
commit | 3598ef88a7a9729b5842818bfa77e28a599ea695 (patch) | |
tree | fd847e83ed3de951788a5b411ce69e2bcaaa1870 /tensorflow.cmake | |
parent | 94ca3558d648c69ee1b5d7771247763a9fdd9514 (diff) | |
download | ethos-u-core-software-3598ef88a7a9729b5842818bfa77e28a599ea695.tar.gz |
Add TFLU_BUILD_TYPE in tensorflow.cmake
The build type for the Tensorflow Lite Micro library can now
controlled with TFLU_BUILD_TYPE, valid values are "release"
and "debug"
Change-Id: I4ee5fa431dd433aaf2989bd91f1521631632364f
Diffstat (limited to 'tensorflow.cmake')
-rw-r--r-- | tensorflow.cmake | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow.cmake b/tensorflow.cmake index d252252..f8b9ea1 100644 --- a/tensorflow.cmake +++ b/tensorflow.cmake @@ -35,6 +35,8 @@ set(TFLU_PATH "${TENSORFLOW_PATH}/tensorflow/lite/micro") set(TFLU_GENDIR ${CMAKE_CURRENT_BINARY_DIR}/tensorflow/) set(TFLU_TARGET "lib") set(TFLU_TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}) +set(TFLU_ETHOSU_LIBS $<TARGET_FILE:ethosu_core_driver>) +set(TFLU_BUILD_TYPE "release" CACHE STRING "Tensorflow Lite Mirco build type, can be release or debug") if(CORE_SOFTWARE_ACCELERATOR STREQUAL NPU) set(TFLU_ETHOSU_LIBS $<TARGET_FILE:ethosu_core_driver>) @@ -49,7 +51,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} TAGS="${TFLU_TAGS}" + 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} WORKING_DIRECTORY ${TENSORFLOW_PATH}) # Create library and link library to custom target |