Fix incorrect running batch size in prefill stats (#15941)

2026-01-03 20:49:54 +08:00
parent c7c0d97fc6
commit b65c9889a0
1 changed files with 6 additions and 1 deletions
--- a/python/sglang/srt/managers/scheduler.py
+++ b/python/sglang/srt/managers/scheduler.py
@@ -2073,7 +2073,12 @@ class Scheduler(

        # Print stats
        if self.current_scheduler_metrics_enabled:
-            self.log_prefill_stats(adder, can_run_list, running_bs, 0)
+            self.log_prefill_stats(
+                adder,
+                can_run_list,
+                running_bs=len(self.running_batch.reqs),
+                running_bs_offline_batch=0,
+            )

        # Record metrics
        for req in can_run_list: