We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1239842 commit 5d561d0Copy full SHA for 5d561d0
torchao/quantization/pt2e/__init__.py
@@ -39,6 +39,8 @@
39
FusedMovingAvgObsFakeQuantize,
40
default_dynamic_fake_quant,
41
default_fake_quant,
42
+ disable_fake_quant,
43
+ disable_observer,
44
enable_fake_quant,
45
enable_observer,
46
)
@@ -114,6 +116,8 @@
114
116
# utils
115
117
"enable_fake_quant",
118
"enable_observer",
119
+ "disable_fake_quant",
120
+ "disable_observer",
121
# export_utils
122
"move_exported_model_to_eval",
123
"move_exported_model_to_train",
torchao/testing/pt2e/utils.py
@@ -78,6 +78,7 @@ def _test_quantizer(
78
m,
79
example_inputs,
80
dynamic_shapes=dynamic_shapes if export_with_dynamic_shape else None,
81
+ strict=True,
82
).module()
83
84
if is_qat:
0 commit comments