chore: Update CI

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

.circleci/config.yml

@@ -162,8 +162,7 @@ commands:
       - run:
           name: Build setup
           command: |
-            mv toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> WORKSPACE
-            cd py
+            mv ~/project/toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> ~/project/WORKSPACE
             python3 -m pip install wheel setuptools
             python3 -m pip install pybind11==2.6.2
       - when:
@@ -172,6 +171,7 @@ commands:
             - run:
                 name: Build torch-tensorrt python release package
                 command: |
+                  cd ~/project/py
                   python3 setup.py bdist_wheel --use-cxx11-abi --release
                   python3 setup.py install --use-cxx11-abi --release
                   mkdir -p /tmp/dist/builds
@@ -182,6 +182,7 @@ commands:
             - run:
                 name: Build torch-tensorrt python package
                 command: |
+                  cd ~/project/py
                   python3 setup.py bdist_wheel --use-cxx11-abi
                   python3 setup.py install --use-cxx11-abi
                   mkdir -p /tmp/dist/builds
@@ -637,14 +638,15 @@ jobs:
                 name: Build cxx11-abi tarball
                 command: |
                   set -e
+                  cd ~/project/py/
                   bazel build //:libtorchtrt -c opt
                   CUDA_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __cuda_version__;print(__cuda_version__)")
                   TORCHTRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __version__;print(__version__)")
                   TRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __tensorrt_version__;print(__tensorrt_version__)")
                   CUDNN_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __cudnn_version__;print(__cudnn_version__)")
                   pip3 install -r ~/project/py/requirements.txt
                   TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
-                  cp ~/project/bazel-bin/libtorchtrt.tar.gz ~/project/py/dist/libtorchtrt-${TORCHTRT_VERSION}-cudnn${CUDNN_VERSION}-tensorrt${TRT_VERSION}-cuda${CUDA_VERSION}-libtorch-${TORCH_VERSION}.tar.gz
+                  cp ~/project/bazel-bin/libtorchtrt.tar.gz ~/project/py/dist/libtorchtrt-${TORCHTRT_VERSION}-cudnn${CUDNN_VERSION}-tensorrt${TRT_VERSION}-cuda${CUDA_VERSION}-libtorch${TORCH_VERSION}-x86_64-linux.tar.gz
             - run:
                 name: Collect packages
                 command: |
@@ -720,14 +722,15 @@ jobs:
           name: Build cxx11-abi tarball
           command: |
             set -e
+            cd ~/project/py/
             bazel build //:libtorchtrt -c opt
             CUDA_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __cuda_version__;print(__cuda_version__)")
             TORCHTRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __version__;print(__version__)")
             TRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __tensorrt_version__;print(__tensorrt_version__)")
             CUDNN_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __cudnn_version__;print(__cudnn_version__)")
             pip3 install -r ~/project/py/requirements.txt
             TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
-            cp ~/project/bazel-bin/libtorchtrt.tar.gz ~/project/py/dist/libtorchtrt-${TORCHTRT_VERSION}-cudnn${CUDNN_VERSION}-tensorrt${TRT_VERSION}-cuda${CUDA_VERSION}-libtorch-${TORCH_VERSION}.tar.gz
+            cp ~/project/bazel-bin/libtorchtrt.tar.gz ~/project/py/dist/libtorchtrt-${TORCHTRT_VERSION}-cudnn${CUDNN_VERSION}-tensorrt${TRT_VERSION}-cuda${CUDA_VERSION}-libtorch${TORCH_VERSION}-aarch64-linux-jp<< parameters.jetpack-version >>.tar.gz
       - run:
           name: Move to release dir
           command: |