diff options
Diffstat (limited to 'scripts/cmake')
-rw-r--r-- | scripts/cmake/common_user_options.cmake | 2 | ||||
-rw-r--r-- | scripts/cmake/tensorflow.cmake | 8 |
2 files changed, 3 insertions, 7 deletions
diff --git a/scripts/cmake/common_user_options.cmake b/scripts/cmake/common_user_options.cmake index a077264..ae35925 100644 --- a/scripts/cmake/common_user_options.cmake +++ b/scripts/cmake/common_user_options.cmake @@ -88,7 +88,7 @@ if (NOT TARGET_PLATFORM STREQUAL native) if (ETHOS_U_NPU_ENABLED) USER_OPTION(ETHOS_U_NPU_TIMING_ADAPTER_SRC_PATH "Path to Ethos-U NPU timing adapter sources" - "${DEPENDENCY_ROOT_DIR}/core-software/drivers/timing_adapter" + "${DEPENDENCY_ROOT_DIR}/core-platform/drivers/timing_adapter" PATH ) diff --git a/scripts/cmake/tensorflow.cmake b/scripts/cmake/tensorflow.cmake index 284c70a..127724b 100644 --- a/scripts/cmake/tensorflow.cmake +++ b/scripts/cmake/tensorflow.cmake @@ -54,7 +54,7 @@ else() if(ETHOS_U_NPU_ENABLED) # Arm Ethos-U55 NPU is the co-processor for ML workload: set(TENSORFLOW_LITE_MICRO_CO_PROCESSOR "ethos_u") - string(TOLOWER ${ETHOS_U_NPU_ID} ETHOSU_ARCH) + set(ETHOS_U_NPU_ID "u55") # Currently only u55 is supported by TFLite Micro. endif() set(TENSORFLOW_LITE_MICRO_OPTIMIZED_KERNEL "cmsis_nn") @@ -68,10 +68,6 @@ if (TENSORFLOW_LITE_MICRO_CLEAN_BUILD) list(APPEND MAKE_TARGETS_LIST "clean") endif() -if (ETHOS_U_NPU_ID) - string(TOLOWER ${ETHOS_U_NPU_ID} ETHOSU_ARCH) -endif() - # Primary target list(APPEND MAKE_TARGETS_LIST "microlite") message(STATUS "TensorFlow Lite Micro build to be called for these targets: ${MAKE_TARGETS_LIST}") @@ -89,7 +85,7 @@ add_custom_target(tensorflow_build ALL BUILD_TYPE=${TENSORFLOW_LITE_MICRO_BUILD_TYPE} CMSIS_PATH=${CMSIS_SRC_PATH} # Conditional arguments - $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_ARCH=${ETHOSU_ARCH}> + $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_ARCH=${ETHOS_U_NPU_ID}> $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_DRIVER_PATH=${ETHOS_U_NPU_DRIVER_SRC_PATH}> $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_DRIVER_LIBS=$<TARGET_FILE:ethosu_core_driver>> |