From f30df723bf146d4a8037010b0b11939a601df423 Mon Sep 17 00:00:00 2001 From: Yilong Zhao <74357408+happierpig@users.noreply.github.com> Date: Wed, 1 Apr 2026 23:33:06 -0700 Subject: [PATCH] scheduler: add prefill-only update in merge batch (#21840) --- python/sglang/srt/managers/schedule_batch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/sglang/srt/managers/schedule_batch.py b/python/sglang/srt/managers/schedule_batch.py index fdbc9b477..0c178204a 100644 --- a/python/sglang/srt/managers/schedule_batch.py +++ b/python/sglang/srt/managers/schedule_batch.py @@ -2266,6 +2266,7 @@ class ScheduleBatch(ScheduleBatchDisaggregationDecodeMixin): self.has_stream |= other.has_stream self.has_grammar |= other.has_grammar self.return_hidden_states |= other.return_hidden_states + self.is_prefill_only = self.is_prefill_only and other.is_prefill_only if self.spec_info: self.spec_info.merge_batch(other.spec_info)