test: addressing flaky spawn in subprocesses (#20940)

Author: Borda

tests/tests_fabric/utilities/test_distributed.py

@@ -105,8 +105,8 @@ def _test_all_reduce(strategy):
         assert result is tensor  # inplace
 
 
-# flaky with "process 0 terminated with signal SIGABRT" (GLOO)
-@pytest.mark.flaky(reruns=3, only_rerun="torch.multiprocessing.spawn.ProcessExitedException")
+# flaky with "torch.multiprocessing.spawn.ProcessExitedException: process 0 terminated with signal SIGABRT" (GLOO)
+@pytest.mark.flaky(reruns=3)
 @RunIf(skip_windows=True)
 @pytest.mark.parametrize(
     "process",

tests/tests_pytorch/callbacks/test_stochastic_weight_avg.py

@@ -354,8 +354,8 @@ def test_swa_resume_training_from_checkpoint_custom_scheduler(tmp_path, crash_on
 
 
 @RunIf(skip_windows=True)
-# flaky with "process 0 terminated with signal SIGABRT" (GLOO)
-@pytest.mark.flaky(reruns=3, only_rerun="torch.multiprocessing.spawn.ProcessExitedException")
+# flaky with "torch.multiprocessing.spawn.ProcessExitedException: process 0 terminated with signal SIGABRT" (GLOO)
+@pytest.mark.flaky(reruns=3)
 def test_swa_resume_training_from_checkpoint_ddp(tmp_path):
     model = SwaTestModel(crash_on_epoch=3)
     resume_model = SwaTestModel()

tests/tests_pytorch/core/test_results.py

@@ -49,8 +49,8 @@ def result_reduce_ddp_fn(strategy):
     assert actual.item() == dist.get_world_size()
 
 
-# flaky with "process 0 terminated with signal SIGABRT"
-@pytest.mark.flaky(reruns=3, only_rerun="torch.multiprocessing.spawn.ProcessExitedException")
+# flaky with "torch.multiprocessing.spawn.ProcessExitedException: process 0 terminated with signal SIGABRT"
+@pytest.mark.flaky(reruns=3)
 @RunIf(skip_windows=True)
 def test_result_reduce_ddp():
     spawn_launch(result_reduce_ddp_fn, [torch.device("cpu")] * 2)