From 0f81eeecbb84e6306ec930a3c13df0e5c1934ac0 Mon Sep 17 00:00:00 2001
From: Kaiyao Ke <47203510+kaiyaok2@users.noreply.github.com>
Date: Wed, 5 Mar 2025 02:43:08 -0600
Subject: [PATCH] Update test_results.py

---
 tests/tests_pytorch/core/test_results.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/tests_pytorch/core/test_results.py b/tests/tests_pytorch/core/test_results.py
index edc6efbfa4889..93982086a6b0a 100644
--- a/tests/tests_pytorch/core/test_results.py
+++ b/tests/tests_pytorch/core/test_results.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 from functools import partial
 
+import pytest
 import torch
 import torch.distributed as dist
 
@@ -48,6 +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")
 @RunIf(skip_windows=True)
 def test_result_reduce_ddp():
     spawn_launch(result_reduce_ddp_fn, [torch.device("cpu")] * 2)