From b202778aa55387e8a5ee4080c00e77e6606e5d83 Mon Sep 17 00:00:00 2001 From: YAMY <74099316+YAMY1234@users.noreply.github.com> Date: Sun, 10 May 2026 16:51:13 -0700 Subject: [PATCH] Fix reduce_scatterv producer contract for SUM_LEN (#24785) --- python/sglang/srt/layers/communicator.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/layers/communicator.py b/python/sglang/srt/layers/communicator.py index 853cf3ad5..7de4337f9 100644 --- a/python/sglang/srt/layers/communicator.py +++ b/python/sglang/srt/layers/communicator.py @@ -700,9 +700,11 @@ class LayerCommunicator: if ( self._communicate_summable_tensor_pair_fn is CommunicateSummableTensorPairFn._scatter_hidden_states - and forward_batch.dp_padding_mode.is_max_len() ): - return True + if should_use_dp_reduce_scatterv(): + return True + if forward_batch.dp_padding_mode.is_max_len(): + return True if nsa_use_prefill_cp(forward_batch): return True if get_attn_tp_context().input_scattered and not self.is_last_layer: