Skip to content

Commit 6ba6f52

Browse files
committed
linspace
Signed-off-by: Justin Chu <[email protected]>
1 parent 7dd4ceb commit 6ba6f52

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -766,9 +766,14 @@ def _where_input_wrangler(
766766
"linspace",
767767
core_ops.aten_linspace,
768768
tolerance={torch.float16: (2e-2, 2e-3)},
769-
).xfail(
769+
)
770+
.xfail(
770771
dtypes=(torch.int64, torch.int32),
771772
reason="fixme: Results do not match with PyTorch. https://github.com/microsoft/onnxscript/issues/854",
773+
)
774+
.skip(
775+
matcher=lambda sample: sample.kwargs.get("dtype") in (torch.int64, torch.int32),
776+
reason="fixme: Results do not match with PyTorch. https://github.com/microsoft/onnxscript/issues/854",
772777
),
773778
TorchLibOpInfo("log", core_ops.aten_log),
774779
TorchLibOpInfo("le", core_ops.aten_le),

0 commit comments

Comments
 (0)