minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Fix PD bootstrap failure handling (#24772)

Browse Source 
Co-authored-by: Cheng Wan <chwan@rice.edu>
This commit is contained in:
Yuhao Yang
2026-05-10 19:02:47 +08:00
committed by GitHub
co-authored by Cheng Wan
parent 8cc16c9974
commit bd0aa22309
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/disaggregation/common/conn.py
+3
View File
@@ -608,6 +608,8 @@ class CommonKVReceiver(BaseKVReceiver):
self.prefill_dp_rank = prefill_dp_rank self.prefill_dp_rank = prefill_dp_rank
self._setup_bootstrap_infos() self._setup_bootstrap_infos()
if self.conclude_state == KVPoll.Failed:
return
self.kv_mgr.update_status(self.bootstrap_room, KVPoll.WaitingForInput) self.kv_mgr.update_status(self.bootstrap_room, KVPoll.WaitingForInput)
def _setup_bootstrap_infos(self): def _setup_bootstrap_infos(self):
@@ -650,6 +652,7 @@ class CommonKVReceiver(BaseKVReceiver):
self.kv_mgr.update_status( self.kv_mgr.update_status(
self.bootstrap_room, KVPoll.Failed self.bootstrap_room, KVPoll.Failed
) )
self.bootstrap_infos = None
return return
self.bootstrap_infos = bootstrap_infos self.bootstrap_infos = bootstrap_infos