diff options
| author | Cisco Cervellera <cisco.cervellera@arm.com> | 2021-08-02 09:32:07 +0100 |
|---|---|---|
| committer | Isabella Gottardi <isabella.gottardi@arm.com> | 2021-08-03 16:10:16 +0100 |
| commit | f085fa5a2859178741f39563b4d9d68ca41d916a (patch) | |
| tree | 62fd8411c63a853be4c32683a3ada0b6affb0399 /scripts/cmake/tensorflow.cmake | |
| parent | 81c4c9178598e5c26af7373eca67842a8fe4179f (diff) | |
| download | ml-embedded-evaluation-kit-f085fa5a2859178741f39563b4d9d68ca41d916a.tar.gz | |
MLECO-2051: Work with Corstone-300 + Ethos-U65
Change-Id: I64ab930a1de5210f435f91bed7600a700581946f
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-on: https://eu-gerrit-2.euhpc.arm.com/c/ml/ecosystem/ml-embedded-evaluation-kit/+/459126
Tested-by: mlecosys <mlecosys@arm.com>
Tested-by: George Gekov <george.gekov@arm.com>
Reviewed-by: George Gekov <george.gekov@arm.com>
Reviewed-by: Kshitij Sisodia <kshitij.sisodia@arm.com>
Diffstat (limited to 'scripts/cmake/tensorflow.cmake')
| -rw-r--r-- | scripts/cmake/tensorflow.cmake | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/scripts/cmake/tensorflow.cmake b/scripts/cmake/tensorflow.cmake index 1123c7f..d0654b8 100644 --- a/scripts/cmake/tensorflow.cmake +++ b/scripts/cmake/tensorflow.cmake @@ -69,7 +69,7 @@ if (TARGET_PLATFORM STREQUAL native) else() set(TENSORFLOW_LITE_MICRO_TARGET "cortex_m_ethos_eval") set(TENSORFLOW_LITE_MICRO_TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}) - if(ETHOS_U55_ENABLED) + if(ETHOS_U_NPU_ENABLED) # Arm Ethos-U55 NPU is the co-processor for ML workload: set(TENSORFLOW_LITE_MICRO_CO_PROCESSOR "ethos_u") endif() @@ -104,7 +104,7 @@ add_custom_target(tensorflow_build ALL TARGET=${TENSORFLOW_LITE_MICRO_TARGET} TARGET_ARCH=${TENSORFLOW_LITE_MICRO_TARGET_ARCH} BUILD_TYPE=${TENSORFLOW_LITE_MICRO_BUILD_TYPE} - ETHOSU_DRIVER_PATH=${ETHOS_U55_DRIVER_SRC_PATH} + ETHOSU_DRIVER_PATH=${ETHOS_U_NPU_DRIVER_SRC_PATH} CMSIS_PATH=${CMSIS_SRC_PATH} # Conditional arguments |