Disable broken tests (#1055)

facebookresearch · Aug 8, 2022 · f81a60b · f81a60b
1 parent 5c60f33
commit f81a60b
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 2 deletions.
diff --git a/tests/experimental/nn/test_ssd_offload.py b/tests/experimental/nn/test_ssd_offload.py
@@ -79,6 +79,7 @@ def test_ssd_handle_dispatch_bwd():
         assert torch.equal(ssd_handle.grad, orig_copy.grad)
 
 
+@pytest.mark.skip("broken at head")
 def test_ssd_handle_dispatch_bwd_hook():
     _init()
 
@@ -277,6 +278,7 @@ def test_ssd_flat_parameter_view_modify():
         assert ssd_flat_param.storage_state == so.StorageState.ON_CPU_DIRTY
 
 
+@pytest.mark.skip("broken at head")
 def test_ssd_flat_parameter_view_bwd():
     _init()
 
@@ -344,6 +346,7 @@ def post_backward_hook(name, hooks_called, *grads):
         assert "one" in hooks_called
 
 
+@pytest.mark.skip("broken at head")
 def test_ssd_flat_parameter_view_bwd_parameterization():
     _init()
 

diff --git a/tests/experimental/nn/test_sync_batchnorm.py b/tests/experimental/nn/test_sync_batchnorm.py
@@ -97,7 +97,12 @@ def parity3d_bn():
     check_parity(torch_bn, fs_bn, x)
 
 
-@pg_test()
+@pytest.mark.skip("broken at head")
+def test_parity3d_checkpoint_syncbn():
+    assert 1 == 2
+
+
+# @pg_test()
 def parity3d_checkpoint_syncbn():
     rank = dist.get_rank()
     torch.cuda.set_device(rank)
@@ -110,7 +115,12 @@ def parity3d_checkpoint_syncbn():
     check_parity_ddp(torch_bn, fs_bn, x)
 
 
-@pg_test()
+@pytest.mark.skip("broken at head")
+def test_parity3d_checkpoint_syncbn_twice():
+    assert 1 == 2
+
+
+# @pg_test()
 def parity3d_checkpoint_syncbn_twice():
     rank = dist.get_rank()
     torch.cuda.set_device(rank)

diff --git a/tests/nn/data_parallel/test_fsdp_memory.py b/tests/nn/data_parallel/test_fsdp_memory.py
@@ -159,6 +159,7 @@ def cmp(results, expected):
 
 
 @skip_if_single_gpu
+@pytest.mark.timeout(120)
 @pytest.mark.parametrize("ckpt", ["no_ckpt", "ckpt"])
 @pytest.mark.parametrize("fsdp", ["ddp", "fsdp", "fsdp_amp_default", "fsdp_amp_compute_dtype32"])
 def test_fsdp_memory(fsdp, ckpt):

diff --git a/tests/nn/data_parallel/test_fsdp_regnet.py b/tests/nn/data_parallel/test_fsdp_regnet.py
@@ -349,6 +349,7 @@ def dump(d):
 
 # We use strings for precision and flatten params instead of bool to
 # make the pytest output more readable.
+@pytest.mark.skip("broken at head")
 @skip_if_single_gpu
 @pytest.mark.parametrize("precision", ["full", "mixed"])
 @pytest.mark.parametrize("flatten", ["flatten", "no_flatten"])

diff --git a/tests/optim/test_oss.py b/tests/optim/test_oss.py
@@ -958,6 +958,7 @@ def closure_sharded(input_tensor=input_tensor):
     dist.destroy_process_group()
 
 
+@pytest.mark.skip("broken at head")
 @skip_if_no_cuda
 @skip_if_single_gpu
 @pytest.mark.parametrize("change_train_graph", [True, False])