Skip to content

Commit 1707e14

Browse files
committed
add fake_quantize_per_channel_affine expected failure case for non-integer zero_point
1 parent e76a616 commit 1707e14

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -698,7 +698,10 @@ def _where_input_wrangler(
698698
TorchLibOpInfo("special.erfcx", special_ops.aten_special_erfcx).xfail(
699699
reason="fixme: The implementation is numerically unstable: https://github.com/microsoft/onnxscript/issues/1223"
700700
),
701-
TorchLibOpInfo("fake_quantize_per_channel_affine", core_ops.aten_fake_quantize_per_channel_affine),
701+
TorchLibOpInfo("fake_quantize_per_channel_affine", core_ops.aten_fake_quantize_per_channel_affine).xfail(
702+
reason="fixme: ONNX (De)QuantizeLinear only supports integer zero_point values",
703+
matcher=lambda sample: sample.args[1].dtype != torch.int32,
704+
),
702705
TorchLibOpInfo("fake_quantize_per_tensor_affine", core_ops.aten_fake_quantize_per_tensor_affine),
703706
TorchLibOpInfo("fill", core_ops.aten_fill),
704707
TorchLibOpInfo("flip", core_ops.aten_flip).skip(

0 commit comments

Comments
 (0)