Skip to content

Commit b387d4d

Browse files
zewenli98zewenli98
authored
fix: CI errors on release 2.6 (#3358)
1 parent afa9547 commit b387d4d

16 files changed

+101
-53
lines changed

‎.github/workflows/assigner.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ jobs:
2222
runs-on: ubuntu-latest
2323
steps:
2424
- name: Checkout
25-
uses: actions/checkout@v3
25+
uses: actions/checkout@v4
2626

2727
- name: Assign
2828
uses: ./.github/actions/assigner

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

+3-3
Original file line numberDiff line numberDiff line change
@@ -114,13 +114,13 @@ jobs:
114114
rm -rf "${RUNNER_TEMP}/*"
115115
fi
116116
echo "::endgroup::"
117-
- uses: actions/checkout@v3
117+
- uses: actions/checkout@v4
118118
with:
119119
# Support the use case where we need to checkout someone's fork
120120
repository: ${{ inputs.test-infra-repository }}
121121
ref: ${{ inputs.test-infra-ref }}
122122
path: test-infra
123-
- uses: actions/checkout@v3
123+
- uses: actions/checkout@v4
124124
if: ${{ env.ARCH == 'aarch64' }}
125125
with:
126126
# Support the use case where we need to checkout someone's fork
@@ -212,7 +212,7 @@ jobs:
212212
# NB: Only upload to GitHub after passing smoke tests
213213
- name: Upload wheel to GitHub
214214
continue-on-error: true
215-
uses: actions/upload-artifact@v3
215+
uses: actions/upload-artifact@v4
216216
with:
217217
name: ${{ env.UPLOAD_ARTIFACT_NAME }}
218218
path: ${{ inputs.repository }}/dist

‎.github/workflows/build-tensorrt-windows.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ jobs:
100100
# to have a conversation
101101
timeout-minutes: 120
102102
steps:
103-
- uses: actions/checkout@v3
103+
- uses: actions/checkout@v4
104104
with:
105105
# Support the use case where we need to checkout someone's fork
106106
repository: ${{ inputs.test-infra-repository }}
@@ -216,7 +216,7 @@ jobs:
216216
# NB: Only upload to GitHub after passing smoke tests
217217
- name: Upload wheel to GitHub
218218
continue-on-error: true
219-
uses: actions/upload-artifact@v3
219+
uses: actions/upload-artifact@v4
220220
with:
221221
name: ${{ env.UPLOAD_ARTIFACT_NAME }}
222222
path: ${{ inputs.repository }}/dist/

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

+19-7
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ jobs:
3333
- uses: actions/setup-python@v4
3434
with:
3535
python-version: '3.10'
36-
- uses: actions/checkout@v3
36+
- uses: actions/checkout@v4
3737
with:
3838
repository: pytorch/tensorrt
3939
- name: Generate release matrix
@@ -136,7 +136,9 @@ jobs:
136136
export USE_HOST_DEPS=1
137137
export CI_BUILD=1
138138
pushd .
139-
cd tests/py/dynamo
139+
cd tests/py
140+
python -m pip install -r requirements.txt
141+
cd dynamo
140142
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
141143
popd
142144
@@ -165,7 +167,9 @@ jobs:
165167
export USE_HOST_DEPS=1
166168
export CI_BUILD=1
167169
pushd .
168-
cd tests/py/dynamo
170+
cd tests/py
171+
python -m pip install -r requirements.txt
172+
cd dynamo
169173
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
170174
popd
171175
@@ -194,7 +198,9 @@ jobs:
194198
export USE_HOST_DEPS=1
195199
export CI_BUILD=1
196200
pushd .
197-
cd tests/py/dynamo
201+
cd tests/py
202+
python -m pip install -r requirements.txt
203+
cd dynamo
198204
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
199205
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/reexport_test_results.xml --ir dynamo models/test_reexport.py
200206
popd
@@ -224,7 +230,9 @@ jobs:
224230
export USE_HOST_DEPS=1
225231
export CI_BUILD=1
226232
pushd .
227-
cd tests/py/dynamo
233+
cd tests/py
234+
python -m pip install -r requirements.txt
235+
cd dynamo
228236
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
229237
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
230238
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
@@ -255,7 +263,9 @@ jobs:
255263
export USE_HOST_DEPS=1
256264
export CI_BUILD=1
257265
pushd .
258-
cd tests/py/dynamo
266+
cd tests/py
267+
python -m pip install -r requirements.txt
268+
cd dynamo
259269
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/
260270
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
261271
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
@@ -286,7 +296,9 @@ jobs:
286296
export USE_HOST_DEPS=1
287297
export CI_BUILD=1
288298
pushd .
289-
cd tests/py/dynamo
299+
cd tests/py
300+
python -m pip install -r requirements.txt
301+
cd dynamo
290302
nvidia-smi
291303
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
292304
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

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

+19-7
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ jobs:
2828
- uses: actions/setup-python@v4
2929
with:
3030
python-version: '3.10'
31-
- uses: actions/checkout@v3
31+
- uses: actions/checkout@v4
3232
with:
3333
repository: pytorch/tensorrt
3434
- name: Generate tensorrt matrix
@@ -128,7 +128,9 @@ jobs:
128128
export USE_HOST_DEPS=1
129129
export CI_BUILD=1
130130
pushd .
131-
cd tests/py/dynamo
131+
cd tests/py
132+
python -m pip install -r requirements.txt
133+
cd dynamo
132134
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
133135
popd
134136
@@ -157,7 +159,9 @@ jobs:
157159
export USE_HOST_DEPS=1
158160
export CI_BUILD=1
159161
pushd .
160-
cd tests/py/dynamo
162+
cd tests/py
163+
python -m pip install -r requirements.txt
164+
cd dynamo
161165
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
162166
popd
163167
@@ -186,7 +190,9 @@ jobs:
186190
export USE_HOST_DEPS=1
187191
export CI_BUILD=1
188192
pushd .
189-
cd tests/py/dynamo
193+
cd tests/py
194+
python -m pip install -r requirements.txt
195+
cd dynamo
190196
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
191197
popd
192198
@@ -215,7 +221,9 @@ jobs:
215221
export USE_HOST_DEPS=1
216222
export CI_BUILD=1
217223
pushd .
218-
cd tests/py/dynamo
224+
cd tests/py
225+
python -m pip install -r requirements.txt
226+
cd dynamo
219227
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
220228
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
221229
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
@@ -246,7 +254,9 @@ jobs:
246254
export USE_HOST_DEPS=1
247255
export CI_BUILD=1
248256
pushd .
249-
cd tests/py/dynamo
257+
cd tests/py
258+
python -m pip install -r requirements.txt
259+
cd dynamo
250260
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/
251261
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
252262
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
@@ -277,7 +287,9 @@ jobs:
277287
export USE_HOST_DEPS=1
278288
export CI_BUILD=1
279289
pushd .
280-
cd tests/py/dynamo
290+
cd tests/py
291+
python -m pip install -r requirements.txt
292+
cd dynamo
281293
nvidia-smi
282294
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
283295
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

‎.github/workflows/build-test-tensorrt-windows.yml

+19-7
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ jobs:
2828
- uses: actions/setup-python@v4
2929
with:
3030
python-version: '3.10'
31-
- uses: actions/checkout@v3
31+
- uses: actions/checkout@v4
3232
with:
3333
repository: pytorch/tensorrt
3434
- name: Generate tensorrt matrix
@@ -131,7 +131,9 @@ jobs:
131131
export USE_HOST_DEPS=1
132132
export CI_BUILD=1
133133
pushd .
134-
cd tests/py/dynamo
134+
cd tests/py
135+
python -m pip install -r requirements.txt
136+
cd dynamo
135137
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
136138
popd
137139
@@ -157,7 +159,9 @@ jobs:
157159
export USE_HOST_DEPS=1
158160
export CI_BUILD=1
159161
pushd .
160-
cd tests/py/dynamo
162+
cd tests/py
163+
python -m pip install -r requirements.txt
164+
cd dynamo
161165
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
162166
popd
163167
@@ -183,7 +187,9 @@ jobs:
183187
export USE_HOST_DEPS=1
184188
export CI_BUILD=1
185189
pushd .
186-
cd tests/py/dynamo
190+
cd tests/py
191+
python -m pip install -r requirements.txt
192+
cd dynamo
187193
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
188194
popd
189195
@@ -209,7 +215,9 @@ jobs:
209215
export USE_HOST_DEPS=1
210216
export CI_BUILD=1
211217
pushd .
212-
cd tests/py/dynamo
218+
cd tests/py
219+
python -m pip install -r requirements.txt
220+
cd dynamo
213221
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
214222
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
215223
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
@@ -237,7 +245,9 @@ jobs:
237245
export USE_HOST_DEPS=1
238246
export CI_BUILD=1
239247
pushd .
240-
cd tests/py/dynamo
248+
cd tests/py
249+
python -m pip install -r requirements.txt
250+
cd dynamo
241251
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/
242252
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
243253
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
@@ -265,7 +275,9 @@ jobs:
265275
export USE_HOST_DEPS=1
266276
export CI_BUILD=1
267277
pushd .
268-
cd tests/py/dynamo
278+
cd tests/py
279+
python -m pip install -r requirements.txt
280+
cd dynamo
269281
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
270282
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
271283
popd

‎.github/workflows/build-test-windows.yml

+18-6
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,9 @@ jobs:
118118
export USE_HOST_DEPS=1
119119
export CI_BUILD=1
120120
pushd .
121-
cd tests/py/dynamo
121+
cd tests/py
122+
python -m pip install -r requirements.txt
123+
cd dynamo
122124
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
123125
popd
124126
@@ -144,7 +146,9 @@ jobs:
144146
export USE_HOST_DEPS=1
145147
export CI_BUILD=1
146148
pushd .
147-
cd tests/py/dynamo
149+
cd tests/py
150+
python -m pip install -r requirements.txt
151+
cd dynamo
148152
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
149153
popd
150154
@@ -170,7 +174,9 @@ jobs:
170174
export USE_HOST_DEPS=1
171175
export CI_BUILD=1
172176
pushd .
173-
cd tests/py/dynamo
177+
cd tests/py
178+
python -m pip install -r requirements.txt
179+
cd dynamo
174180
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
175181
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/reexport_test_results.xml --ir dynamo models/test_reexport.py
176182
popd
@@ -197,7 +203,9 @@ jobs:
197203
export USE_HOST_DEPS=1
198204
export CI_BUILD=1
199205
pushd .
200-
cd tests/py/dynamo
206+
cd tests/py
207+
python -m pip install -r requirements.txt
208+
cd dynamo
201209
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
202210
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
203211
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
@@ -225,7 +233,9 @@ jobs:
225233
export USE_HOST_DEPS=1
226234
export CI_BUILD=1
227235
pushd .
228-
cd tests/py/dynamo
236+
cd tests/py
237+
python -m pip install -r requirements.txt
238+
cd dynamo
229239
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/
230240
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
231241
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
@@ -253,7 +263,9 @@ jobs:
253263
export USE_HOST_DEPS=1
254264
export CI_BUILD=1
255265
pushd .
256-
cd tests/py/dynamo
266+
cd tests/py
267+
python -m pip install -r requirements.txt
268+
cd dynamo
257269
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
258270
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
259271
popd

‎.github/workflows/docker_builder.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ jobs:
3030

3131
steps:
3232
- name: Checkout repository
33-
uses: actions/checkout@v3
33+
uses: actions/checkout@v4
3434

3535
- name: Fix Slashes Repo Name
3636
id: fix_slashes

‎.github/workflows/linter.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ jobs:
2626
with:
2727
ref: ${{ github.event.pull_request.head.sha }}
2828
- name: Set up Python 3.9
29-
uses: actions/setup-python@v3
29+
uses: actions/setup-python@v4
3030
with:
3131
python-version: '3.9'
3232
- name: Setup env
@@ -66,7 +66,7 @@ jobs:
6666
with:
6767
ref: ${{ github.event.pull_request.head.sha }}
6868
- name: Set up Python 3.9
69-
uses: actions/setup-python@v3
69+
uses: actions/setup-python@v4
7070
with:
7171
python-version: '3.9'
7272
- name: Setup env

‎.github/workflows/linux-test.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ jobs:
8585
rm -rfv "${GITHUB_WORKSPACE}"
8686
mkdir -p "${GITHUB_WORKSPACE}"
8787
echo "::endgroup::"
88-
- uses: actions/checkout@v3
88+
- uses: actions/checkout@v4
8989
with:
9090
# Support the use case where we need to checkout someone's fork
9191
repository: ${{ inputs.test-infra-repository }}
@@ -120,7 +120,7 @@ jobs:
120120
path: /opt/torch-tensorrt-builds/
121121
- name: Download artifacts
122122
if: ${{ matrix.tensorrt != '' }}
123-
uses: actions/download-artifact@v3
123+
uses: actions/download-artifact@v4
124124
with:
125125
name: ${{ env.DOWNLOAD_ARTIFACT_NAME }}
126126
path: /opt/torch-tensorrt-builds/
@@ -184,7 +184,7 @@ jobs:
184184
echo "upload-docs=${upload_docs}" >> "${GITHUB_OUTPUT}"
185185
186186
- name: Upload artifacts to GitHub (if any)
187-
uses: actions/upload-artifact@v3
187+
uses: actions/upload-artifact@v4
188188
if: ${{ inputs.upload-artifact != '' }}
189189
with:
190190
name: ${{ inputs.upload-artifact }}

0 commit comments

Comments
 (0)