@@ -77,10 +77,12 @@ jobs:
7777 trigger-event : ${{ github.event_name }}
7878 architecture : " x86_64"
7979 use-rtx : false
80+ pip-install-torch-extra-args : " --extra-index-url https://pypi.org/simple"
8081
8182 L0-dynamo-converter-tests :
8283 name : L0 dynamo converter tests
8384 needs : [filter-matrix, build]
85+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
8486 strategy :
8587 fail-fast : false
8688 matrix :
@@ -109,6 +111,7 @@ jobs:
109111 L0-dynamo-core-tests :
110112 name : L0 dynamo core tests
111113 needs : [filter-matrix, build]
114+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
112115 strategy :
113116 fail-fast : false
114117 matrix :
@@ -140,6 +143,7 @@ jobs:
140143 L0-py-core-tests :
141144 name : L0 core python tests
142145 needs : [filter-matrix, build]
146+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
143147 strategy :
144148 fail-fast : false
145149 matrix :
@@ -168,6 +172,7 @@ jobs:
168172 L0-torchscript-tests :
169173 name : L0 torchscript tests
170174 needs : [filter-matrix, build]
175+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
171176 strategy :
172177 fail-fast : false
173178 matrix :
@@ -200,6 +205,7 @@ jobs:
200205 L1-dynamo-core-tests :
201206 name : L1 dynamo core tests
202207 needs : [filter-matrix, build, L0-dynamo-converter-tests, L0-dynamo-core-tests, L0-py-core-tests, L0-torchscript-tests]
208+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
203209 strategy :
204210 fail-fast : false
205211 matrix :
@@ -228,6 +234,7 @@ jobs:
228234 L1-dynamo-compile-tests :
229235 name : L1 dynamo compile tests
230236 needs : [filter-matrix, build, L0-dynamo-converter-tests, L0-dynamo-core-tests, L0-py-core-tests, L0-torchscript-tests]
237+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
231238 strategy :
232239 fail-fast : false
233240 matrix :
@@ -256,6 +263,7 @@ jobs:
256263 L1-torch-compile-tests :
257264 name : L1 torch compile tests
258265 needs : [filter-matrix, build, L0-dynamo-converter-tests, L0-dynamo-core-tests, L0-py-core-tests, L0-torchscript-tests]
266+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
259267 strategy :
260268 fail-fast : false
261269 matrix :
@@ -286,6 +294,7 @@ jobs:
286294 L1-torchscript-tests :
287295 name : L1 torch script tests
288296 needs : [filter-matrix, build, L0-dynamo-core-tests, L0-dynamo-converter-tests, L0-py-core-tests, L0-torchscript-tests]
297+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
289298 strategy :
290299 fail-fast : false
291300 matrix :
@@ -318,6 +327,7 @@ jobs:
318327 L2-torch-compile-tests :
319328 name : L2 torch compile tests
320329 needs : [filter-matrix, build, L1-torch-compile-tests, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torchscript-tests]
330+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
321331 strategy :
322332 fail-fast : false
323333 matrix :
@@ -347,6 +357,7 @@ jobs:
347357 L2-dynamo-compile-tests :
348358 name : L2 dynamo compile tests
349359 needs : [filter-matrix, build, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torch-compile-tests, L1-torchscript-tests]
360+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
350361 strategy :
351362 fail-fast : false
352363 matrix :
@@ -376,6 +387,7 @@ jobs:
376387 L2-dynamo-core-tests :
377388 name : L2 dynamo core tests
378389 needs : [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
390+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
379391 strategy :
380392 fail-fast : false
381393 matrix :
@@ -404,6 +416,7 @@ jobs:
404416 L2-dynamo-plugin-tests :
405417 name : L2 dynamo plugin tests
406418 needs : [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
419+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
407420 strategy :
408421 fail-fast : false
409422 matrix :
@@ -426,12 +439,16 @@ jobs:
426439 set -euo pipefail
427440 pushd .
428441 cd tests/py/dynamo
429- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_plugins_tests_results.xml automatic_plugin/
442+ python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
443+ python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml automatic_plugin/test_automatic_plugin.py
444+ python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml automatic_plugin/test_automatic_plugin_with_attrs.py
445+ python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml automatic_plugin/test_flashinfer_rmsnorm.py
430446 popd
431447
432448 L2-torchscript-tests :
433449 name : L2 torch script tests
434450 needs : [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
451+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
435452 strategy :
436453 fail-fast : false
437454 matrix :
@@ -464,6 +481,7 @@ jobs:
464481 L2-dynamo-distributed-tests :
465482 name : L2 dynamo distributed tests
466483 needs : [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
484+ if : ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
467485 strategy :
468486 fail-fast : false
469487 matrix :
0 commit comments