Skip to content

Commit

Permalink
size-0
Browse files Browse the repository at this point in the history
  • Loading branch information
justinchuby committed Jul 30, 2024
1 parent 8df0edb commit 78f8557
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions tests/function_libs/torch_lib/ops_test_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -713,19 +713,21 @@ def _where_input_wrangler(
dtypes=(torch.bool,),
reason="fixme: ORT does not implement SplitToSequence for bool inputs: https://github.com/microsoft/onnxruntime/issues/16905",
),
TorchLibOpInfo("clamp_max", core_ops.aten_clamp).skip(
enabled_if=sys.version_info[:2] >= (3, 9) or sys.platform != "win32",
reason="fails in this particular case",
),
TorchLibOpInfo("clamp_max", core_ops.aten_clamp_max).skip(
matcher=lambda sample: len(sample.input.shape) == 0,
enabled_if=version_utils.onnxruntime_older_than("1.16"),
reason="fixme (core dump): ORT aborts on scalar inputs to Reduce*-18. https://github.com/microsoft/onnxruntime/issues/16492",
).skip(
reason="Size 0 inputs are not handled by design",
matcher=lambda sample: sample.input.numel() == 0,
),
TorchLibOpInfo("clamp_min", core_ops.aten_clamp_min).skip(
matcher=lambda sample: len(sample.input.shape) == 0,
enabled_if=version_utils.onnxruntime_older_than("1.16"),
reason="fixme (core dump): ORT aborts on scalar inputs to Reduce*-18. https://github.com/microsoft/onnxruntime/issues/16492",
).skip(
reason="Size 0 inputs are not handled by design",
matcher=lambda sample: sample.input.numel() == 0,
),
TorchLibOpInfo("clone", core_ops.aten_clone),
TorchLibOpInfo("complex", core_ops.aten_complex),
Expand Down

0 comments on commit 78f8557

Please sign in to comment.