Skip to content

Commit 25c434b

Browse files
jaglinuxjithunnair-amd
authored andcommitted
[ test_shape_ops] Increase system memory requirement (#1026)
Increase system memory requirement for TestShapeOpsCUDA.test_flip_large_tensor_cuda Signed-off-by: Jagadish Krishnamoorthy <[email protected]>
1 parent 572c17c commit 25c434b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/test_shape_ops.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -475,6 +475,7 @@ def test_flip_numpy(self, device, dtype):
475475

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

0 commit comments

Comments
 (0)