Justin Chu
773d80747c
[ONNX] Clean up unit tests, rename files and improve import style ( #81141 )
...
- Rename `test_pytorch_common` -> `pytorch_test_common`, `test_onnx_common` -> `onnx_test_common`, removing the test_ prefix to show that the files are not test cases
- Remove import * in `test_pytorch_common` and adjust to import from `testing._internal.common_utils` (where functions are actually defined) instead
- Import modules only in `test_pytorch_onnx_onnxruntime` (too many to handle in a single PR in other tests) (The skips are exceptions)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81141
Approved by: https://github.com/BowenBao
2022-07-12 00:00:49 +00:00
Jane Xu
d2fbfe7fce
[ONNX] subscribe onnx to our custom test infra ( #79546 )
...
Remove as many references as can be easily done of unittest in favor of our custom infra.
Left a todo where I could not easily replace unittest.main with run_tests()
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79546
Approved by: https://github.com/seemethere
2022-06-15 15:00:04 +00:00
BowenBao
6883b0ce9f
[ONNX][WIP] Refactor verification.py
...
* Rename arguments, code clean up.
* Refactor functions to smaller reusable functions.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77289
Approved by: https://github.com/justinchuby , https://github.com/garymm
2022-05-31 18:49:39 +00:00
BowenBao
483bb4f0cb
[ONNX] Extract export verification as standalone api from unittest
...
The verification logic is refactored and extracted from
`test_pytorch_onnx_onnxruntime.py`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76929
Approved by: https://github.com/justinchuby , https://github.com/garymm
2022-05-26 19:49:19 +00:00
Justin Chu
5dd1c67776
[ONNX] Format ONNX python with black
...
Format all onnx python code with black and isort with
```sh
isort torch/onnx/ test/onnx
black torch/onnx/ test/onnx
```
Updated lintrunner config to include these paths.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76754
Approved by: https://github.com/suo , https://github.com/BowenBao
2022-05-05 00:19:22 +00:00
BowenBao
1793ab0bd2
[ONNX] Initial test suite for Torch IR to ONNX
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74492
Approved by: https://github.com/shubhambhokare1 , https://github.com/garymm
2022-04-11 19:28:28 +00:00
