|
1 | 1 | name: Build and test Linux wheels
|
2 | 2 |
|
3 | 3 | 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: |
15 | 15 |
|
16 | 16 | 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 |
37 | 19 | 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}" |
47 | 47 |
|
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" |
77 | 81 |
|
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 |
113 | 117 |
|
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 |
142 | 146 |
|
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 |
171 | 175 |
|
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 |
201 | 205 |
|
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 |
232 | 236 |
|
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 |
263 | 267 |
|
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 |
294 | 298 |
|
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 |
323 | 327 |
|
324 | 328 | 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 |
0 commit comments