BowenBao
60a68477a6
Bump black version to 23.1.0 ( #96578 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96578
Approved by: https://github.com/ezyang
2023-03-15 06:27:59 +00:00
BowenBao
500fd65531
[ONNX] Create common ExportTestCase base class ( #88145 )
...
Refactor out a common base class `ExportTestCase`, for common things in `setUp`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88145
Approved by: https://github.com/justinchuby , https://github.com/abock , https://github.com/AllenTiTaiWang
2022-11-10 21:51:59 +00:00
BowenBao
b0d80f4355
[ONNX] Clarify phrasing of skipScriptTest/skipTraceTest decorators ( #86216 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86216
Approved by: https://github.com/justinchuby , https://github.com/AllenTiTaiWang , https://github.com/abock
2022-10-13 17:20:35 +00:00
BowenBao
45274c56a4
[ONNX] Partially re-enable RoiAlign and RoiPool unit tests ( #86169 )
...
This PR depends on https://github.com/pytorch/vision/pull/6685
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86169
Approved by: https://github.com/justinchuby , https://github.com/AllenTiTaiWang , https://github.com/abock
2022-10-13 14:39:44 +00:00
Li-Huai (Allan) Lin
d9a7e93aaf
[ONNX] Add dtype check in onnx verification ( #79263 )
...
Currently we don't have a dtype check in verifying the consistency between PyTorch and ONNX outputs. As a result, some of dtype inconsistencies were found and reported: #77842 #77845
This is a POC.
Failed workflows:
- [linux-xenial-py3.7-clang7-onnx / test (default, 2, 2, linux.2xlarge)]
- inconsistent shape
- TestONNXRuntime_opset10.test_all (#79371 )
- TestONNXRuntime_opset10.test_any (#79371 )
- TestONNXRuntime_opset10.test_argmin_argmax (#79503 )
- TestONNXRuntime_opset10.test_hardshrink (#79695 )
- TestONNXRuntime_opset10.test_linalg_norm (#79506 )
- TestONNXRuntime_opset10.test_linalg_vector_norm (#79506 )
- TestONNXRuntime_opset10.test_prelu_scalar (#79846 )
- TestONNXRuntime_opset10.test_softshrink (#79695 )
- TestONNXRuntime_opset10.test_sum_empty_tensor (skipped)
- TestONNXRuntime_opset10.test_tolist (skipped)
- inconsistent dtype
- test_arithmetic_prim_bool (skipped)
- test_arithmeticOps_with_low_precision (skipped)
- test_arithmetic_prim_float (skipped)
- test_logical_and (#79339 )
- test_logical_or (#79339 )
- test_logical_xor (#79339 )
- test_pow (skipped)
- test_primitive_input_floating (skipped)
- test_quantize_per_tensor (#79690 )
- test_quantized_adaptive_avg_pool2d (#79690 )
- test_quantized_arithmetic (#79690 )
- test_quantized_arithmetic_qfunctional (#79690 )
- test_quantized_conv2d (#79690 )
- test_quantized_conv2d_relu (#79690 )
- test_quantized_flatten (#79690 )
- test_quantized_hardsigmoid (#79690 )
- test_quantized_hardswish (#79690 )
- test_quantized_linear (#79690 )
- test_quantized_sigmoid (#79690 )
- test_item (skipped)
- test_full_like_value (skipped)
- TestONNXRuntime_opset7.test_div_rounding_mode (skipped)
- TestONNXRuntime_opset8.test_div_rounding_mode (skipped)
- TestONNXRuntime_opset9.test_div_rounding_mode (skipped)
- TestONNXRuntime_opset9_IRv4.test_div_rounding_mode (skipped)
- test_outer (skipped)
- test_symbolic_shape_inference_arange_2 (skipped)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79263
Approved by: https://github.com/justinchuby , https://github.com/BowenBao
2022-08-10 07:14:12 +00:00
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
