pytorch
dd667b6e - [quant][fx] Move all fusion registrations to backend_config_dict (#75318)

Commit
2 years ago
[quant][fx] Move all fusion registrations to backend_config_dict (#75318) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/75318 This PR moves the registrations for fusion patterns to backend_config_dict Also fixed one issue in numeric suite graph matcher, since now (torch.nn.ReLU, torch.nn.BatchNorm3d) would appear in quant patterns, (previously only in fusion pattern), and we need to match sure (torch.nn.ReLU, (torch.nn.BatchNorm3d, torch.nn.Conv3d)) can match before (torch.nn.ReLU, torch.nn.BatchNorm3d), but previously, it looks like (torch.nn.ReLU, (torch.nn.BatchNorm3d, torch.nn.Conv3d)) is not really matched since `end_node_matches_reversed_fusion` is expecting a flattened pattern like (torch.nn.ReLU, torch.nn.BatchNorm3d, torch.nn.Conv3d), for now we'll manually flatten this pattern, but in the future I think we might want to use the matching function `is_match` under torch.ao.quantization.fx.match_utils to do this matching. Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps python test/test_quantization.py TestFXNumericSuiteCoreAPIs Imported from OSS Reviewed By: vkuzo, andrewor14 Differential Revision: D35423788 fbshipit-source-id: a54093ccebae9c59aeee9399669ddb2c48bfb9aa (cherry picked from commit 6a55ea8eb2740cedafb9972888fedf68e927586d)
Author
Committer
Parents
Loading