Skip to content

Commit 5d561d0

Browse files
metascroyfacebook-github-bot
authored andcommitted
Migrate xnnpack/vulkan/boltnn pt2e from torch.ao to torchao (#2302)
Summary: X-link: pytorch/executorch#11363 Reviewed By: jerryzh168 Differential Revision: D75492104
1 parent 1239842 commit 5d561d0

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

torchao/quantization/pt2e/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,8 @@
3939
FusedMovingAvgObsFakeQuantize,
4040
default_dynamic_fake_quant,
4141
default_fake_quant,
42+
disable_fake_quant,
43+
disable_observer,
4244
enable_fake_quant,
4345
enable_observer,
4446
)
@@ -114,6 +116,8 @@
114116
# utils
115117
"enable_fake_quant",
116118
"enable_observer",
119+
"disable_fake_quant",
120+
"disable_observer",
117121
# export_utils
118122
"move_exported_model_to_eval",
119123
"move_exported_model_to_train",

torchao/testing/pt2e/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ def _test_quantizer(
7878
m,
7979
example_inputs,
8080
dynamic_shapes=dynamic_shapes if export_with_dynamic_shape else None,
81+
strict=True,
8182
).module()
8283

8384
if is_qat:

0 commit comments

Comments
 (0)