pytorch
02e30a09 - [ao][sparsity] make sparsity and PTQ compose (#74845)

Commit
3 years ago
[ao][sparsity] make sparsity and PTQ compose (#74845) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74845 This PR adds support for quantization flow to detect parametrized modules and match them using their original module types. This mainly involved using the new type_before_parametrizations function rather than type to check for module mathcing Test Plan: python test/test_ao_sparsity.py TestComposability Imported from OSS Reviewed By: jerryzh168 Differential Revision: D35240274 fbshipit-source-id: 7294d89c9c2e069e51d8b9bafa45c15f92bed124 (cherry picked from commit ed5cdb7b636c42e040d1b4a67b6b94604d06e1ff)
Author
Committer
Parents
Loading