Skip to content

Commit 8c0a6eb

Browse files
committed
chore: Update CI
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 4267406 commit 8c0a6eb

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

.circleci/config.yml

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -162,8 +162,7 @@ commands:
162162
- run:
163163
name: Build setup
164164
command: |
165-
mv toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> WORKSPACE
166-
cd py
165+
mv ~/project/toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> ~/project/WORKSPACE
167166
python3 -m pip install wheel setuptools
168167
python3 -m pip install pybind11==2.6.2
169168
- when:
@@ -172,6 +171,7 @@ commands:
172171
- run:
173172
name: Build torch-tensorrt python release package
174173
command: |
174+
cd ~/project/py
175175
python3 setup.py bdist_wheel --use-cxx11-abi --release
176176
python3 setup.py install --use-cxx11-abi --release
177177
mkdir -p /tmp/dist/builds
@@ -182,6 +182,7 @@ commands:
182182
- run:
183183
name: Build torch-tensorrt python package
184184
command: |
185+
cd ~/project/py
185186
python3 setup.py bdist_wheel --use-cxx11-abi
186187
python3 setup.py install --use-cxx11-abi
187188
mkdir -p /tmp/dist/builds
@@ -637,14 +638,15 @@ jobs:
637638
name: Build cxx11-abi tarball
638639
command: |
639640
set -e
641+
cd ~/project/py/
640642
bazel build //:libtorchtrt -c opt
641643
CUDA_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __cuda_version__;print(__cuda_version__)")
642644
TORCHTRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __version__;print(__version__)")
643645
TRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __tensorrt_version__;print(__tensorrt_version__)")
644646
CUDNN_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __cudnn_version__;print(__cudnn_version__)")
645647
pip3 install -r ~/project/py/requirements.txt
646648
TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
647-
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
649+
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
648650
- run:
649651
name: Collect packages
650652
command: |
@@ -720,14 +722,15 @@ jobs:
720722
name: Build cxx11-abi tarball
721723
command: |
722724
set -e
725+
cd ~/project/py/
723726
bazel build //:libtorchtrt -c opt
724727
CUDA_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __cuda_version__;print(__cuda_version__)")
725728
TORCHTRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __version__;print(__version__)")
726729
TRT_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __tensorrt_version__;print(__tensorrt_version__)")
727730
CUDNN_VERSION=$(cd torch_tensorrt && python3 -c "from _version import __cudnn_version__;print(__cudnn_version__)")
728731
pip3 install -r ~/project/py/requirements.txt
729732
TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
730-
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
733+
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
731734
- run:
732735
name: Move to release dir
733736
command: |
@@ -928,19 +931,19 @@ workflows:
928931
enabled: << pipeline.parameters.packaging-enabled >>
929932
torch-build: << pipeline.parameters.torch-release-build >>
930933
torch-build-index: << pipeline.parameters.torch-release-build-index >>
931-
requires:
932-
- test-core-cpp-x86_64-pyt-release
933-
- test-py-ts-x86_64-pyt-release
934-
- test-py-fx-x86_64-pyt-release
934+
#requires:
935+
# - test-core-cpp-x86_64-pyt-release
936+
# - test-py-ts-x86_64-pyt-release
937+
# - test-py-fx-x86_64-pyt-release
935938

936939
- package-jetson:
937940
name: package-release-aarch64-jetson
938941
enabled: << pipeline.parameters.packaging-enabled >>
939942
torch-build: << pipeline.parameters.torch-jetson-build >>
940943
jetpack-version: << pipeline.parameters.jetpack-version >>
941944
python-version: 3.8.10
942-
requires:
943-
- build-aarch64-pyt-jetson
945+
#requires:
946+
# - build-aarch64-pyt-jetson
944947

945948
on-push:
946949
jobs:

0 commit comments

Comments
 (0)