Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit abe92d8

Browse files
authoredJan 9, 2025··
Update build-test-linux.yml (#3345)
Signed-off-by: Naren Dasan <narens@nvidia.com>
1 parent 37cb8a7 commit abe92d8

File tree

4 files changed

+317
-312
lines changed

4 files changed

+317
-312
lines changed
 
Lines changed: 313 additions & 309 deletions
Original file line numberDiff line numberDiff line change
@@ -1,326 +1,330 @@
11
name: Build and test Linux wheels
22

33
on:
4-
pull_request:
5-
push:
6-
branches:
7-
- main
8-
- nightly
9-
- release/*
10-
tags:
11-
# NOTE: Binary build pipelines should only get triggered on release candidate builds
12-
# Release candidate tags look like: v1.11.0-rc1
13-
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
14-
workflow_dispatch:
4+
pull_request:
5+
push:
6+
branches:
7+
- main
8+
- nightly
9+
- release/*
10+
tags:
11+
# NOTE: Binary build pipelines should only get triggered on release candidate builds
12+
# Release candidate tags look like: v1.11.0-rc1
13+
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
14+
workflow_dispatch:
1515

1616
jobs:
17-
generate-matrix:
18-
uses: ./.github/workflows/generate_binary_build_matrix.yml
19-
with:
20-
package-type: wheel
21-
os: linux
22-
test-infra-repository: pytorch/test-infra
23-
test-infra-ref: main
24-
with-rocm: false
25-
with-cpu: false
26-
27-
filter-matrix:
28-
needs: [generate-matrix]
29-
outputs:
30-
matrix: ${{ steps.generate.outputs.matrix }}
31-
runs-on: ubuntu-latest
32-
steps:
33-
- uses: actions/setup-python@v4
34-
with:
35-
python-version: '3.10'
36-
- uses: actions/checkout@v3
17+
generate-matrix:
18+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
3719
with:
38-
repository: pytorch/tensorrt
39-
- name: Generate release matrix
40-
id: generate
41-
run: |
42-
set -eou pipefail
43-
MATRIX_BLOB=${{ toJSON(needs.generate-matrix.outputs.matrix) }}
44-
MATRIX_BLOB="$(python3 .github/scripts/filter-matrix.py --matrix "${MATRIX_BLOB}")"
45-
echo "${MATRIX_BLOB}"
46-
echo "matrix=${MATRIX_BLOB}" >> "${GITHUB_OUTPUT}"
20+
package-type: wheel
21+
os: linux
22+
test-infra-repository: pytorch/test-infra
23+
test-infra-ref: main
24+
with-rocm: false
25+
with-cpu: false
26+
27+
filter-matrix:
28+
needs: [generate-matrix]
29+
outputs:
30+
matrix: ${{ steps.generate.outputs.matrix }}
31+
runs-on: ubuntu-latest
32+
steps:
33+
- uses: actions/setup-python@v4
34+
with:
35+
python-version: "3.10"
36+
- uses: actions/checkout@v3
37+
with:
38+
repository: pytorch/tensorrt
39+
- name: Generate release matrix
40+
id: generate
41+
run: |
42+
set -eou pipefail
43+
MATRIX_BLOB=${{ toJSON(needs.generate-matrix.outputs.matrix) }}
44+
MATRIX_BLOB="$(python3 .github/scripts/filter-matrix.py --matrix "${MATRIX_BLOB}")"
45+
echo "${MATRIX_BLOB}"
46+
echo "matrix=${MATRIX_BLOB}" >> "${GITHUB_OUTPUT}"
4747
48-
build:
49-
needs: filter-matrix
50-
permissions:
51-
id-token: write
52-
contents: read
53-
strategy:
54-
fail-fast: false
55-
matrix:
56-
include:
57-
- repository: pytorch/tensorrt
58-
pre-script: packaging/pre_build_script.sh
59-
env-var-script: packaging/env_vars.txt
60-
post-script: packaging/post_build_script.sh
61-
smoke-test-script: packaging/smoke_test_script.sh
62-
package-name: torch_tensorrt
63-
name: Build torch-tensorrt whl package
64-
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
65-
with:
66-
repository: ${{ matrix.repository }}
67-
ref: ""
68-
test-infra-repository: pytorch/test-infra
69-
test-infra-ref: main
70-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
71-
pre-script: ${{ matrix.pre-script }}
72-
env-var-script: ${{ matrix.env-var-script }}
73-
post-script: ${{ matrix.post-script }}
74-
package-name: ${{ matrix.package-name }}
75-
smoke-test-script: ${{ matrix.smoke-test-script }}
76-
trigger-event: ${{ github.event_name }}
48+
build:
49+
needs: filter-matrix
50+
permissions:
51+
id-token: write
52+
contents: read
53+
strategy:
54+
fail-fast: false
55+
matrix:
56+
include:
57+
- repository: pytorch/tensorrt
58+
pre-script: packaging/pre_build_script.sh
59+
env-var-script: packaging/env_vars.txt
60+
post-script: packaging/post_build_script.sh
61+
smoke-test-script: packaging/smoke_test_script.sh
62+
package-name: torch_tensorrt
63+
name: Build torch-tensorrt whl package
64+
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
65+
with:
66+
repository: ${{ matrix.repository }}
67+
ref: ""
68+
test-infra-repository: pytorch/test-infra
69+
test-infra-ref: main
70+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
71+
pre-script: ${{ matrix.pre-script }}
72+
env-var-script: ${{ matrix.env-var-script }}
73+
post-script: ${{ matrix.post-script }}
74+
package-name: ${{ matrix.package-name }}
75+
smoke-test-script: ${{ matrix.smoke-test-script }}
76+
trigger-event: ${{ github.event_name }}
77+
architecture: "x86_64"
78+
setup-miniconda: false
79+
build-platform: "python-build-package"
80+
build-command: "python -m pip wheel --no-deps -w dist . --extra-index-url https://download.pytorch.org/whl/test/$WHEEL_DIR"
7781

78-
tests-py-torchscript-fe:
79-
name: Test torchscript frontend [Python]
80-
needs: [filter-matrix, build]
81-
strategy:
82-
fail-fast: false
83-
matrix:
84-
include:
85-
- repository: pytorch/tensorrt
86-
package-name: torch_tensorrt
87-
pre-script: packaging/pre_build_script.sh
88-
post-script: packaging/post_build_script.sh
89-
smoke-test-script: packaging/smoke_test_script.sh
90-
uses: ./.github/workflows/linux-test.yml
91-
with:
92-
job-name: tests-py-torchscript-fe
93-
repository: "pytorch/tensorrt"
94-
ref: ""
95-
test-infra-repository: pytorch/test-infra
96-
test-infra-ref: main
97-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
98-
pre-script: ${{ matrix.pre-script }}
99-
script: |
100-
export USE_HOST_DEPS=1
101-
export CI_BUILD=1
102-
export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
103-
pushd .
104-
cd tests/modules
105-
python hub.py
106-
popd
107-
pushd .
108-
cd tests/py/ts
109-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
110-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
111-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
112-
popd
82+
tests-py-torchscript-fe:
83+
name: Test torchscript frontend [Python]
84+
needs: [filter-matrix, build]
85+
strategy:
86+
fail-fast: false
87+
matrix:
88+
include:
89+
- repository: pytorch/tensorrt
90+
package-name: torch_tensorrt
91+
pre-script: packaging/pre_build_script.sh
92+
post-script: packaging/post_build_script.sh
93+
smoke-test-script: packaging/smoke_test_script.sh
94+
uses: ./.github/workflows/linux-test.yml
95+
with:
96+
job-name: tests-py-torchscript-fe
97+
repository: "pytorch/tensorrt"
98+
ref: ""
99+
test-infra-repository: pytorch/test-infra
100+
test-infra-ref: main
101+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
102+
pre-script: ${{ matrix.pre-script }}
103+
script: |
104+
export USE_HOST_DEPS=1
105+
export CI_BUILD=1
106+
export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
107+
pushd .
108+
cd tests/modules
109+
python hub.py
110+
popd
111+
pushd .
112+
cd tests/py/ts
113+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
114+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
115+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
116+
popd
113117
114-
tests-py-dynamo-converters:
115-
name: Test dynamo converters [Python]
116-
needs: [filter-matrix, build]
117-
strategy:
118-
fail-fast: false
119-
matrix:
120-
include:
121-
- repository: pytorch/tensorrt
122-
package-name: torch_tensorrt
123-
pre-script: packaging/pre_build_script.sh
124-
post-script: packaging/post_build_script.sh
125-
smoke-test-script: packaging/smoke_test_script.sh
126-
uses: ./.github/workflows/linux-test.yml
127-
with:
128-
job-name: tests-py-dynamo-converters
129-
repository: "pytorch/tensorrt"
130-
ref: ""
131-
test-infra-repository: pytorch/test-infra
132-
test-infra-ref: main
133-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
134-
pre-script: ${{ matrix.pre-script }}
135-
script: |
136-
export USE_HOST_DEPS=1
137-
export CI_BUILD=1
138-
pushd .
139-
cd tests/py/dynamo
140-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
141-
popd
118+
tests-py-dynamo-converters:
119+
name: Test dynamo converters [Python]
120+
needs: [filter-matrix, build]
121+
strategy:
122+
fail-fast: false
123+
matrix:
124+
include:
125+
- repository: pytorch/tensorrt
126+
package-name: torch_tensorrt
127+
pre-script: packaging/pre_build_script.sh
128+
post-script: packaging/post_build_script.sh
129+
smoke-test-script: packaging/smoke_test_script.sh
130+
uses: ./.github/workflows/linux-test.yml
131+
with:
132+
job-name: tests-py-dynamo-converters
133+
repository: "pytorch/tensorrt"
134+
ref: ""
135+
test-infra-repository: pytorch/test-infra
136+
test-infra-ref: main
137+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
138+
pre-script: ${{ matrix.pre-script }}
139+
script: |
140+
export USE_HOST_DEPS=1
141+
export CI_BUILD=1
142+
pushd .
143+
cd tests/py/dynamo
144+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
145+
popd
142146
143-
tests-py-dynamo-fe:
144-
name: Test dynamo frontend [Python]
145-
needs: [filter-matrix, build]
146-
strategy:
147-
fail-fast: false
148-
matrix:
149-
include:
150-
- repository: pytorch/tensorrt
151-
package-name: torch_tensorrt
152-
pre-script: packaging/pre_build_script.sh
153-
post-script: packaging/post_build_script.sh
154-
smoke-test-script: packaging/smoke_test_script.sh
155-
uses: ./.github/workflows/linux-test.yml
156-
with:
157-
job-name: tests-py-dynamo-fe
158-
repository: "pytorch/tensorrt"
159-
ref: ""
160-
test-infra-repository: pytorch/test-infra
161-
test-infra-ref: main
162-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
163-
pre-script: ${{ matrix.pre-script }}
164-
script: |
165-
export USE_HOST_DEPS=1
166-
export CI_BUILD=1
167-
pushd .
168-
cd tests/py/dynamo
169-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
170-
popd
147+
tests-py-dynamo-fe:
148+
name: Test dynamo frontend [Python]
149+
needs: [filter-matrix, build]
150+
strategy:
151+
fail-fast: false
152+
matrix:
153+
include:
154+
- repository: pytorch/tensorrt
155+
package-name: torch_tensorrt
156+
pre-script: packaging/pre_build_script.sh
157+
post-script: packaging/post_build_script.sh
158+
smoke-test-script: packaging/smoke_test_script.sh
159+
uses: ./.github/workflows/linux-test.yml
160+
with:
161+
job-name: tests-py-dynamo-fe
162+
repository: "pytorch/tensorrt"
163+
ref: ""
164+
test-infra-repository: pytorch/test-infra
165+
test-infra-ref: main
166+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
167+
pre-script: ${{ matrix.pre-script }}
168+
script: |
169+
export USE_HOST_DEPS=1
170+
export CI_BUILD=1
171+
pushd .
172+
cd tests/py/dynamo
173+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
174+
popd
171175
172-
tests-py-dynamo-serde:
173-
name: Test dynamo export serde [Python]
174-
needs: [filter-matrix, build]
175-
strategy:
176-
fail-fast: false
177-
matrix:
178-
include:
179-
- repository: pytorch/tensorrt
180-
package-name: torch_tensorrt
181-
pre-script: packaging/pre_build_script.sh
182-
post-script: packaging/post_build_script.sh
183-
smoke-test-script: packaging/smoke_test_script.sh
184-
uses: ./.github/workflows/linux-test.yml
185-
with:
186-
job-name: tests-py-dynamo-serde
187-
repository: "pytorch/tensorrt"
188-
ref: ""
189-
test-infra-repository: pytorch/test-infra
190-
test-infra-ref: main
191-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
192-
pre-script: ${{ matrix.pre-script }}
193-
script: |
194-
export USE_HOST_DEPS=1
195-
export CI_BUILD=1
196-
pushd .
197-
cd tests/py/dynamo
198-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
199-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/reexport_test_results.xml --ir dynamo models/test_reexport.py
200-
popd
176+
tests-py-dynamo-serde:
177+
name: Test dynamo export serde [Python]
178+
needs: [filter-matrix, build]
179+
strategy:
180+
fail-fast: false
181+
matrix:
182+
include:
183+
- repository: pytorch/tensorrt
184+
package-name: torch_tensorrt
185+
pre-script: packaging/pre_build_script.sh
186+
post-script: packaging/post_build_script.sh
187+
smoke-test-script: packaging/smoke_test_script.sh
188+
uses: ./.github/workflows/linux-test.yml
189+
with:
190+
job-name: tests-py-dynamo-serde
191+
repository: "pytorch/tensorrt"
192+
ref: ""
193+
test-infra-repository: pytorch/test-infra
194+
test-infra-ref: main
195+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
196+
pre-script: ${{ matrix.pre-script }}
197+
script: |
198+
export USE_HOST_DEPS=1
199+
export CI_BUILD=1
200+
pushd .
201+
cd tests/py/dynamo
202+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
203+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/reexport_test_results.xml --ir dynamo models/test_reexport.py
204+
popd
201205
202-
tests-py-torch-compile-be:
203-
name: Test torch compile backend [Python]
204-
needs: [filter-matrix, build]
205-
strategy:
206-
fail-fast: false
207-
matrix:
208-
include:
209-
- repository: pytorch/tensorrt
210-
package-name: torch_tensorrt
211-
pre-script: packaging/pre_build_script.sh
212-
post-script: packaging/post_build_script.sh
213-
smoke-test-script: packaging/smoke_test_script.sh
214-
uses: ./.github/workflows/linux-test.yml
215-
with:
216-
job-name: tests-py-torch-compile-be
217-
repository: "pytorch/tensorrt"
218-
ref: ""
219-
test-infra-repository: pytorch/test-infra
220-
test-infra-ref: main
221-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
222-
pre-script: ${{ matrix.pre-script }}
223-
script: |
224-
export USE_HOST_DEPS=1
225-
export CI_BUILD=1
226-
pushd .
227-
cd tests/py/dynamo
228-
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
229-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_complete_be_e2e_test_results.xml --ir torch_compile models/test_models.py
230-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
231-
popd
206+
tests-py-torch-compile-be:
207+
name: Test torch compile backend [Python]
208+
needs: [filter-matrix, build]
209+
strategy:
210+
fail-fast: false
211+
matrix:
212+
include:
213+
- repository: pytorch/tensorrt
214+
package-name: torch_tensorrt
215+
pre-script: packaging/pre_build_script.sh
216+
post-script: packaging/post_build_script.sh
217+
smoke-test-script: packaging/smoke_test_script.sh
218+
uses: ./.github/workflows/linux-test.yml
219+
with:
220+
job-name: tests-py-torch-compile-be
221+
repository: "pytorch/tensorrt"
222+
ref: ""
223+
test-infra-repository: pytorch/test-infra
224+
test-infra-ref: main
225+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
226+
pre-script: ${{ matrix.pre-script }}
227+
script: |
228+
export USE_HOST_DEPS=1
229+
export CI_BUILD=1
230+
pushd .
231+
cd tests/py/dynamo
232+
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
233+
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_complete_be_e2e_test_results.xml --ir torch_compile models/test_models.py
234+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
235+
popd
232236
233-
tests-py-dynamo-core:
234-
name: Test dynamo core [Python]
235-
needs: [filter-matrix, build]
236-
strategy:
237-
fail-fast: false
238-
matrix:
239-
include:
240-
- repository: pytorch/tensorrt
241-
package-name: torch_tensorrt
242-
pre-script: packaging/pre_build_script.sh
243-
post-script: packaging/post_build_script.sh
244-
smoke-test-script: packaging/smoke_test_script.sh
245-
uses: ./.github/workflows/linux-test.yml
246-
with:
247-
job-name: tests-py-dynamo-core
248-
repository: "pytorch/tensorrt"
249-
ref: ""
250-
test-infra-repository: pytorch/test-infra
251-
test-infra-ref: main
252-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
253-
pre-script: ${{ matrix.pre-script }}
254-
script: |
255-
export USE_HOST_DEPS=1
256-
export CI_BUILD=1
257-
pushd .
258-
cd tests/py/dynamo
259-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml --ignore runtime/test_002_cudagraphs_py.py --ignore runtime/test_002_cudagraphs_cpp.py runtime/
260-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
261-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
262-
popd
237+
tests-py-dynamo-core:
238+
name: Test dynamo core [Python]
239+
needs: [filter-matrix, build]
240+
strategy:
241+
fail-fast: false
242+
matrix:
243+
include:
244+
- repository: pytorch/tensorrt
245+
package-name: torch_tensorrt
246+
pre-script: packaging/pre_build_script.sh
247+
post-script: packaging/post_build_script.sh
248+
smoke-test-script: packaging/smoke_test_script.sh
249+
uses: ./.github/workflows/linux-test.yml
250+
with:
251+
job-name: tests-py-dynamo-core
252+
repository: "pytorch/tensorrt"
253+
ref: ""
254+
test-infra-repository: pytorch/test-infra
255+
test-infra-ref: main
256+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
257+
pre-script: ${{ matrix.pre-script }}
258+
script: |
259+
export USE_HOST_DEPS=1
260+
export CI_BUILD=1
261+
pushd .
262+
cd tests/py/dynamo
263+
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml --ignore runtime/test_002_cudagraphs_py.py --ignore runtime/test_002_cudagraphs_cpp.py runtime/
264+
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
265+
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
266+
popd
263267
264-
tests-py-dynamo-cudagraphs:
265-
name: Test dynamo cudagraphs [Python]
266-
needs: [filter-matrix, build]
267-
strategy:
268-
fail-fast: false
269-
matrix:
270-
include:
271-
- repository: pytorch/tensorrt
272-
package-name: torch_tensorrt
273-
pre-script: packaging/pre_build_script.sh
274-
post-script: packaging/post_build_script.sh
275-
smoke-test-script: packaging/smoke_test_script.sh
276-
uses: ./.github/workflows/linux-test.yml
277-
with:
278-
job-name: tests-py-dynamo-cudagraphs
279-
repository: "pytorch/tensorrt"
280-
ref: ""
281-
test-infra-repository: pytorch/test-infra
282-
test-infra-ref: main
283-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
284-
pre-script: ${{ matrix.pre-script }}
285-
script: |
286-
export USE_HOST_DEPS=1
287-
export CI_BUILD=1
288-
pushd .
289-
cd tests/py/dynamo
290-
nvidia-smi
291-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_cpp_test_results.xml runtime/test_002_cudagraphs_cpp.py || true
292-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_py_test_results.xml runtime/test_002_cudagraphs_py.py || true
293-
popd
268+
tests-py-dynamo-cudagraphs:
269+
name: Test dynamo cudagraphs [Python]
270+
needs: [filter-matrix, build]
271+
strategy:
272+
fail-fast: false
273+
matrix:
274+
include:
275+
- repository: pytorch/tensorrt
276+
package-name: torch_tensorrt
277+
pre-script: packaging/pre_build_script.sh
278+
post-script: packaging/post_build_script.sh
279+
smoke-test-script: packaging/smoke_test_script.sh
280+
uses: ./.github/workflows/linux-test.yml
281+
with:
282+
job-name: tests-py-dynamo-cudagraphs
283+
repository: "pytorch/tensorrt"
284+
ref: ""
285+
test-infra-repository: pytorch/test-infra
286+
test-infra-ref: main
287+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
288+
pre-script: ${{ matrix.pre-script }}
289+
script: |
290+
export USE_HOST_DEPS=1
291+
export CI_BUILD=1
292+
pushd .
293+
cd tests/py/dynamo
294+
nvidia-smi
295+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_cpp_test_results.xml runtime/test_002_cudagraphs_cpp.py || true
296+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_py_test_results.xml runtime/test_002_cudagraphs_py.py || true
297+
popd
294298
295-
tests-py-core:
296-
name: Test core [Python]
297-
needs: [filter-matrix, build]
298-
strategy:
299-
fail-fast: false
300-
matrix:
301-
include:
302-
- repository: pytorch/tensorrt
303-
package-name: torch_tensorrt
304-
pre-script: packaging/pre_build_script.sh
305-
post-script: packaging/post_build_script.sh
306-
smoke-test-script: packaging/smoke_test_script.sh
307-
uses: ./.github/workflows/linux-test.yml
308-
with:
309-
job-name: tests-py-core
310-
repository: "pytorch/tensorrt"
311-
ref: ""
312-
test-infra-repository: pytorch/test-infra
313-
test-infra-ref: main
314-
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
315-
pre-script: ${{ matrix.pre-script }}
316-
script: |
317-
export USE_HOST_DEPS=1
318-
export CI_BUILD=1
319-
pushd .
320-
cd tests/py/core
321-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_core_test_results.xml .
322-
popd
299+
tests-py-core:
300+
name: Test core [Python]
301+
needs: [filter-matrix, build]
302+
strategy:
303+
fail-fast: false
304+
matrix:
305+
include:
306+
- repository: pytorch/tensorrt
307+
package-name: torch_tensorrt
308+
pre-script: packaging/pre_build_script.sh
309+
post-script: packaging/post_build_script.sh
310+
smoke-test-script: packaging/smoke_test_script.sh
311+
uses: ./.github/workflows/linux-test.yml
312+
with:
313+
job-name: tests-py-core
314+
repository: "pytorch/tensorrt"
315+
ref: ""
316+
test-infra-repository: pytorch/test-infra
317+
test-infra-ref: main
318+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
319+
pre-script: ${{ matrix.pre-script }}
320+
script: |
321+
export USE_HOST_DEPS=1
322+
export CI_BUILD=1
323+
pushd .
324+
cd tests/py/core
325+
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_core_test_results.xml .
326+
popd
323327
324328
concurrency:
325-
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
326-
cancel-in-progress: true
329+
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
330+
cancel-in-progress: true

‎packaging/env_vars.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,5 @@ if [ $CU_VERSION == "cu126" ]; then
44
export USE_CXX11_ABI="1"
55
else
66
export USE_CXX11_ABI="0"
7-
fi
7+
fi
8+
export CHANNEL="test"

‎packaging/pre_build_script.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ python3 -m pip install pyyaml
77

88
yum install -y ninja-build gettext
99

10-
wget https://github.com/bazelbuild/bazelisk/releases/download/v1.17.0/bazelisk-linux-amd64 \
10+
wget https://github.com/bazelbuild/bazelisk/releases/download/v1.25.0/bazelisk-linux-amd64 \
1111
&& mv bazelisk-linux-amd64 /usr/bin/bazel \
1212
&& chmod +x /usr/bin/bazel
1313

‎py/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,6 @@ packaging
33
pybind11==2.6.2
44
--extra-index-url https://download.pytorch.org/whl/test/cu126
55
torch>=2.6.0.dev,<2.7.0
6-
torchvision>=0.20.0.dev,<0.21.0
6+
torchvision>=0.20.0.dev,<0.22.0
77
--extra-index-url https://pypi.ngc.nvidia.com
88
pyyaml

0 commit comments

Comments
 (0)
Please sign in to comment.