From bd0aa2230922590e5bb6e0254b74e7d92730aaac Mon Sep 17 00:00:00 2001 From: Yuhao Yang <47235274+yhyang201@users.noreply.github.com> Date: Sun, 10 May 2026 19:02:47 +0800 Subject: [PATCH] Fix PD bootstrap failure handling (#24772) Co-authored-by: Cheng Wan --- python/sglang/srt/disaggregation/common/conn.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/sglang/srt/disaggregation/common/conn.py b/python/sglang/srt/disaggregation/common/conn.py index 840063094..1fcd91d9c 100644 --- a/python/sglang/srt/disaggregation/common/conn.py +++ b/python/sglang/srt/disaggregation/common/conn.py @@ -608,6 +608,8 @@ class CommonKVReceiver(BaseKVReceiver): self.prefill_dp_rank = prefill_dp_rank self._setup_bootstrap_infos() + if self.conclude_state == KVPoll.Failed: + return self.kv_mgr.update_status(self.bootstrap_room, KVPoll.WaitingForInput) def _setup_bootstrap_infos(self): @@ -650,6 +652,7 @@ class CommonKVReceiver(BaseKVReceiver): self.kv_mgr.update_status( self.bootstrap_room, KVPoll.Failed ) + self.bootstrap_infos = None return self.bootstrap_infos = bootstrap_infos