pytorch
1c21cd22 - [quant][pt2e][refactor] Add input_output_share_observers to node.meta["target_dtype_info"] (#97949)

Commit
1 year ago
[quant][pt2e][refactor] Add input_output_share_observers to node.meta["target_dtype_info"] (#97949) Summary: The goal for this PR is to unify the flow of information to reduce fragmentation of implementations between fx graph mode quantization and quantize_pt2e, since quantize_pt2e will be using node.meta to store this information, we'd like to make sure fx graph mode quantization get this information from the same place Test Plan: python test/test_quantization.py TestQuantizeFx Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/97949 Approved by: https://github.com/andrewor14
Author
Committer
Parents
Loading