diff --git a/torchrec/distributed/test_utils/multi_process.py b/torchrec/distributed/test_utils/multi_process.py index 5cfd3339f..6c65d2793 100644 --- a/torchrec/distributed/test_utils/multi_process.py +++ b/torchrec/distributed/test_utils/multi_process.py @@ -126,7 +126,7 @@ def _run_multi_process_test( # pyre-ignore **kwargs, ) -> None: - ctx = multiprocessing.get_context("forkserver") + ctx = multiprocessing.get_context("spawn") processes = [] for rank in range(world_size): kwargs["rank"] = rank