Skip to content

Commit

Permalink
Update base for Update on "[ONNX] Add test_fx_op_consistency.py"
Browse files Browse the repository at this point in the history
Should be merged after #99434

<!--
copilot:all
-->
### <samp>馃 Generated by Copilot at f59c428</samp>

### Summary
馃摑馃攧馃殌

<!--
1.  馃摑 - This emoji represents the improvement of the documentation and type annotation of the ONNX exporter and its tests.
2.  馃攧 - This emoji represents the renaming of the `dont_care` function to `skip` and the update of the references and documentation. This change reflects a refactoring and improvement of the code quality and readability.
3.  馃殌 - This emoji represents the enhancement of the ONNX exporter's ability to handle more input and output types for PyTorch models. This change increases the performance and functionality of the exporter.
-->
This pull request enhances the ONNX exporter and its tests to handle more input and output types for PyTorch models. It improves the type annotation, tolerance handling, and documentation of the exporter and its test functions. It also renames a test function to make the code more consistent and clear.

> _To export PyTorch models with ONNX_
> _We need to handle various contexts_
> _We improved the annotation_
> _And the tolerance function_
> _And renamed `dont_care` to `skip` for the tests_

### Walkthrough
*  Expand the `_InputArgsType` type annotation to include int, float, and bool types, in addition to torch.Tensor, Sequence, and Mapping, to support more types of inputs for PyTorch models ([link](https://github.com/pytorch/pytorch/pull/99465/files?diff=unified&w=0#diff-c8fa56eefd7f98fb4f9739d57df57f02ede77e28528133736010a6d06651ebcbL44-R46), [link](https://github.com/pytorch/pytorch/pull/99465/files?diff=unified&w=0#diff-0795f54fd1f38cfbf2c4a863a4efc9f40f2ea020a2b1612605c361b8d8d35862L197-R199), [link](https://github.com/pytorch/pytorch/pull/99465/files?diff=unified&w=0#diff-0795f54fd1f38cfbf2c4a863a4efc9f40f2ea020a2b1612605c361b8d8d35862L291-R295)).



[ghstack-poisoned]
  • Loading branch information
titaiwangms committed Apr 27, 2023
1 parent e69163c commit dc8467e
Showing 0 changed files with 0 additions and 0 deletions.

0 comments on commit dc8467e

Please sign in to comment.