We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d15eeff commit 71793c6Copy full SHA for 71793c6
tests/tests_pytorch/core/test_results.py
@@ -13,6 +13,7 @@
13
# limitations under the License.
14
from functools import partial
15
16
+import pytest
17
import torch
18
import torch.distributed as dist
19
@@ -48,6 +49,8 @@ def result_reduce_ddp_fn(strategy):
48
49
assert actual.item() == dist.get_world_size()
50
51
52
+# flaky with "process 0 terminated with signal SIGABRT"
53
+@pytest.mark.flaky(reruns=3, only_rerun="torch.multiprocessing.spawn.ProcessExitedException")
54
@RunIf(skip_windows=True)
55
def test_result_reduce_ddp():
56
spawn_launch(result_reduce_ddp_fn, [torch.device("cpu")] * 2)
0 commit comments