Skip to content

Commit 4459b67

Browse files
increase tensor size to force out of memory exception on MI300X (#1449)
1 parent 5de711c commit 4459b67

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/profiler/test_profiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -717,7 +717,7 @@ def run_profiler(tensor_creation_fn):
717717

718718
def create_cuda_tensor_oom():
719719
device = torch.device("cuda:0")
720-
return torch.empty(1024, 1024, 1024, 20, dtype=torch.float32, device=device)
720+
return torch.empty(1024, 1024, 1024, 1024, dtype=torch.float32, device=device)
721721

722722
def check_trace(fname):
723723
prof.export_chrome_trace(fname)

0 commit comments

Comments
 (0)