[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>
2025-12-26 19:54:35 +08:00
parent 3778c2fc6d
commit cf34d0ab32
1 changed files with 2 additions and 0 deletions
@@ -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()