Skip to content

Commit 80899dd

Browse files
author
zhangyue
committed
test(torch): skip high-memory cuda svd reference
1 parent 4273849 commit 80899dd

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

tests/test_torch_ops.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,13 @@ def _list_default(aten_type):
218218
}
219219
)
220220

221+
# Ops whose reference path needs too much memory for the generic large CI shape.
222+
_HIGH_MEMORY_REFERENCE_CASES = frozenset(
223+
{
224+
("cuda", "svd", (4, 4, 5632)),
225+
}
226+
)
227+
221228
# Ops where the ATen `_out` schema and the Python reference (`torch.<op>`,
222229
# `torch.nn.functional.<op>`) diverge in positional-argument ordering, so
223230
# the harness's purely-positional reference call lands an InfiniOps
@@ -451,6 +458,9 @@ def test_op(op_meta, shape, dtype, device, rtol, atol):
451458
if (device, aten_name) in _VENDOR_CRASH_OPS:
452459
pytest.skip(f"`{aten_name}` crashes on `{device}` vendor kernel")
453460

461+
if (device, aten_name, shape) in _HIGH_MEMORY_REFERENCE_CASES:
462+
pytest.skip(f"`{aten_name}` reference path needs too much memory")
463+
454464
if device == "cuda" and aten_name in _DEVICE_ASSERTING_OPS:
455465
pytest.skip(
456466
f"`{aten_name}` triggers a CUDA device-side assert on random inputs"

0 commit comments

Comments
 (0)