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 |
|