pytorch
d1445945 - [Quant][fx] Remove WEIGHT_INDEX_DICT and BIAS_INDEX_DICT (Part 2) (#83853)

Commit
3 years ago
[Quant][fx] Remove WEIGHT_INDEX_DICT and BIAS_INDEX_DICT (Part 2) (#83853) Summary: - Finishes the second part of https://github.com/pytorch/pytorch/pull/83263 - Removes WEIGHT_INDEX_DICT and BIAS_INDEX_DICT from utils.py - Moves two funcitons, `node_arg_is_weight` and `node_arg_is_bias` into utils.py from prepare.py convert.py and _equalize.py now use node_arg_is_weight instead of the dictionaries - Adds in quantization support for `F.groupnorm`. Add in missing BackendPatternConfigs for layernorm, instancenorm, and groupnorm Test Plan: ``` python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps ``` Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: 2b157e0dc4f1553be1f4813b4693db952e6fc558 Pull Request resolved: https://github.com/pytorch/pytorch/pull/83848 Fixes #83093 Pull Request resolved: https://github.com/pytorch/pytorch/pull/83853 Approved by: https://github.com/jerryzh168, https://github.com/andrewor14
Author
Committer
Parents
Loading