pytorch
9bb21fac - [ao][sparsity] make sparsity compose with PTQ convert (#74846)

Commit
3 years ago
[ao][sparsity] make sparsity compose with PTQ convert (#74846) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74846 This PR primarily allows the PTQ convert function to work with parametrized modules. Given that the parametrized weight is what is used by default in convert, as long as sparsifier.step() has already been called, the converted model will use the sparisified weights. There is currently no way to handle things if sparsifier.step() has not been called. Lastly, added the is_leaf_or_only_parametrized function because parametrized modules no longer look like leaves due to the parametrizations module attached to them Test Plan: python test/test_ao_sparsity.py TestComposability Imported from OSS Reviewed By: vkuzo Differential Revision: D35240275 fbshipit-source-id: 48529f2a83edfe6d8a2d2dff8ca3d08a3fb0d553 (cherry picked from commit 9d6361482e2885db964e02b0222cd23c9f4d469e)
Author
Committer
Parents
Loading