pytorch
cebf08af - [Quant] Remove weight from DTypeConfig for non-weighted ops (#86335)

Commit
2 years ago
[Quant] Remove weight from DTypeConfig for non-weighted ops (#86335) Summary: Weight dtypes should be specified only for weighted ops like conv and linear. This commit removes weight dtypes from the DTypeConfigs used in binary ops and fixed qparams ops. Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps Reviewers: jerryzh168, vkuzo Subscribers: jerryzh168, vkuzo Pull Request resolved: https://github.com/pytorch/pytorch/pull/86335 Approved by: https://github.com/vkuzo
Author
Committer
Parents
Loading