Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update and have more specific xfail
  • Loading branch information
titaiwangms committed Jul 18, 2023
commit a425eb901b81fab76445237f9dde5dc2d1881b4e
14 changes: 10 additions & 4 deletions onnxscript/tests/function_libs/torch_lib/ops_test_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -1406,8 +1406,14 @@ def _where_input_wrangler(
reason="ONNX doesn't support divisor_override argument",
)
.xfail(
matcher=lambda sample: sample.kwargs.get("ceil_mode", True),
reason="ONNXRUNTIME doesn't match PyTorch when ceil_mode=True until opset 19",
matcher=lambda sample: (sample.kwargs.get("ceil_mode") is True)
and (
sample.kwargs.get("count_include_pad") is True
or sample.input.shape[2]
% (sample.args[0][0] if isinstance(sample.args[0], tuple) else sample.args[0])
!= 0
),
reason="fixme: ORT doesn't match PyTorch when ceil_mode=True until opset 19",
),
TorchLibOpInfo(
"nn.functional.avg_pool2d",
Expand All @@ -1431,8 +1437,8 @@ def _where_input_wrangler(
reason="ONNX doesn't support divisor_override argument",
)
.xfail(
matcher=lambda sample: sample.kwargs.get("ceil_mode", True),
reason="ONNXRUNTIME doesn't match PyTorch when ceil_mode=True until opset 19",
matcher=lambda sample: sample.kwargs.get("ceil_mode") is True,
reason="fixme: ORT doesn't match PyTorch when ceil_mode=True until opset 19",
),
TorchLibOpInfo(
"nn.functional.conv1d",
Expand Down