Skip to content

Commit 7ea502c

Browse files
committed
NXP backend: Condition fused bias quantization pass to QAT flow only
1 parent 35638e5 commit 7ea502c

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

backends/nxp/quantizer/utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,7 @@ def calibrate_and_quantize(
219219

220220
m = convert_pt2e(m)
221221

222-
m = QuantizeFusedConvBnBiasAtenPass(default_zero_bias=True)(m).graph_module
222+
if is_qat:
223+
m = QuantizeFusedConvBnBiasAtenPass(default_zero_bias=False)(m).graph_module
223224

224225
return m

0 commit comments

Comments
 (0)