From a582ed5062c5a89008996fccc0be8dbed990d557 Mon Sep 17 00:00:00 2001
From: Luke Hutton <luke.hutton@arm.com>
Date: Thu, 21 Mar 2024 13:57:53 +0100
Subject: [PATCH] [SME][Docker] Add Fixed Virtual Platform (FVP) and toolchain
 install (#16755)

This commit adds the installation of the AArch64 Architecture Envelope
Model (AEM) Fixed Virtual Platform (FVP) which can be used to test SME
code generation functional correctness.

It also adds the installation of a baremetal toolchain which can be used
to for compiling functions to run on the FVP.

Change-Id: If13d0cb07855ecf8c9e1c8cd0496c54678335d30
---
 docker/Dockerfile.ci_cpu                      |  5 ++
 docker/install/ubuntu_install_aprofile_aem.sh | 54 +++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100755 docker/install/ubuntu_install_aprofile_aem.sh

diff --git a/docker/Dockerfile.ci_cpu b/docker/Dockerfile.ci_cpu
index 208a0f272a008..ae088f5c9e634 100644
--- a/docker/Dockerfile.ci_cpu
+++ b/docker/Dockerfile.ci_cpu
@@ -149,3 +149,8 @@ RUN bash /install/ubuntu_install_libxsmm.sh
 # ONNX and PyTorch
 COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
 RUN bash /install/ubuntu_install_onnx.sh
+
+# AArch64 Architecture Envelope Model (AEM)
+COPY install/ubuntu_install_aprofile_aem.sh /install
+RUN bash /install/ubuntu_install_aprofile_aem.sh
+ENV PATH $PATH:/opt/arm/fvp/Base_RevC_AEMvA_pkg/models/Linux64_GCC-9.3/:/opt/arm/gcc-aarch64-none-elf/bin
diff --git a/docker/install/ubuntu_install_aprofile_aem.sh b/docker/install/ubuntu_install_aprofile_aem.sh
new file mode 100755
index 0000000000000..4288cded0b557
--- /dev/null
+++ b/docker/install/ubuntu_install_aprofile_aem.sh
@@ -0,0 +1,54 @@
+#!/bin/bash
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# Install the AArch64 Architecture Envelope Model (AEM)
+
+set -e
+set -u
+set -o pipefail
+
+tmpdir=$(mktemp -d)
+
+cleanup()
+{
+    rm -rf "$tmpdir"
+}
+
+trap cleanup 0
+
+pushd "$tmpdir"
+
+# Install GCC toolchain
+gcc_install_dir="/opt/arm/gcc-aarch64-none-elf"
+gcc_url="https://developer.arm.com/-/media/Files/downloads/gnu/13.2.rel1/binrel/arm-gnu-toolchain-13.2.rel1-x86_64-aarch64-none-elf.tar.xz?rev=28d5199f6db34e5980aae1062e5a6703&hash=D87D4B558F0A2247B255BA15C32A94A9F354E6A8"
+gcc_sha="7fe7b8548258f079d6ce9be9144d2a10bd2bf93b551dafbf20fe7f2e44e014b8"
+gcc_tar="arm-gnu-toolchain-13.2.rel1-x86_64-aarch64-none-linux-gnu.tar.xz"
+mkdir -p $gcc_install_dir
+curl --retry 64 -sSL $gcc_url -o $gcc_tar
+echo "$gcc_sha $gcc_tar" | sha256sum --check
+tar -xf $gcc_tar -C $gcc_install_dir --strip-components=1
+
+# Download FVP
+fvp_dir="/opt/arm/fvp"
+fvp_url="https://developer.arm.com/-/media/Files/downloads/ecosystem-models/FVP_Base_RevC-2xAEMvA_11.24_11_Linux64.tgz"
+fvp_sha="0f132334834cbc66889a62dd72057c976d7c7dfcfeec21799e9c78fb2ce24720"
+curl --retry 64 -sSL $fvp_url -o fvp.tgz
+echo "$fvp_sha fvp.tgz" | sha256sum --check
+mkdir -p "$fvp_dir"
+tar -xzf fvp.tgz -C "$fvp_dir"
+rm -rf doc  # Remove some documentation bundled with the package