chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

Revert "[Feature] PD-Multiplexing Context and Scheduler." (#12267)

Browse Source
This commit is contained in:
Yineng Zhang
2025-10-28 02:00:37 -07:00
committed by GitHub
parent 41a113356a
commit ea39952797
9 changed files with 26 additions and 457 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/sglang/srt/layers/logits_processor.py
View File

@@ -135,7 +135,10 @@ class LogitsMetadata:
@classmethod
def from_forward_batch(cls, forward_batch: ForwardBatch):
if (
forward_batch.forward_mode.is_extend()
(
forward_batch.forward_mode.is_extend()
or forward_batch.forward_mode.is_split_prefill()
)
and forward_batch.return_logprob
and not forward_batch.forward_mode.is_target_verify()
):
@@ -386,8 +389,8 @@ class LogitsProcessor(nn.Module):
input_logprob_indices = None
elif (
logits_metadata.forward_mode.is_extend()
and not logits_metadata.extend_return_logprob
):
or logits_metadata.forward_mode.is_split_prefill()
) and not logits_metadata.extend_return_logprob:
# Prefill without input logprobs.
if logits_metadata.padded_static_len < 0:
last_index = torch.cumsum(logits_metadata.extend_seq_lens, dim=0) - 1