diff --git a/tests/test_torch_ops.py b/tests/test_torch_ops.py index 2f28b24f2..5447e710e 100644 --- a/tests/test_torch_ops.py +++ b/tests/test_torch_ops.py @@ -218,6 +218,13 @@ def _list_default(aten_type): } ) +# Ops whose reference path needs too much memory for the generic large CI shape. +_HIGH_MEMORY_REFERENCE_CASES = frozenset( + { + ("cuda", "svd", (4, 4, 5632)), + } +) + # Ops where the ATen `_out` schema and the Python reference (`torch.`, # `torch.nn.functional.`) diverge in positional-argument ordering, so # the harness's purely-positional reference call lands an InfiniOps @@ -451,6 +458,9 @@ def test_op(op_meta, shape, dtype, device, rtol, atol): if (device, aten_name) in _VENDOR_CRASH_OPS: pytest.skip(f"`{aten_name}` crashes on `{device}` vendor kernel") + if (device, aten_name, shape) in _HIGH_MEMORY_REFERENCE_CASES: + pytest.skip(f"`{aten_name}` reference path needs too much memory") + if device == "cuda" and aten_name in _DEVICE_ASSERTING_OPS: pytest.skip( f"`{aten_name}` triggers a CUDA device-side assert on random inputs"