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

Revert "[Scheduler] Decouple maybe_send_health_check_signal from process_batch_result" (#20259)

Browse Source
This commit is contained in:
Liangsheng Yin
2026-03-10 01:58:48 -07:00
committed by GitHub
parent 8cd1de3354
commit ac07a6d439
5 changed files with 1 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/sglang/srt/disaggregation/decode.py
View File

@@ -980,8 +980,6 @@ class SchedulerDisaggregationDecodeMixin:
# Update last_batch
self.last_batch = batch
self.maybe_send_health_check_signal()
@torch.no_grad()
def event_loop_overlap_disagg_decode(self: Scheduler):
self.result_queue = deque()
@@ -1019,8 +1017,6 @@ class SchedulerDisaggregationDecodeMixin:
# Update last_batch
self.last_batch = batch
self.maybe_send_health_check_signal()
def _run_batch_prebuilt(
self: Scheduler, batch: ScheduleBatch
) -> GenerationBatchResult:

4
python/sglang/srt/disaggregation/prefill.py
View File

@@ -381,8 +381,6 @@ class SchedulerDisaggregationPrefillMixin:
# Update last_batch
self.last_batch = batch
self.maybe_send_health_check_signal()
@torch.no_grad()
def event_loop_overlap_disagg_prefill(self: Scheduler) -> None:
self.result_queue = deque()
@@ -423,8 +421,6 @@ class SchedulerDisaggregationPrefillMixin:
# Update last_batch
self.last_batch = batch
self.maybe_send_health_check_signal()
def process_batch_result_disagg_prefill(
self: Scheduler,
batch: ScheduleBatch,

5
python/sglang/srt/managers/scheduler.py
View File

@@ -1259,8 +1259,6 @@ class Scheduler(
# Update last_batch
self.last_batch = batch
self.maybe_send_health_check_signal()
if envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.get():
self.self_check_during_busy()
@@ -1316,8 +1314,6 @@ class Scheduler(
# Update last_batch
self.last_batch = batch
self.maybe_send_health_check_signal()
if envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.get():
self.self_check_during_busy()
@@ -2621,6 +2617,7 @@ class Scheduler(
self.log_batch_result_stats(batch, result)
self._maybe_clear_mm_inputs(batch)
self.maybe_send_health_check_signal()
def maybe_send_health_check_signal(self):
if self.return_health_check_ct:

6
python/sglang/srt/managers/scheduler_pp_mixin.py
View File

@@ -139,8 +139,6 @@ class SchedulerPPMixin:
self.pp_outputs = next_pp_outputs
self.maybe_send_health_check_signal()
# When the server is idle, self-check and re-init some states
if server_is_idle:
self.self_check_during_idle()
@@ -316,8 +314,6 @@ class SchedulerPPMixin:
self.running_batch.batch_is_full = False
self.maybe_send_health_check_signal()
# When the server is idle, self-check and re-init some states
if server_is_idle and len(self.disagg_prefill_inflight_queue) == 0:
self.self_check_during_idle()
@@ -508,8 +504,6 @@ class SchedulerPPMixin:
if self.server_args.disaggregation_decode_enable_offload_kvcache:
queue_size += len(self.decode_offload_manager.ongoing_offload)
self.maybe_send_health_check_signal()
if server_is_idle and queue_size == 0:
self.self_check_during_idle()

2
python/sglang/srt/multiplex/multiplexing_mixin.py
View File

@@ -219,5 +219,3 @@ class SchedulerMultiplexMixin:
self.split_prefill_batch = None
wait_prefill_kernel_done = False
adjust_stream_group = True
self.maybe_send_health_check_signal()