From 2c7f4ca2f2afe47e474fc278eb9519529ee4adb6 Mon Sep 17 00:00:00 2001 From: Shangming Cai Date: Fri, 3 Oct 2025 23:07:02 +0800 Subject: [PATCH] Optimize debug log position of PD abort request (#11090) Signed-off-by: Shangming Cai --- python/sglang/srt/managers/scheduler.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 5a08f810d..a34a8d4d1 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -2597,31 +2597,31 @@ class Scheduler( # Delete requests not in the waiting queue when PD disaggregation is enabled if self.disaggregation_mode == DisaggregationMode.PREFILL: # Abort requests that have not yet been bootstrapped - for i, req in enumerate(self.disagg_prefill_bootstrap_queue.queue): - logger.debug(f"Abort bootstrap queue request. {req.rid=}") + for req in self.disagg_prefill_bootstrap_queue.queue: if recv_req.abort_all or req.rid.startswith(recv_req.rid): + logger.debug(f"Abort bootstrap queue request. {req.rid=}") if hasattr(req.disagg_kv_sender, "abort"): req.disagg_kv_sender.abort() # Abort in-flight requests - for i, req in enumerate(self.disagg_prefill_inflight_queue): - logger.debug(f"Abort inflight queue request. {req.rid=}") + for req in self.disagg_prefill_inflight_queue: if recv_req.abort_all or req.rid.startswith(recv_req.rid): + logger.debug(f"Abort inflight queue request. {req.rid=}") if hasattr(req.disagg_kv_sender, "abort"): req.disagg_kv_sender.abort() elif self.disaggregation_mode == DisaggregationMode.DECODE: # Abort requests that have not yet finished preallocation - for i, decode_req in enumerate(self.disagg_decode_prealloc_queue.queue): - logger.debug(f"Abort prealloc queue request. {decode_req.req.rid=}") + for decode_req in self.disagg_decode_prealloc_queue.queue: if recv_req.abort_all or decode_req.req.rid.startswith(recv_req.rid): + logger.debug(f"Abort prealloc queue request. {decode_req.req.rid=}") if hasattr(decode_req.kv_receiver, "abort"): decode_req.kv_receiver.abort() # Abort requests waiting for kvcache to release tree cache - for i, decode_req in enumerate(self.disagg_decode_transfer_queue.queue): - logger.debug(f"Abort transfer queue request. {decode_req.req.rid=}") + for decode_req in self.disagg_decode_transfer_queue.queue: if recv_req.abort_all or decode_req.req.rid.startswith(recv_req.rid): + logger.debug(f"Abort transfer queue request. {decode_req.req.rid=}") if hasattr(decode_req.kv_receiver, "abort"): decode_req.kv_receiver.abort()