pytorch
5937c48f - dbr quant function fusion [1/x]: record matches for functions (#71764)

Commit
2 years ago
dbr quant function fusion [1/x]: record matches for functions (#71764) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71764 For DBR quant, adds the code for matching seen ops to function fusion patterns. After we have the full DAG, we have a separate pass over the dag and add matched fusion patterns to the seen op data structure. This is the first PR in the stack which implements matching and recording the match results. Future PRs in this stack will use the match results to modify observer insertion and inference. Test Plan: ``` python test/test_quantization.py TestQuantizeDBR.test_fusion_functions ``` Reviewed By: jerryzh168 Differential Revision: D33775098 Pulled By: vkuzo fbshipit-source-id: 488aac902bf568d41c863ee49248990411ed9c53 (cherry picked from commit 4ad1ca1abc60af813805497573c097759184e293)
Author
Committer
Parents
Loading