Skip to content

Commit 08a584b

Browse files
authored
Fix broken fbcode//on_device_ai/Assistant/Jarvis/tests:test_custom_ops - test_quantized_w8a32_conv1d
Differential Revision: D104996877 Pull Request resolved: pytorch#19602
1 parent 933b476 commit 08a584b

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

backends/transforms/quantize_fused_convbn_bias_pass.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -227,9 +227,8 @@ def _quantize_fused_conv_bias(
227227
if input_dequant.target in unsqueeze_targets:
228228
input_dequant = input_dequant.args[0]
229229

230-
assert (
231-
input_dequant.target == dq_per_tensor
232-
), f"Expected dequantize_per_tensor, got {input_dequant.target}"
230+
if input_dequant.target != dq_per_tensor:
231+
continue
233232

234233
bias_val = bias_node.meta.get("val")
235234
dequant_val = (

0 commit comments

Comments
 (0)