Skip to content

Commit 71793c6

Browse files
authored
test: flaky terminated with signal SIGABRT (#20621)
Update test_results.py
1 parent d15eeff commit 71793c6

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/tests_pytorch/core/test_results.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
# limitations under the License.
1414
from functools import partial
1515

16+
import pytest
1617
import torch
1718
import torch.distributed as dist
1819

@@ -48,6 +49,8 @@ def result_reduce_ddp_fn(strategy):
4849
assert actual.item() == dist.get_world_size()
4950

5051

52+
# flaky with "process 0 terminated with signal SIGABRT"
53+
@pytest.mark.flaky(reruns=3, only_rerun="torch.multiprocessing.spawn.ProcessExitedException")
5154
@RunIf(skip_windows=True)
5255
def test_result_reduce_ddp():
5356
spawn_launch(result_reduce_ddp_fn, [torch.device("cpu")] * 2)

0 commit comments

Comments
 (0)