diff --git a/python/sglang/srt/disaggregation/prefill.py b/python/sglang/srt/disaggregation/prefill.py index cf63ff48a..8cf1e5ebb 100644 --- a/python/sglang/srt/disaggregation/prefill.py +++ b/python/sglang/srt/disaggregation/prefill.py @@ -331,6 +331,8 @@ class SchedulerDisaggregationPrefillMixin: if batch: trace_event_batch("schedule", batch.reqs) + self.log_prefill_stats_late(batch) + return batch @torch.no_grad()