[test] disable a flaky test (#1020)

Co-authored-by: Min Xu <min.xu.public@gmail.com>
facebookresearch · Jun 29, 2022 · 775a0f0 · 775a0f0
1 parent a5116ec
commit 775a0f0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/fair_dev/testing/testing.py b/fair_dev/testing/testing.py
@@ -81,6 +81,10 @@
     reason="Python3.9 without CUDA is skipped",
 )
 
+skip_due_to_flakyness = pytest.mark.skip(
+    reason="Flaky test to be fixed or removed",
+)
+
 available_devices = ["cpu"]
 if torch.cuda.is_available():
     available_devices.append("cuda")

diff --git a/tests/experimental/nn/test_multiprocess_pipe.py b/tests/experimental/nn/test_multiprocess_pipe.py
@@ -20,7 +20,7 @@
 import torch.multiprocessing as mp
 import torch.nn as nn
 
-from fair_dev.testing.testing import skip_if_single_gpu
+from fair_dev.testing.testing import skip_due_to_flakyness, skip_if_single_gpu
 from fairscale.experimental.nn.distributed_pipeline import DistributedLoss, DistributedPipeline, PipelineModulesGraph
 from fairscale.internal import torch_version
 
@@ -113,6 +113,7 @@ def create_multiple_layers():
 @rpc_test(world_size=2)
 @pytest.mark.parametrize("devices", DEVICES)
 @skip_if_single_gpu
+@skip_due_to_flakyness
 def create_multiple_workers(devices):
     model = [RemoteModuleParams(nn.Linear, (4, 4), {}), RemoteModuleParams(nn.ReLU, (), {})]
     pipe = create_sequence_pipeline(model, balance=[1, 1], chunks=1, devices=devices[:2])