[CUDA][CPU] Bump system memory requirement for test_cross_entropy_large_tensor (#151812)

eqy · pytorchmergebot · commit 6efc57222179 · 2025-04-24T19:25:29.000Z
`/usr/bin/time` seems to show max resident pages at 119GiB Pull Request resolved: #151812 Approved by: https://github.com/colesbury
diff --git a/test/test_nn.py b/test/test_nn.py
@@ -12152,7 +12152,7 @@ def check_equal(loss, inp_targ_1, inp_targ_2):
 
     # Ref: https://github.com/pytorch/pytorch/issues/85005
     @onlyCUDA
-    @largeTensorTest("45GB", "cpu")
+    @largeTensorTest("120GB", "cpu")
     @largeTensorTest("70GB", "cuda")
     @parametrize_test("reduction", ("none", "mean", "sum"))
     def test_cross_entropy_large_tensor(self, device, reduction):
