[aarch64] add gcc-10 option to support SVE based library compilation

Author: snadampal

build_aarch64_wheel.py

@@ -420,6 +420,14 @@ def configure_system(host: RemoteHost, *,
 
     host.run_cmd("pip3 install dataclasses typing-extensions")
 
+    # Install and switch to gcc-10 on Ubuntu-20.04. This is required to support
+    # SVE instruction set for newer aarch64 architectures
+    if not host.using_docker() and host.ami == ubuntu20_04_ami and compiler == 'gcc-10':
+        host.run_cmd("sudo apt-get install -y g++-10 gfortran-10")
+        host.run_cmd("sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-10 100")
+        host.run_cmd("sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-10 100")
+        host.run_cmd("sudo update-alternatives --install /usr/bin/gfortran gfortran /usr/bin/gfortran-10 100")
+
 
 def start_build(host: RemoteHost, *,
                 branch="master",
@@ -444,9 +452,9 @@ def start_build(host: RemoteHost, *,
         # HACK: pypa gforntran.a is compiled without PIC, which leads to the following error
         # libgfortran.a(error.o)(.text._gfortrani_st_printf+0x34): unresolvable R_AARCH64_ADR_PREL_PG_HI21 relocation against symbol `__stack_chk_guard@@GLIBC_2.17'
         # Workaround by copying gfortran library from the host
-        host.run_ssh_cmd("sudo apt-get install -y gfortran-8")
-        host.run_cmd("mkdir -p /usr/lib/gcc/aarch64-linux-gnu/8")
-        host.run_ssh_cmd(["docker", "cp", "/usr/lib/gcc/aarch64-linux-gnu/8/libgfortran.a",
+        host.run_ssh_cmd("sudo apt-get install -y gfortran-10")
+        host.run_cmd("mkdir -p /usr/lib/gcc/aarch64-linux-gnu/10")
+        host.run_ssh_cmd(["docker", "cp", "/usr/lib/gcc/aarch64-linux-gnu/10/libgfortran.a",
                          f"{host.container_id}:/opt/rh/devtoolset-10/root/usr/lib/gcc/aarch64-redhat-linux/10/"
                           ])
 
@@ -613,7 +621,7 @@ def parse_arguments():
     parser.add_argument("--instance-type", type=str, default="t4g.2xlarge")
     parser.add_argument("--branch", type=str, default="master")
     parser.add_argument("--use-docker", action="store_true")
-    parser.add_argument("--compiler", type=str, choices=['gcc-7', 'gcc-8', 'gcc-9', 'clang'], default="gcc-8")
+    parser.add_argument("--compiler", type=str, choices=['gcc-7', 'gcc-8', 'gcc-9', 'gcc-10', 'clang'], default="gcc-8")
     parser.add_argument("--use-torch-from-pypi", action="store_true")
     return parser.parse_args()