Run ONNX tests as part of the standard run_test script

[huydhn]

🐛 Describe the bug

At the moment, ONNX tests are handled in a different way than other Python test using run_test.py. Specifically, it shells out to pytest directly in https://github.com/pytorch/pytorch/blob/master/scripts/onnx/test.sh. The main issue with the current approach is that none of the feature of run_test.py is available to ONNX such as flaky bot, retry, slow test to name a fews. For example, it won't work when I try to disable a flaky ONNX test #98622. Now that run_test.py uses pytest by default, it's time to have onnx as part of it too

Versions

PyTorch CI, ONNX

cc @seemethere @malfet @pytorch/pytorch-dev-infra @ZainRizvi @kit1980 @clee2000

[kit1980]

cc @BowenBao

[huydhn]

Interestingly, I saw an example ONNX flaky failure today https://github.com/pytorch/pytorch/actions/runs/4662975326/jobs/8254048180 where the retry logic ran. But tries failed with the same failure:

=================================== FAILURES ===================================
�[31m�[1m________________ test/onnx/test_fx_to_onnx_with_onnxruntime.py _________________�[0m
[gw1] linux -- Python 3.8.16 /opt/conda/envs/py_3.8/bin/python
worker 'gw1' crashed while running 'test/onnx/test_fx_to_onnx_with_onnxruntime.py::TestFxToOnnxWithOnnxRuntime_op_level_debug_False_dynamic_shapes_True::test_flatten_dynamic_axes'

....
RETRY
...

=================================== FAILURES ===================================
�[31m�[1m________________ test/onnx/test_fx_to_onnx_with_onnxruntime.py _________________�[0m
[gw0] linux -- Python 3.8.16 /opt/conda/envs/py_3.8/bin/python
worker 'gw0' crashed while running 'test/onnx/test_fx_to_onnx_with_onnxruntime.py::TestFxToOnnxWithOnnxRuntime_op_level_debug_False_dynamic_shapes_True::test_flatten_dynamic_axes'

May be this is just a rare case, an infra issue, or the test is just really flaky.

[BowenBao]

cc @titaiwangms maybe we should decorate every dynamic=True test cases

[titaiwangms]

Will do