[test_shape_ops] Increase system memory requirement (pytorch#80369)

Increase system memory requirement for TestShapeOpsCUDA.test_flip_large_tensor_cuda

Signed-off-by: Jagadish Krishnamoorthy <[email protected]>

Fixes pytorch#80373

Pull Request resolved: pytorch#80369
Approved by: https://github.com/soulitzer

Author: jaglinux

test/test_shape_ops.py

@@ -475,6 +475,7 @@ def test_flip_numpy(self, device, dtype):
 
     @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,))