Run test_serialization serially (for 2xlarge runners) (#94613)

huydhn · pytorchmergebot · commit d51ca38ef0e3 · 2023-02-11T00:15:10.000Z
Fixes #92746 Pull Request resolved: #94613 Approved by: https://github.com/clee2000
diff --git a/test/run_test.py b/test/run_test.py
@@ -319,6 +319,7 @@ def skip_test_p(name: str) -> bool:
     'functorch/test_vmap',  # OOM
     'test_fx',  # gets SIGKILL
     'test_dataloader',  # frequently hangs for ROCm
+    'test_serialization',   # test_serialization_2gb_file allocates a tensor of 2GB, and could cause OOM
 ]
 
 # A subset of our TEST list that validates PyTorch's ops, modules, and autograd function as expected
diff --git a/test/test_serialization.py b/test/test_serialization.py
@@ -5,6 +5,7 @@
 import io
 import tempfile
 import os
+import gc
 import sys
 import zipfile
 import warnings
@@ -905,6 +906,8 @@ def test_serialization_zipfile_actually_jit(self):
 
     # Ensure large zip64 serialization works properly
     def test_serialization_2gb_file(self):
+        # Run GC to clear up as much memory as possible before running this test
+        gc.collect()
         big_model = torch.nn.Conv2d(20000, 3200, kernel_size=3)
 
         with BytesIOContext() as f:

Original file line number	Diff line number	Diff line change
`@@ -319,6 +319,7 @@ def skip_test_p(name: str) -> bool:`
`319`	`319`	`'functorch/test_vmap', # OOM`
`320`	`320`	`'test_fx', # gets SIGKILL`
`321`	`321`	`'test_dataloader', # frequently hangs for ROCm`
	`322`	`+ 'test_serialization', # test_serialization_2gb_file allocates a tensor of 2GB, and could cause OOM`
`322`	`323`	`]`
`323`	`324`
`324`	`325`	`# A subset of our TEST list that validates PyTorch's ops, modules, and autograd function as expected`