Skip to content

Commit

Permalink
[quant][pt2e] Disable remove_qconfig (pytorch#111000)
Browse files Browse the repository at this point in the history
Summary:
This is a hacky flag that we had before in fx flow, and we don't want this in the new flow

Test Plan:
python test/test_quantization.py TestQuantizePT2E

Reviewers:

Subscribers:

Tasks:

Tags:
Pull Request resolved: pytorch#111000
Approved by: https://github.com/andrewor14
  • Loading branch information
jerryzh168 authored and pytorchmergebot committed Oct 11, 2023
1 parent cf1da9b commit d589106
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions torch/ao/quantization/quantize_fx.py
Original file line number Diff line number Diff line change
Expand Up @@ -655,7 +655,6 @@ def convert_to_reference_fx(
def _convert_to_reference_decomposed_fx(
graph_module: GraphModule,
convert_custom_config: Union[ConvertCustomConfig, Dict[str, Any], None] = None,
_remove_qconfig: bool = True,
qconfig_mapping: Union[QConfigMapping, Dict[str, Any], None] = None,
backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
) -> GraphModule:
Expand Down Expand Up @@ -699,7 +698,7 @@ def _convert_to_reference_decomposed_fx(
graph_module,
is_reference=True,
convert_custom_config=convert_custom_config,
_remove_qconfig=_remove_qconfig,
_remove_qconfig=False,
qconfig_mapping=qconfig_mapping,
backend_config=backend_config,
is_decomposed=True,
Expand Down

0 comments on commit d589106

Please sign in to comment.