Skip to content

Commit 7765146

Browse files
committed
add some test
1 parent 67cbaf3 commit 7765146

File tree

1 file changed

+37
-1
lines changed

1 file changed

+37
-1
lines changed

.github/workflows/build-test-tensorrt-linux.yml

Lines changed: 37 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,4 +66,40 @@ jobs:
6666
post-script: ${{ matrix.post-script }}
6767
package-name: ${{ matrix.package-name }}
6868
smoke-test-script: ${{ matrix.smoke-test-script }}
69-
trigger-event: ${{ github.event_name }}
69+
trigger-event: ${{ github.event_name }}
70+
71+
tests-py-torchscript-fe:
72+
name: Test torchscript frontend [Python]
73+
needs: [generate-tensorrt-matrix, build]
74+
strategy:
75+
fail-fast: false
76+
matrix:
77+
include:
78+
- repository: pytorch/tensorrt
79+
package-name: torch_tensorrt
80+
pre-script: packaging/pre_build_script.sh
81+
post-script: packaging/post_build_script.sh
82+
smoke-test-script: packaging/smoke_test_script.sh
83+
uses: ./.github/workflows/linux-test.yml
84+
with:
85+
job-name: tests-py-torchscript-fe
86+
repository: "pytorch/tensorrt"
87+
ref: ""
88+
test-infra-repository: pytorch/test-infra
89+
test-infra-ref: main
90+
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
91+
pre-script: ${{ matrix.pre-script }}
92+
script: |
93+
export USE_HOST_DEPS=1
94+
export CI_BUILD=1
95+
export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
96+
pushd .
97+
cd tests/modules
98+
python hub.py
99+
popd
100+
pushd .
101+
cd tests/py/ts
102+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
103+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
104+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
105+
popd

0 commit comments

Comments
 (0)