Skip to content

Commit d6be4ba

Browse files
zewenli98zewenli98
authored
fix: torch 2.7 bump bug on the main branch (#3353)
1 parent 543bc9b commit d6be4ba

22 files changed

+121
-77
lines changed

‎.github/scripts/generate_binary_build_matrix.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@
6464
XPU = "xpu"
6565

6666

67-
CURRENT_NIGHTLY_VERSION = "2.6.0"
67+
CURRENT_NIGHTLY_VERSION = "2.7.0"
6868
CURRENT_CANDIDATE_VERSION = "2.5.1"
6969
CURRENT_STABLE_VERSION = "2.5.1"
7070
CURRENT_VERSION = CURRENT_STABLE_VERSION

‎.github/workflows/assigner.yml‎

Lines changed: 1 addition & 1 deletion
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‎

Lines changed: 3 additions & 3 deletions
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‎

Lines changed: 2 additions & 2 deletions
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‎

Lines changed: 20 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,10 @@ jobs:
3030
matrix: ${{ steps.generate.outputs.matrix }}
3131
runs-on: ubuntu-latest
3232
steps:
33-
- uses: actions/setup-python@v4
33+
- uses: actions/setup-python@v5
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‎

Lines changed: 20 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,10 @@ jobs:
2727
matrix: ${{ steps.generate.outputs.matrix }}
2828
runs-on: ubuntu-latest
2929
steps:
30-
- uses: actions/setup-python@v4
30+
- uses: actions/setup-python@v5
3131
with:
3232
python-version: '3.10'
33-
- uses: actions/checkout@v3
33+
- uses: actions/checkout@v4
3434
with:
3535
repository: pytorch/tensorrt
3636
- name: Generate tensorrt matrix
@@ -132,7 +132,9 @@ jobs:
132132
export USE_HOST_DEPS=1
133133
export CI_BUILD=1
134134
pushd .
135-
cd tests/py/dynamo
135+
cd tests/py
136+
python -m pip install -r requirements.txt
137+
cd dynamo
136138
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
137139
popd
138140
@@ -161,7 +163,9 @@ jobs:
161163
export USE_HOST_DEPS=1
162164
export CI_BUILD=1
163165
pushd .
164-
cd tests/py/dynamo
166+
cd tests/py
167+
python -m pip install -r requirements.txt
168+
cd dynamo
165169
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
166170
popd
167171
@@ -190,7 +194,9 @@ jobs:
190194
export USE_HOST_DEPS=1
191195
export CI_BUILD=1
192196
pushd .
193-
cd tests/py/dynamo
197+
cd tests/py
198+
python -m pip install -r requirements.txt
199+
cd dynamo
194200
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
195201
popd
196202
@@ -219,7 +225,9 @@ jobs:
219225
export USE_HOST_DEPS=1
220226
export CI_BUILD=1
221227
pushd .
222-
cd tests/py/dynamo
228+
cd tests/py
229+
python -m pip install -r requirements.txt
230+
cd dynamo
223231
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
224232
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
225233
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
@@ -250,7 +258,9 @@ jobs:
250258
export USE_HOST_DEPS=1
251259
export CI_BUILD=1
252260
pushd .
253-
cd tests/py/dynamo
261+
cd tests/py
262+
python -m pip install -r requirements.txt
263+
cd dynamo
254264
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/
255265
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
256266
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
@@ -281,7 +291,9 @@ jobs:
281291
export USE_HOST_DEPS=1
282292
export CI_BUILD=1
283293
pushd .
284-
cd tests/py/dynamo
294+
cd tests/py
295+
python -m pip install -r requirements.txt
296+
cd dynamo
285297
nvidia-smi
286298
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
287299
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‎

Lines changed: 20 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,10 @@ jobs:
2727
matrix: ${{ steps.generate.outputs.matrix }}
2828
runs-on: ubuntu-latest
2929
steps:
30-
- uses: actions/setup-python@v4
30+
- uses: actions/setup-python@v5
3131
with:
3232
python-version: '3.10'
33-
- uses: actions/checkout@v3
33+
- uses: actions/checkout@v4
3434
with:
3535
repository: pytorch/tensorrt
3636
- name: Generate tensorrt matrix
@@ -135,7 +135,9 @@ jobs:
135135
export USE_HOST_DEPS=1
136136
export CI_BUILD=1
137137
pushd .
138-
cd tests/py/dynamo
138+
cd tests/py
139+
python -m pip install -r requirements.txt
140+
cd dynamo
139141
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
140142
popd
141143
@@ -161,7 +163,9 @@ jobs:
161163
export USE_HOST_DEPS=1
162164
export CI_BUILD=1
163165
pushd .
164-
cd tests/py/dynamo
166+
cd tests/py
167+
python -m pip install -r requirements.txt
168+
cd dynamo
165169
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
166170
popd
167171
@@ -187,7 +191,9 @@ jobs:
187191
export USE_HOST_DEPS=1
188192
export CI_BUILD=1
189193
pushd .
190-
cd tests/py/dynamo
194+
cd tests/py
195+
python -m pip install -r requirements.txt
196+
cd dynamo
191197
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
192198
popd
193199
@@ -213,7 +219,9 @@ jobs:
213219
export USE_HOST_DEPS=1
214220
export CI_BUILD=1
215221
pushd .
216-
cd tests/py/dynamo
222+
cd tests/py
223+
python -m pip install -r requirements.txt
224+
cd dynamo
217225
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
218226
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
219227
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
@@ -241,7 +249,9 @@ jobs:
241249
export USE_HOST_DEPS=1
242250
export CI_BUILD=1
243251
pushd .
244-
cd tests/py/dynamo
252+
cd tests/py
253+
python -m pip install -r requirements.txt
254+
cd dynamo
245255
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/
246256
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
247257
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
@@ -269,7 +279,9 @@ jobs:
269279
export USE_HOST_DEPS=1
270280
export CI_BUILD=1
271281
pushd .
272-
cd tests/py/dynamo
282+
cd tests/py
283+
python -m pip install -r requirements.txt
284+
cd dynamo
273285
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
274286
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
275287
popd

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

Lines changed: 18 additions & 6 deletions
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‎

Lines changed: 1 addition & 1 deletion
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‎

Lines changed: 2 additions & 2 deletions
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@v5
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@v5
7070
with:
7171
python-version: '3.9'
7272
- name: Setup env

0 commit comments

Comments
 (0)