Add Linux CI build for aarch64 (#3516)

Author: lanluo-nvidia

.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,10 +2,12 @@
 
 import argparse
 import json
+import os
 import sys
 from typing import List
 
-disabled_python_versions: List[str] = []
+# currently we don't support python 3.13t due to tensorrt does not support 3.13t
+disabled_python_versions: List[str] = ["3.13t"]
 
 
 def main(args: list[str]) -> None:
@@ -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

.github/scripts/generate-tensorrt-test-matrix.py

@@ -21,8 +21,8 @@
 # channel: test if the future tensorRT version test workflow is triggered from the release branch(release/2.5 etc....)
 PYTHON_VERSIONS_DICT = {
     "nightly": ["3.11"],
-    "test": ["3.9", "3.10", "3.11", "3.12"],
-    "release": ["3.9", "3.10", "3.11", "3.12"],
+    "test": ["3.9", "3.10", "3.11", "3.12", "3.13"],
+    "release": ["3.9", "3.10", "3.11", "3.12", "3.13"],
 }
 
 # please update the future tensorRT version you want to test here