Skip to content

Commit b20ae85

Browse files
committed
NXP backend: Add quantization of fused biassless convolution to integration tests pipeline
1 parent 4ac044b commit b20ae85

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

backends/nxp/tests_models/utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,9 @@
3333
)
3434
from executorch.backends.nxp.quantizer.neutron_quantizer import NeutronQuantizer
3535
from executorch.backends.nxp.tests_models.model_input_spec import ModelInputSpec
36+
from executorch.backends.transforms.quantize_fused_convbn_bias_pass import (
37+
QuantizeFusedConvBnBiasAtenPass,
38+
)
3639
from executorch.devtools.visualization.visualization_utils import (
3740
visualize_with_clusters,
3841
)
@@ -149,6 +152,11 @@ def to_quantized_edge_program(
149152

150153
exir_program_aten_quant = convert_pt2e(m)
151154

155+
if use_qat:
156+
exir_program_aten_quant = QuantizeFusedConvBnBiasAtenPass(
157+
default_zero_bias=True
158+
)(exir_program_aten_quant).graph_module
159+
152160
# To ATen
153161
core_aten_ep = _to_core_aten(
154162
exir_program_aten_quant, example_input, None, verbose=True

0 commit comments

Comments
 (0)