[CI] Update ci_minimal docker image to cross-compile TVM to aarch64 (#…

…13776)

This PR is a prerequisite to #13714 and needs to be merged before. It contains the changes to the ci_minimal docker image to support minimal cross-compilation of TVM to aarch64.

docker/Dockerfile.ci_minimal

@@ -28,6 +28,10 @@ RUN bash /install/ubuntu_setup_tz.sh
 COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh
 RUN bash /install/ubuntu_install_core.sh
 
+# Install libraries for cross-compiling TVM to Aarch64
+COPY install/ubuntu1804_install_aarch64_cross_compile.sh /install/ubuntu1804_install_aarch64_cross_compile.sh
+RUN bash /install/ubuntu1804_install_aarch64_cross_compile.sh
+
 COPY install/ubuntu_install_cmake_source.sh /install/ubuntu_install_cmake_source.sh
 RUN bash /install/ubuntu_install_cmake_source.sh
 
@@ -47,6 +51,10 @@ RUN bash /install/ubuntu_install_python_package.sh
 COPY install/ubuntu1804_manual_install_llvm.sh /install/ubuntu1804_manual_install_llvm.sh
 RUN bash /install/ubuntu1804_manual_install_llvm.sh
 
+# Cross build LLVM to Aarch64
+COPY install/ubuntu1804_manual_install_llvm_cross_aarch64.sh /install/ubuntu1804_manual_install_llvm_cross_aarch64.sh
+RUN bash /install/ubuntu1804_manual_install_llvm_cross_aarch64.sh
+
 # Rust env (build early; takes a while)
 COPY install/ubuntu_install_rust.sh /install/ubuntu_install_rust.sh
 RUN bash /install/ubuntu_install_rust.sh

docker/install/ubuntu1804_install_aarch64_cross_compile.sh

@@ -0,0 +1,51 @@
+#!/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.
+
+set -e
+set -u
+# Used for debugging RVM build
+set -x
+set -o pipefail
+
+architecture_type=$(uname -i)
+if [ "$architecture_type" != "aarch64" ]; then
+  # Install gcc and g++ for cross-compiling c++ on ubuntu
+  apt-get update && apt-install-and-clear -y --no-install-recommends \
+      g++-aarch64-linux-gnu \
+      gcc-aarch64-linux-gnu \
+
+  # Add Aarch64 packages to the apt sources list
+  echo >> /etc/apt/sources.list.d/arm64.list
+  echo "deb [arch=arm64] http://ports.ubuntu.com/ bionic main restricted" >> /etc/apt/sources.list.d/arm64.list
+  echo "deb [arch=arm64] http://ports.ubuntu.com/ bionic-updates main restricted" >> /etc/apt/sources.list.d/arm64.list
+  echo "deb [arch=arm64] http://ports.ubuntu.com/ bionic universe" >> /etc/apt/sources.list.d/arm64.list
+  echo "deb [arch=arm64] http://ports.ubuntu.com/ bionic-updates universe" >> /etc/apt/sources.list.d/arm64.list
+  echo "deb [arch=arm64] http://ports.ubuntu.com/ bionic multiverse" >> /etc/apt/sources.list.d/arm64.list
+  echo "deb [arch=arm64] http://ports.ubuntu.com/ bionic-updates multiverse" >> /etc/apt/sources.list.d/arm64.list
+  echo "deb [arch=arm64] http://ports.ubuntu.com/ bionic-backports main restricted universe multiverse" >> /etc/apt/sources.list.d/arm64.list
+
+  # Fix apt-get update by specifying the amd64 architecture in sources.list
+  sed -i -e 's/deb /deb [arch=amd64] /g' /etc/apt/sources.list
+
+  # Install the required packages for cross-compiling
+  dpkg --add-architecture arm64
+  apt-get update && apt-install-and-clear -y --no-install-recommends \
+      zlib1g-dev:arm64 \
+      libtinfo-dev:arm64
+
+fi

docker/install/ubuntu1804_manual_install_llvm_cross_aarch64.sh

@@ -0,0 +1,65 @@
+#!/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.
+
+set -e
+set -u
+set -o pipefail
+
+architecture_type=$(uname -i)
+# Cross-build LLVM for aarch64 when not building natively.
+if [ "$architecture_type" != "aarch64" ]; then
+  git clone --depth 1 --branch release/11.x https://github.com/llvm/llvm-project.git
+  pushd llvm-project
+
+  # First build clang-tblgen and llvm-tblgen
+  mkdir build-host
+  pushd build-host
+  cmake \
+    -G Ninja \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DLLVM_ENABLE_ASSERTIONS=ON \
+    -DLLVM_ENABLE_PROJECTS="llvm;clang;clang-tools-extra" \
+    ../llvm
+  ninja clang-tblgen llvm-tblgen
+  popd
+
+  # Then cross-compile LLVM for Aarch64
+  mkdir build
+  pushd build
+  CC=aarch64-linux-gnu-gcc CXX=aarch64-linux-gnu-g++ cmake \
+    -G Ninja \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DCMAKE_INSTALL_PREFIX=/usr/llvm-aarch64 \
+    -DLLVM_ENABLE_ASSERTIONS=ON \
+    -DLLVM_ENABLE_PROJECTS="llvm;clang" \
+    -DCMAKE_SYSTEM_NAME=Linux \
+    -DLLVM_TABLEGEN=/llvm-project/build-host/bin/llvm-tblgen \
+    -DCLANG_TABLEGEN=/llvm-project/build-host/bin/clang-tblgen \
+    -DLLVM_DEFAULT_TARGET_TRIPLE=aarch64-linux-gnu \
+    -DLLVM_TARGET_ARCH=AArch64 \
+    -DCMAKE_CXX_FLAGS='-march=armv8-a -mtune=cortex-a72' \
+    ../llvm
+  ninja install
+  popd
+  popd
+  rm -rf llvm-project
+
+  # This is a hack. Cross-compiling LLVM with gcc will cause the llvm-config to be an Aarch64 executable.
+  # We need it to be x86 to be able to call it when building TVM. We just copy and use the x86 one instead.
+  cp /usr/bin/llvm-config /usr/llvm-aarch64/bin/llvm-config
+fi