Skip to content

Commit

Permalink
Update on "[quant][fix] Fix quant type classification for float_qpara…
Browse files Browse the repository at this point in the history
…m qconfig"

Summary:
also renamed float_qparam_dynamic_qconfig to float_qparam_weight_only_qconfig
It's not used in user code yet so we only need to update the tests.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: [D25010175](https://our.internmc.facebook.com/intern/diff/D25010175)

[ghstack-poisoned]
  • Loading branch information
jerryzh168 committed Nov 18, 2020
1 parent 618459f commit f35cdc6
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions torch/quantization/qconfig.py
Expand Up @@ -67,8 +67,11 @@ def __new__(cls, activation=torch.nn.Identity, weight=torch.nn.Identity):
per_channel_dynamic_qconfig = QConfigDynamic(activation=default_dynamic_quant_observer,
weight=default_per_channel_weight_observer)

float_qparams_weight_only_qconfig = QConfig(activation=default_placeholder_observer,
weight=default_float_qparams_observer)
# TODO: this is weight only quant, change this to QConfigWeightOnly
# or remove the QConfigDynamic later
float_qparams_weight_only_qconfig = QConfigDynamic(
activation=default_placeholder_observer,
weight=default_float_qparams_observer)

default_qat_qconfig = QConfig(activation=default_fake_quant,
weight=default_weight_fake_quant)
Expand Down

0 comments on commit f35cdc6

Please sign in to comment.