Skip to content

Commit f490656

Browse files
committed
NXP backend: Add quantization of fused biassless convolution to integration tests pipeline
1 parent 37b12c8 commit f490656

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

backends/nxp/tests_models/utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@
2626
)
2727
from executorch.backends.nxp.quantizer.neutron_quantizer import NeutronQuantizer
2828
from executorch.backends.nxp.tests_models.model_input_spec import ModelInputSpec
29+
from executorch.backends.transforms.quantize_fused_convbn_bias_pass import (
30+
QuantizeFusedConvBnBiasAtenPass,
31+
)
2932
from executorch.devtools.visualization.visualization_utils import (
3033
visualize_with_clusters,
3134
)
@@ -138,6 +141,11 @@ def to_quantized_edge_program(
138141

139142
exir_program_aten_quant = convert_pt2e(m)
140143

144+
if use_qat:
145+
exir_program_aten_quant = QuantizeFusedConvBnBiasAtenPass(
146+
default_zero_bias=True
147+
)(exir_program_aten_quant).graph_module
148+
141149
# To ATen
142150
core_aten_ep = _to_core_aten(
143151
exir_program_aten_quant, example_input, None, verbose=True

0 commit comments

Comments
 (0)