Add Linux CI build for aarch64

.bazelrc

@@ -36,6 +36,8 @@ build:cxx11_abi --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=1"
 build:cxx11_abi --linkopt="-D_GLIBCXX_USE_CXX11_ABI=1"
 build:cxx11_abi --define=abi=cxx11_abi
 
+build:jetpack --//toolchains/dep_collection:compute_libs=jetpack
+
 build:ci_testing --define=torchtrt_src=prebuilt --cxxopt="-DDISABLE_TEST_IN_CI" --action_env "NVIDIA_TF32_OVERRIDE=0"
 build:use_precompiled_torchtrt --define=torchtrt_src=prebuilt
 

.bazelversion

@@ -1 +1 @@
-7.2.1
+8.1.1

.github/scripts/filter-matrix.py

@@ -2,9 +2,11 @@
 
 import argparse
 import json
+import os
 import sys
 from typing import List
 
+# this was introduced to avoid build for py3.13, currently we support py3.13, but keep this for future use with other python versions
 disabled_python_versions: List[str] = []
 
 
@@ -21,11 +23,20 @@ def main(args: list[str]) -> None:
 
     if options.matrix == "":
         raise Exception("--matrix needs to be provided")
+
     matrix_dict = json.loads(options.matrix)
     includes = matrix_dict["include"]
     filtered_includes = []
     for item in includes:
-        if item["python_version"] not in disabled_python_versions:
+        if item["python_version"] in disabled_python_versions:
+            continue
+        if item["gpu_arch_type"] == "cuda-aarch64":
+            # pytorch image:pytorch/manylinuxaarch64-builder:cuda12.8 comes with glibc2.28
+            # however, TensorRT requires glibc2.31 on aarch64 platform
+            # TODO: in future, if pytorch supports aarch64 with glibc2.31, we should switch to use the pytorch image
+            item["container_image"] = "quay.io/pypa/manylinux_2_34_aarch64"
+            filtered_includes.append(item)
+        else:
             filtered_includes.append(item)
     filtered_matrix_dict = {}
     filtered_matrix_dict["include"] = filtered_includes