increase baby gloo timeout

pytorch · Feb 20, 2025 · d908392 · d908392
1 parent 9cf9554
commit d908392
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchft/process_group_test.py b/torchft/process_group_test.py
@@ -712,7 +712,7 @@ def _create_pg(cls, backend: str) -> ProcessGroup:
         if backend == "gloo":
             return ProcessGroupGloo(timeout=timedelta(seconds=1))
         elif backend == "baby_gloo":
-            return ProcessGroupBabyGloo(timeout=timedelta(seconds=2))
+            return ProcessGroupBabyGloo(timeout=timedelta(seconds=5))
         elif backend == "baby_nccl":
             return ProcessGroupBabyNCCL(timeout=timedelta(seconds=10))
         else: