Skip to content

Commit 6d27aaf

Browse files
committed
NXP backend: Condition fused bias quantization pass to QAT flow only
1 parent 405d85f commit 6d27aaf

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

backends/nxp/quantizer/utils.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -219,8 +219,9 @@ def calibrate_and_quantize(
219219

220220
m = convert_pt2e(m)
221221

222-
m = QuantizeFusedConvBnBiasAtenPass(default_zero_bias=False, symmetric_quant=True)(
223-
m
224-
).graph_module
222+
if is_qat:
223+
m = QuantizeFusedConvBnBiasAtenPass(
224+
default_zero_bias=False, symmetric_quant=True
225+
)(m).graph_module
225226

226227
return m

0 commit comments

Comments
 (0)