diff --git a/tests/tests_fabric/utilities/test_distributed.py b/tests/tests_fabric/utilities/test_distributed.py index fa9cc0ed40e93..5c6570eae9b0e 100644 --- a/tests/tests_fabric/utilities/test_distributed.py +++ b/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", diff --git a/tests/tests_pytorch/callbacks/test_stochastic_weight_avg.py b/tests/tests_pytorch/callbacks/test_stochastic_weight_avg.py index c24e33c80c6df..abcd302149fcf 100644 --- a/tests/tests_pytorch/callbacks/test_stochastic_weight_avg.py +++ b/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() diff --git a/tests/tests_pytorch/core/test_results.py b/tests/tests_pytorch/core/test_results.py index 93982086a6b0a..c1d50e8458da6 100644 --- a/tests/tests_pytorch/core/test_results.py +++ b/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)