Re-enable temp_prefill_info assertion after pairing fix (#16203)

This commit is contained in:
Hudson Xing
2026-01-07 18:05:17 +08:00
committed by GitHub
parent b86bbf841e
commit 98a107d491

View File

@@ -137,7 +137,7 @@ class SchedulerMetricsMixin:
self.last_input_throughput = self.last_prefill_tokens / gap_latency
self.last_prefill_tokens = adder.log_input_tokens
# assert self.temp_prefill_info is None # TODO re-enable
assert self.temp_prefill_info is None
self.temp_prefill_info = dict(
adder_log_input_tokens=adder.log_input_tokens,
adder_log_hit_tokens=adder.log_hit_tokens,