From 32d893730f64632f3255944f553023d0d7704fdb Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Tue, 23 Sep 2025 17:02:01 -0700 Subject: [PATCH] Revert "[fix][pd-disag]no need set next batch sampling info done in prefill" (#10828) --- python/sglang/srt/managers/scheduler.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index b84608e06..31294749d 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -2297,9 +2297,8 @@ class Scheduler( def set_next_batch_sampling_info_done(self, batch: ScheduleBatch): if batch.next_batch_sampling_info: if batch.next_batch_sampling_info.grammars is not None: - if self.disaggregation_mode != DisaggregationMode.PREFILL: - batch.next_batch_sampling_info.update_regex_vocab_mask() - self.current_stream.synchronize() + batch.next_batch_sampling_info.update_regex_vocab_mask() + self.current_stream.synchronize() batch.next_batch_sampling_info.sampling_info_done.set() def watchdog_thread(self):