test

Author: lanluo-nvidia

packaging/pre_build_script.sh

@@ -10,15 +10,16 @@ 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_S=$(uname -s)
+PLATFORM_M=$(uname -m)
+if [[ ${PLATFORM_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-`uname -s`-`uname -m` -o bazelisk \
-    && mv bazelisk /usr/bin/bazel \
+wget https://github.com/bazelbuild/bazelisk/releases/download/v1.25.0/bazelisk-${PLATFORM_S}-${PLATFORM_M} -o bazelisk-${PLATFORM_S}-${PLATFORM_M} \
+    && mv bazelisk-${PLATFORM_S}-${PLATFORM_M} /usr/bin/bazel \
     && chmod +x /usr/bin/bazel
 
 TORCH_TORCHVISION=$(grep "^torch" py/requirements.txt)

setup.py

@@ -190,8 +190,9 @@ def build_libtorchtrt_cxx11_abi(
         cmd.append("--config=jetpack")
 
     if IS_SBSA:
+        cmd.append("--config=use_torch_whl")
         if CI_BUILD:
-            cmd.append("--platforms=//toolchains/dep_src:torch=whl")
+            cmd.append("--platforms=//toolchains:aarch64_linux")
     else:
         if CI_BUILD:
             cmd.append("--platforms=//toolchains:ci_rhel_x86_64_linux")