From d82097a0df0640ea28587fb9762b9dc43bf07be3 Mon Sep 17 00:00:00 2001 From: Shangming Cai Date: Fri, 3 Apr 2026 15:13:44 +0800 Subject: [PATCH] [PD] Tiny register info field cleanup for mooncake backend (#22016) --- python/sglang/srt/disaggregation/mooncake/conn.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/sglang/srt/disaggregation/mooncake/conn.py b/python/sglang/srt/disaggregation/mooncake/conn.py index fc74d555d..64d97f5c6 100644 --- a/python/sglang/srt/disaggregation/mooncake/conn.py +++ b/python/sglang/srt/disaggregation/mooncake/conn.py @@ -81,6 +81,7 @@ class TransferInfo: dst_state_indices: List[int] required_dst_info_num: int is_dummy: bool + # Note: always put the optional staging field at the final (it will be set through 'STAGING_RSP' pkg when needed) staging: Optional[StagingTransferInfo] = None @classmethod @@ -129,6 +130,7 @@ class KVArgsRegisterInfo: dst_state_dim_per_tensor: list[int] # HiSparse: decode host pool stores KV at token granularity enable_hisparse: bool = False + # Note: always put the staging field at the final (since the staging field is optional and contains multiple inputs) staging: Optional[StagingRegisterInfo] = None @classmethod @@ -157,6 +159,7 @@ class KVArgsRegisterInfo: enable_hisparse=( msg[12].decode("ascii") == "1" if len(msg) > 12 else False ), + # Note: always put the staging field at the final staging=StagingRegisterInfo.from_zmq_fields(msg, 13), )