From dccb11881fae72886353bb7f278c020a0e3d08f0 Mon Sep 17 00:00:00 2001 From: Shangming Cai Date: Sun, 5 Apr 2026 23:13:06 +0800 Subject: [PATCH] [PD] Fix staging warmup for GQA prefill decode different tp (#22153) --- python/sglang/srt/disaggregation/decode.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/sglang/srt/disaggregation/decode.py b/python/sglang/srt/disaggregation/decode.py index f54c882cc..5c152feaa 100644 --- a/python/sglang/srt/disaggregation/decode.py +++ b/python/sglang/srt/disaggregation/decode.py @@ -776,6 +776,7 @@ class DecodePreallocQueue: ) if ( self.transfer_queue.enable_staging + and hasattr(decode_req.kv_receiver, "require_staging") and decode_req.kv_receiver.require_staging ): self.transfer_queue.staging_handler.register_decode_req(