pytorch
23bcab19 - [quant][refactor] Refactor find_matches for easier future extension (#74878)

Commit
3 years ago
[quant][refactor] Refactor find_matches for easier future extension (#74878) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74878 Previously we record the matched node as a list of nodes: `List[Node]`, this does not generalize to a graph, which is needed for future use cases, in this PR we changed the recorded node as NodePattern instead, currently defined as ``` NodePattern = Union[Tuple[Node, Node], Tuple[Node, Tuple[Node, Node]], Any] ``` but can be more general. This will allow us to support more general patterns with backend_config_dict api, and is also needed for BinaryOpQuantizeHandler refactor Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps Imported from OSS Reviewed By: vkuzo Differential Revision: D35203616 fbshipit-source-id: f4bf5b056cfc0955455eea9c2bf1ac9f6dde3974 (cherry picked from commit b290c047e1861bbb62fb1bb576761e801b210220)
Author
Committer
Parents
Loading