Skip to content

Commit b4bec49

Browse files
committed
enable tests for fake_quantize_per_tensor_affine
1 parent 5db4954 commit b4bec49

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -699,6 +699,7 @@ def _where_input_wrangler(
699699
reason="fixme: The implementation is numerically unstable: https://github.com/microsoft/onnxscript/issues/1223"
700700
),
701701
TorchLibOpInfo("fake_quantize_per_channel_affine", core_ops.aten_fake_quantize_per_channel_affine),
702+
TorchLibOpInfo("fake_quantize_per_tensor_affine", core_ops.aten_fake_quantize_per_tensor_affine),
702703
TorchLibOpInfo("fill", core_ops.aten_fill),
703704
TorchLibOpInfo("flip", core_ops.aten_flip).skip(
704705
reason="fixme: size 0 inputs are not handled yet",

0 commit comments

Comments
 (0)