Skip to content

Commit 31d0a0a

Browse files
committed
ci: Parallelize packaging builds
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent c808089 commit 31d0a0a

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

.circleci/config.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -754,6 +754,7 @@ jobs:
754754
type: string
755755
torch-build-index:
756756
type: string
757+
parallelism: 4
757758
machine:
758759
image: ubuntu-2004-cuda-11.4:202110-01
759760
resource_class: gpu.nvidia.small
@@ -772,7 +773,7 @@ jobs:
772773
command: |
773774
cd ~/project/py/
774775
cp ~/project/toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel ~/project/WORKSPACE
775-
docker run -it --rm --gpus all -v ~/project:/workspace/project torch_tensorrt_release_env /bin/bash /workspace/project/py/ci/build_whl.sh
776+
docker run -it --rm --gpus all -v ~/project:/workspace/project torch_tensorrt_release_env /bin/bash /workspace/project/py/ci/build_whl.sh $(cat /workspace/project/py/ci/build_manifest.txt | circleci tests split)
776777
- create-env:
777778
os: "ubuntu2004"
778779
platform: "x86_64"

py/ci/build_manifest.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
build_py37
2+
build_py38
3+
build_py39
4+
build_py310
5+
build_libtorchtrt_pre_cxx11_abi

0 commit comments

Comments
 (0)