diff options
author | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-05-17 11:16:22 +0100 |
---|---|---|
committer | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-05-17 11:16:22 +0100 |
commit | 8c61c0a3cb8d6b534d1e423211e06b89f45bf223 (patch) | |
tree | 24895ed5f20d4cad92fbcd679fb9637637931e18 /scripts/cmake/tensorflow.cmake | |
parent | b76b855448d58b85f53642532375c9b7808ab14d (diff) | |
download | ml-embedded-evaluation-kit-8c61c0a3cb8d6b534d1e423211e06b89f45bf223.tar.gz |
MLECO-2985 Adding Corstone-310 support
Change-Id: Ifa4b11154478355c10cb3e747b9938a74afd242b
Signed-off-by: Eanna O Cathain <eanna.ocathain@arm.com>
Diffstat (limited to 'scripts/cmake/tensorflow.cmake')
-rw-r--r-- | scripts/cmake/tensorflow.cmake | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/scripts/cmake/tensorflow.cmake b/scripts/cmake/tensorflow.cmake index 2374c86..8a790bc 100644 --- a/scripts/cmake/tensorflow.cmake +++ b/scripts/cmake/tensorflow.cmake @@ -50,7 +50,16 @@ if (TARGET_PLATFORM STREQUAL native) set(TENSORFLOW_LITE_MICRO_TARGET_ARCH x86_64) else() set(TENSORFLOW_LITE_MICRO_TARGET "cortex_m_generic") - set(TENSORFLOW_LITE_MICRO_TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}) + + if ("${CMAKE_SYSTEM_ARCH}" STREQUAL "armv8.1-m.main") + # TensorFlow's generic makefile doesn't currently have a flow for Cortex-M85. + # We build for Arm Cortex-M55 instead. + # @TODO: check with latest TensorFlow package. + set(TENSORFLOW_LITE_MICRO_TARGET_ARCH "cortex-m55") + else() + set(TENSORFLOW_LITE_MICRO_TARGET_ARCH "${CMAKE_SYSTEM_PROCESSOR}") + endif() + if(ETHOS_U_NPU_ENABLED) # Arm Ethos-U55 NPU is the co-processor for ML workload: set(TENSORFLOW_LITE_MICRO_CO_PROCESSOR "ethos_u") |