[pruner] amend base pruner API to match base sparsifier (#63178)
authorKaren Zhou <kazhou@fb.com>
Tue, 24 Aug 2021 17:17:28 +0000 (10:17 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 24 Aug 2021 17:25:43 +0000 (10:25 -0700)
commit16ba20507a7a8fcb62d88c719eceab578e09e210
treea7d5961f2696f0e6f6dad54541266ff5acb93f9c
parent5dee15401c7730ae2a0e49e774c9da2d067527b7
[pruner] amend base pruner API to match base sparsifier (#63178)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63178

Update base pruner API to match base sparsifier API as defined in D28970960 / PR58955

Changes include:
- `enable_mask_update = True` in `__init__`
- `prepare` takes model and config instead of constructor
- convert functionality renamed to `squash_mask`, `convert` method call now raises Error
- `activation_handles` ad `bias_handles` initialized in `_prepare` instead of constructor
ghstack-source-id: 136467595

Test Plan:
Function names updates according to changes

`buck test mode/dev-nosan //caffe2/test:ao -- TestBasePruner`

https://pxl.cl/1MTgH

TODO will need to modify `fbcode/scripts/kazhou/fusion_tests.py` to use new API

Reviewed By: z-a-f

Differential Revision: D30287179

fbshipit-source-id: d4727bea1873b500f2d4bb784db26d532bf26cce
test/ao/sparsity/test_pruner.py
torch/ao/sparsity/__init__.py
torch/ao/sparsity/experimental/pruner/base_pruner.py
torch/ao/sparsity/sparsifier/base_sparsifier.py
torch/ao/sparsity/sparsifier/utils.py