Add cortex300 download for aarch64 linux host (tensorflow#2573)

* Updates x86 corstone300 version BUG=Corstone300 tests were not able to be ran on aarch64 hosts Author: Ryan O'Shea <[email protected]>
plumerai · May 16, 2024 · 1973d40 · 1973d40
1 parent b281a0d
commit 1973d40
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/tensorflow/lite/micro/tools/make/corstone_300_download.sh b/tensorflow/lite/micro/tools/make/corstone_300_download.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -46,10 +46,19 @@ DOWNLOADED_CORSTONE_PATH=${DOWNLOADS_DIR}/corstone300
 if [ -d ${DOWNLOADED_CORSTONE_PATH} ]; then
   echo >&2 "${DOWNLOADED_CORSTONE_PATH} already exists, skipping the download."
 else
-  UNAME_S=`uname -s`
+  UNAME_S=$(uname -s)
+  UNAME_M=$(uname -m)
   if [ ${UNAME_S} == Linux ]; then
-    CORSTONE_URL=https://developer.arm.com/-/media/Arm%20Developer%20Community/Downloads/OSS/FVP/Corstone-300/FVP_Corstone_SSE-300_11.16_26.tgz
-    EXPECTED_MD5=29d9208127b24a0d83356efb8343162d
+    if [ ${UNAME_M} == x86_64 ]; then
+      CORSTONE_URL=https://developer.arm.com/-/media/Arm%20Developer%20Community/Downloads/OSS/FVP/Corstone-300/FVP_Corstone_SSE-300_11.24_13_Linux64.tgz
+      EXPECTED_MD5=42500e49a4b9e8e0f633d1bad9b7c052
+    elif [ ${UNAME_M} == aarch64 ]; then
+      CORSTONE_URL=https://developer.arm.com/-/media/Arm%20Developer%20Community/Downloads/OSS/FVP/Corstone-300/FVP_Corstone_SSE-300_11.24_13_Linux64_armv8l.tgz
+      EXPECTED_MD5=89904e875c863235635e1570c4f6459e
+    else
+          echo "Cpu type ${UNAME_M} with OS type ${UNAME_S} not supported."
+          exit 1
+    fi
   else
     echo "OS type ${UNAME_S} not supported."
     exit 1