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

[test_shape_ops] Increase system memory requirement (#80369)

Browse Source 
Increase system memory requirement for TestShapeOpsCUDA.test_flip_large_tensor_cuda

Signed-off-by: Jagadish Krishnamoorthy <jagdish.krishna@gmail.com>

Fixes https://github.com/pytorch/pytorch/issues/80373

Pull Request resolved: https://github.com/pytorch/pytorch/pull/80369
Approved by: https://github.com/soulitzer
This commit is contained in:
Jagadish Krishnamoorthy
2022-06-29 23:11:12 +00:00
committed by PyTorch MergeBot
parent c44317704a
commit 70e86b4562
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_shape_ops.py
View File

@ -475,6 +475,7 @@ class TestShapeOps(TestCase):
@onlyCUDA # CPU is too slow
@largeTensorTest('17GB') # 4 tensors of 4GB (in, out) x (torch, numpy) + 1GB
@largeTensorTest("81GB", "cpu") # even for CUDA test, sufficient system memory is required
def test_flip_large_tensor(self, device):
t_in = torch.empty(2**32 + 1, dtype=torch.uint8).random_()
torch_fn = partial(torch.flip, dims=(0,))