pytorch
dc8b5db5 - [quant][graphmode] relax the constraint for supported_dtypes for reference option (Linear and Conv) (#62348)

Commit
3 years ago
[quant][graphmode] relax the constraint for supported_dtypes for reference option (Linear and Conv) (#62348) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62348 Originally we have a supported_dtypes check for linear and conv, but it's only valid for non reference option, this PR removes the constraint when is_reference=True and enables producing reference patterns for the dtype combinations that's not supported by fbgemm/qnnpack, for example qint8 activation dtypes Test Plan: python test/test_quantization.py TestQuantizeFx.test_linear_qint8_activation Imported from OSS Reviewed By: vkuzo Differential Revision: D29968675 fbshipit-source-id: 2abe37940eb62e16fcf0cbb700c174de49719223
Author
Parents
Loading