test

Author: lanluo-nvidia

packaging/pre_build_script.sh

@@ -9,6 +9,8 @@ install -y ninja-build gettext
 
 BAZEL_PLATFORM=amd64
 
+nvidia-smi
+
 if [[ $(uname -m) == "aarch64" ]]; then
     BAZEL_PLATFORM=arm64
     rm -rf /opt/openssl # Not sure whats up with the openssl mismatch

setup.py

@@ -192,7 +192,7 @@ def build_libtorchtrt_cxx11_abi(
     if IS_SBSA:
         if CI_BUILD:
             cmd.append("--platforms=//toolchains:aarch64_linux")
-            cmd.append("--//toolchains/dep_src:torch=whl")
+            # cmd.append("--//toolchains/dep_src:torch=whl")
     else:
         if CI_BUILD:
             cmd.append("--platforms=//toolchains:ci_rhel_x86_64_linux")