From dc194a309641a68c16d29cb904e5b8a100a13395 Mon Sep 17 00:00:00 2001 From: ankushwahaRH Date: Fri, 15 Aug 2025 00:11:52 +0000 Subject: [PATCH] Test multiprocessing spawn timing fix (#160672) Submitting PR to fix #160511. Pull Request resolved: https://github.com/pytorch/pytorch/pull/160672 Approved by: https://github.com/mikaylagawarecki --- test/test_multiprocessing_spawn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_multiprocessing_spawn.py b/test/test_multiprocessing_spawn.py index 25eea49b755c..d093e01921dc 100644 --- a/test/test_multiprocessing_spawn.py +++ b/test/test_multiprocessing_spawn.py @@ -47,7 +47,7 @@ def _test_terminate_signal_func(i): def _test_terminate_exit_func(i, arg): if i == 0: sys.exit(arg) - time.sleep(1.0) + time.sleep(4.0) def _test_success_first_then_exception_func(i, arg): @@ -145,7 +145,7 @@ class _TestMultiProcessing: with self.assertRaisesRegex(Exception, message): mp.start_processes(_test_terminate_signal_func, nprocs=2, start_method=self.start_method) - @parametrize("grace_period", [None, 5]) + @parametrize("grace_period", [None, 20]) def test_terminate_exit(self, grace_period): exitcode = 123 ctx = mp.start_processes(_test_terminate_exit_func, args=(exitcode,), nprocs=2, start_method=self.start_method, join=False)