frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CI] Relax per proc memory by a little bit, mark a test as serial (#125960)

Browse Source 
test failure is here https://github.com/pytorch/pytorch/actions/runs/9036789873/job/24836020415

* OOMs etc rel to https://github.com/pytorch/pytorch/pull/125598
Pull Request resolved: https://github.com/pytorch/pytorch/pull/125960
Approved by: https://github.com/huydhn
This commit is contained in:
Catherine Lee
2024-05-10 21:11:37 +00:00
committed by PyTorch MergeBot
parent c61bfd24c1
commit b08072f645
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_cuda_multigpu.py
View File

@ -30,6 +30,7 @@ from torch.testing._internal.common_utils import (
IS_SANDCASTLE,
NoTest,
run_tests,
serialTest,
skipCUDANonDefaultStreamIf,
skipIfRocm,
TEST_CUDA,
@ -280,6 +281,7 @@ class TestCudaMultiGPU(TestCase):
assert_change(0, reset_peak=True)
@unittest.skipIf(TEST_CUDAMALLOCASYNC, "temporarily disabled")
@serialTest()
def test_memory_stats(self):
gc.collect()
torch.cuda.empty_cache()