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

[CUDA] Add serialTest decorator to largeTensorTest in test_cuda.py (#159271)

Browse Source 
Hopefully helps with disabled tests due to OOM such as https://github.com/pytorch/pytorch/issues/159069

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159271
Approved by: https://github.com/Skylion007, https://github.com/ngimel
This commit is contained in:
eqy
2025-07-31 16:27:16 +00:00
committed by PyTorch MergeBot
parent f946b25865
commit 05aade1b6d
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_cuda.py
View File

@ -1503,6 +1503,7 @@ except RuntimeError as e:
)
@largeTensorTest("20GB", "cuda")
@serialTest()
def test_randint_generation_for_large_numel(self) -> None:
numel = 2**31 + 1
s = torch.randint(2, (numel,), device="cuda", dtype=torch.int8).sum()