pytorch
bfe6e232 - Early version of fx graph matcher for NS (#51588)

Commit
3 years ago
Early version of fx graph matcher for NS (#51588) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51588 Early version of utility to match nodes between graph A and graph B, for Numerical Suite for FX graph mode quantization. The main goal of this utility is to reliably match the nodes of graph A to the nodes of graph B, and throw an easy to read error message. This will be used in future PRs to create the APIs for matching activations. It also could potentially be used to match weights. Test Plan: For now, we have bare bones test coverage on some toy models, and a single torchvision model. ``` python test/test_quantization.py TestFXGraphMatcher ``` Future PRs will add more testing. Imported from OSS Reviewed By: jerryzh168 Differential Revision: D26403093 fbshipit-source-id: 60e318d51e6fefe65265488c4967629d946048ef
Author
Parents
Loading