test

Author: lanluo-nvidia

.github/scripts/filter-matrix.py

@@ -28,7 +28,8 @@ def main(args: List[str]) -> None:
         if all(
             [
                 item["python_version"] not in disabled_python_versions,
-                item["desired_cuda"] != cpu_tag,
+                # only test cu128 for aarch64 for now for quick CI
+                item["desired_cuda"] == "cu128",
             ]
         ):
             filtered_includes.append(item)

packaging/pre_build_script.sh

@@ -7,18 +7,18 @@ python3 -m pip install pyyaml
 
 install -y ninja-build gettext
 
-PLATFORM="amd64"
-PLATFORM=x86_64
-BAZEL_PLATFORM=amd64
-if [[ $(uname -m) == "aarch64" ]]; then
-    PLATFORM=aarch64
-    BAZEL_PLATFORM=arm64
-
-    rm -rf /opt/openssl # Not sure whats up with the openssl mismatch
-fi
+# PLATFORM="amd64"
+# PLATFORM=x86_64
+# BAZEL_PLATFORM=amd64
+# if [[ $(uname -m) == "aarch64" ]]; then
+#     PLATFORM=aarch64
+#     BAZEL_PLATFORM=arm64
+
+#     rm -rf /opt/openssl # Not sure whats up with the openssl mismatch
+# fi
 
-wget https://github.com/bazelbuild/bazelisk/releases/download/v1.25.0/bazelisk-linux-${BAZEL_PLATFORM} \
-    && mv bazelisk-linux-${BAZEL_PLATFORM} /usr/bin/bazel \
+wget https://github.com/bazelbuild/bazelisk/releases/download/v1.25.0/bazelisk-`uname -s`-`uname -m` -o bazelisk \
+    && mv bazelisk /usr/bin/bazel \
     && chmod +x /usr/bin/bazel
 
 TORCH_TORCHVISION=$(grep "^torch" py/requirements.txt)
@@ -49,6 +49,8 @@ if [[ "${CU_VERSION::4}" < "cu12" ]]; then
          -e "s/tensorrt-cu12-libs/tensorrt-${CU_VERSION::4}-libs/g" \
          pyproject.toml
 fi
+
+
 curl -L  https://github.com/a8m/envsubst/releases/download/v1.4.2/envsubst-Linux-arm64 -o envsubst \
 && mv envsubst /usr/bin/envsubst && chmod +x /usr/bin/envsubst
 

setup.py

@@ -128,6 +128,7 @@ def load_dep_info():
 
 if RELEASE:
     __version__ = os.environ.get("BUILD_VERSION")
+    print(f"lan added BUILD_VERSION: {__version__}")
 else:
     __version__ = f"{get_base_version()}.dev0+{get_git_revision_short_hash()}"
 
@@ -190,11 +191,11 @@ def build_libtorchtrt_cxx11_abi(
 
     if IS_SBSA:
         if CI_BUILD:
-            cmd.append("--//toolchains/dep_src:torch=whl")
-
-    if CI_BUILD:
-        cmd.append("--platforms=//toolchains:ci_rhel_x86_64_linux")
-        print("CI based build")
+            cmd.append("--platforms=//toolchains/dep_src:torch=whl")
+    else:
+        if CI_BUILD:
+            cmd.append("--platforms=//toolchains:ci_rhel_x86_64_linux")
+            print("CI based build")
 
     print(f"building libtorchtrt {cmd=}")
     status_code = subprocess.run(cmd).returncode