From cf34d0ab3292f11f8668d2ecce8b93e6f70dc67c Mon Sep 17 00:00:00 2001
From: Huaixin Chang <61184708+changhuaixin@users.noreply.github.com>
Date: Fri, 26 Dec 2025 19:54:35 +0800
Subject: [PATCH] [Fix] assert error in log_prefill_stats (#15881)

Signed-off-by: Chang Huaixin (OpenAnolis) <changhuaixin@linux.alibaba.com>
Co-authored-by: Junjie Mao (Alibaba) <banxing.mjj@alibaba-inc.com>
---
 python/sglang/srt/disaggregation/prefill.py | 2 ++
 1 file changed, 2 insertions(+)

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()