[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