Skip to content

Commit 827b730

Browse files
malfetpytorchmergebot
authored andcommitted
[CI] Skip test_copy_large_tensor on M2-15 runners (pytorch#150377)
They have more than 12Gb memory, but may be running this test causes OOM in CI Pull Request resolved: pytorch#150377 Approved by: https://github.com/atalman
1 parent 6470b37 commit 827b730

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/test_mps.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7476,6 +7476,7 @@ def compare_mm(m, n, k, dtype=torch.float):
74767476

74777477
@unittest.skipIf(total_memory < 12_000_000_000, "Needs at least 12Gb RAM to run the test")
74787478
@unittest.skipIf(MACOS_VERSION < 14.0, "Can't allocate 4Gb tensor on MacOS 13")
7479+
@unittest.skipIf(IS_CI, "May be fixes https://github.com/pytorch/pytorch/issues/149999")
74797480
def test_copy_large(self):
74807481
""" Test that copy of 4Gb+ tensors works """
74817482
x = torch.ones((2**30 + 11,), dtype=torch.float32)

0 commit comments

Comments
 (0)