|
33 | 33 | - uses: actions/setup-python@v4 |
34 | 34 | with: |
35 | 35 | python-version: '3.10' |
36 | | - - uses: actions/checkout@v3 |
| 36 | + - uses: actions/checkout@v4 |
37 | 37 | with: |
38 | 38 | repository: pytorch/tensorrt |
39 | 39 | - name: Generate release matrix |
@@ -136,7 +136,9 @@ jobs: |
136 | 136 | export USE_HOST_DEPS=1 |
137 | 137 | export CI_BUILD=1 |
138 | 138 | pushd . |
139 | | - cd tests/py/dynamo |
| 139 | + cd tests/py |
| 140 | + python -m pip install -r requirements.txt |
| 141 | + cd dynamo |
140 | 142 | python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/ |
141 | 143 | popd |
142 | 144 |
|
@@ -165,7 +167,9 @@ jobs: |
165 | 167 | export USE_HOST_DEPS=1 |
166 | 168 | export CI_BUILD=1 |
167 | 169 | pushd . |
168 | | - cd tests/py/dynamo |
| 170 | + cd tests/py |
| 171 | + python -m pip install -r requirements.txt |
| 172 | + cd dynamo |
169 | 173 | python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/ |
170 | 174 | popd |
171 | 175 |
|
@@ -194,7 +198,9 @@ jobs: |
194 | 198 | export USE_HOST_DEPS=1 |
195 | 199 | export CI_BUILD=1 |
196 | 200 | pushd . |
197 | | - cd tests/py/dynamo |
| 201 | + cd tests/py |
| 202 | + python -m pip install -r requirements.txt |
| 203 | + cd dynamo |
198 | 204 | python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py |
199 | 205 | python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/reexport_test_results.xml --ir dynamo models/test_reexport.py |
200 | 206 | popd |
@@ -224,7 +230,9 @@ jobs: |
224 | 230 | export USE_HOST_DEPS=1 |
225 | 231 | export CI_BUILD=1 |
226 | 232 | pushd . |
227 | | - cd tests/py/dynamo |
| 233 | + cd tests/py |
| 234 | + python -m pip install -r requirements.txt |
| 235 | + cd dynamo |
228 | 236 | python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/ |
229 | 237 | 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 | 238 | 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: |
255 | 263 | export USE_HOST_DEPS=1 |
256 | 264 | export CI_BUILD=1 |
257 | 265 | pushd . |
258 | | - cd tests/py/dynamo |
| 266 | + cd tests/py |
| 267 | + python -m pip install -r requirements.txt |
| 268 | + cd dynamo |
259 | 269 | 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 | 270 | python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/ |
261 | 271 | 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: |
286 | 296 | export USE_HOST_DEPS=1 |
287 | 297 | export CI_BUILD=1 |
288 | 298 | pushd . |
289 | | - cd tests/py/dynamo |
| 299 | + cd tests/py |
| 300 | + python -m pip install -r requirements.txt |
| 301 | + cd dynamo |
290 | 302 | nvidia-smi |
291 | 303 | 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 | 304 | 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 |
|
0 commit comments