diff options
author | Måns Nilsson <mans.nilsson@arm.com> | 2020-09-09 16:56:16 +0200 |
---|---|---|
committer | Måns Nilsson <mans.nilsson@arm.com> | 2020-09-10 17:35:19 +0200 |
commit | 95e5a229331b2ff6a4a4fb92544b002720f6b034 (patch) | |
tree | 1c0026d58ad643029d3e2793a1026650d98510d6 /tensorflow.cmake | |
parent | 25480143de2066d58be5c0c305007acbcb65e75a (diff) | |
download | ethos-u-core-software-95e5a229331b2ff6a4a4fb92544b002720f6b034.tar.gz |
Update tflu_gen target
-Use real target instead of "lib".
-Remove TAGS=cmsis-nn when build for Ethos-U.
Change-Id: I8825759df9be5614ba572c2e005e7f66034937f9
Diffstat (limited to 'tensorflow.cmake')
-rw-r--r-- | tensorflow.cmake | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow.cmake b/tensorflow.cmake index 3e051f1..ee9c68a 100644 --- a/tensorflow.cmake +++ b/tensorflow.cmake @@ -39,7 +39,7 @@ endif() set(TFLU_PATH "${TENSORFLOW_PATH}/tensorflow/lite/micro") set(TFLU_GENDIR ${CMAKE_CURRENT_BINARY_DIR}/tensorflow/) -set(TFLU_TARGET "lib") +set(TFLU_TARGET "cortex_generic") 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_DEBUG_OPTIMIZATION CACHE STRING "Tensorflow Lite Micro default optimization level for debug builds") @@ -47,8 +47,6 @@ set(TFLU_RELEASE_OPTIMIZATION CACHE STRING "Tensorflow Lite Micro default optimi if(CORE_SOFTWARE_ACCELERATOR STREQUAL NPU) set(TFLU_ETHOSU_LIBS $<TARGET_FILE:ethosu_core_driver>) - # Set preference for ethos-u over cmsis-nn - list(APPEND TFLU_TAGS "cmsis-nn") list(APPEND TFLU_TAGS "ethos-u") elseif(CORE_SOFTWARE_ACCELERATOR STREQUAL CMSIS-NN) list(APPEND TFLU_TAGS "cmsis-nn") |