diff options
| -rw-r--r-- | tensorflow.cmake | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow.cmake b/tensorflow.cmake index 355a673..d304a84 100644 --- a/tensorflow.cmake +++ b/tensorflow.cmake @@ -30,14 +30,14 @@ set(TFLU_TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}) set(TFLU_ETHOSU_LIBS $<TARGET_FILE:ethosu_core_driver>) if(CORE_SOFTWARE_BACKEND STREQUAL NPU) - list(APPEND TFLU_TAGS "ARM_NPU") + list(APPEND TFLU_TAGS "ethos-u") endif() 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} ARM_NPU_PATH=${CORE_DRIVER_PATH} ETHOSU_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}" WORKING_DIRECTORY ${TENSORFLOW_PATH}) # Create library and link library to custom target |