pytorch
b347b8c1 - [quant][fx] Support some default ops in the native backend config (#74600)

Commit
2 years ago
[quant][fx] Support some default ops in the native backend config (#74600) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74600 Following https://github.com/pytorch/pytorch/pull/74210, this PR adds the support for some ops using the DefaultNodeQuantizeHandler in the backend_config_dict defintion for pytorch native backend TODO: There is still a few ops we didn't handle with backend_config_dict path: gelu and softmax, need to discuss if we still need them, if so we can change the test to use backend_config_dict and remove the DefaultNodeQuantizeHandler after that Test Plan: python test/test_quantization.py TestQuantizeFxOps Imported from OSS Reviewed By: andrewor14 Differential Revision: D35071437 fbshipit-source-id: 70351d2810ca1ac7dc09d4a9c239f6757ccb51ca (cherry picked from commit 5e68f755a32ba7d90d6c73db9c2017f9c58d7fa5)
Author
Committer
Parents
Loading