Skip to content

Commit 923377c

Browse files
committed
chore: update post-build script
1 parent 33ba8b2 commit 923377c

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

.github/workflows/build-test.yml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ jobs:
3636
- repository: pytorch/tensorrt
3737
pre-script: packaging/pre_build_script.sh
3838
env-var-script: packaging/env_vars.txt
39-
post-script: ""
39+
post-script: packaging/post_build_script.sh
4040
smoke-test-script: ""
4141
package-name: torch_tensorrt
4242
name: Build torch-tensorrt whl package
@@ -64,6 +64,7 @@ jobs:
6464
- repository: pytorch/tensorrt
6565
package-name: torch_tensorrt
6666
pre-script: packaging/pre_build_script.sh
67+
post-script: packaging/post_build_script.sh
6768
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
6869
with:
6970
job-name: tests-py-torchscript-fe
@@ -99,6 +100,7 @@ jobs:
99100
- repository: pytorch/tensorrt
100101
package-name: torch_tensorrt
101102
pre-script: packaging/pre_build_script.sh
103+
post-script: packaging/post_build_script.sh
102104
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
103105
with:
104106
job-name: tests-py-dynamo-converters
@@ -126,6 +128,7 @@ jobs:
126128
- repository: pytorch/tensorrt
127129
package-name: torch_tensorrt
128130
pre-script: packaging/pre_build_script.sh
131+
post-script: packaging/post_build_script.sh
129132
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
130133
with:
131134
job-name: tests-py-dynamo-fe
@@ -154,6 +157,7 @@ jobs:
154157
- repository: pytorch/tensorrt
155158
package-name: torch_tensorrt
156159
pre-script: packaging/pre_build_script.sh
160+
post-script: packaging/post_build_script.sh
157161
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
158162
with:
159163
job-name: tests-py-dynamo-serde
@@ -181,6 +185,7 @@ jobs:
181185
- repository: pytorch/tensorrt
182186
package-name: torch_tensorrt
183187
pre-script: packaging/pre_build_script.sh
188+
post-script: packaging/post_build_script.sh
184189
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
185190
with:
186191
job-name: tests-py-torch-compile-be
@@ -209,6 +214,7 @@ jobs:
209214
- repository: pytorch/tensorrt
210215
package-name: torch_tensorrt
211216
pre-script: packaging/pre_build_script.sh
217+
post-script: packaging/post_build_script.sh
212218
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
213219
with:
214220
job-name: tests-py-dynamo-core

packaging/post_build_script.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
python -m pip install /opt/torch-tensorrt-builds/TensorRT-10.0.0.6/python/tensorrt-10.0.0b6-cp38-none-linux_x86_64.whl

0 commit comments

Comments
 (0)