Read process groups through the runtime context (#40068)

2026-09-18 17:40:32 -07:00
parent 5931fd60ee
commit afe71f4b9e
165 changed files with 569 additions and 608 deletions
@@ -270,8 +270,9 @@ class TestHostMemoryBudget(CustomTestCase):
            unittest.mock.patch.object(
                torch.distributed, "is_initialized", return_value=True
            ),
-            unittest.mock.patch.object(
-                base, "get_world_group", return_value=fake_group
+            unittest.mock.patch(
+                "sglang.srt.distributed.parallel_state.get_world_group",
+                return_value=fake_group,
            ),
        ):
            self.assertEqual(base.ranks_per_host(), 8)