[Fix] Include disagg prefill waiting queue in FPM (#32122)

Signed-off-by: hongkuanz <hongkuanz@nvidia.com>
This commit is contained in:
Hongkuan Zhou
2026-07-23 09:30:27 -07:00
committed by GitHub
parent b98a577fbe
commit d0b9689805
2 changed files with 5 additions and 2 deletions
@@ -304,6 +304,8 @@ class SchedulerMetricsReporter:
if self.scheduler.disaggregation_mode == DisaggregationMode.PREFILL:
for req in self.scheduler.disagg_prefill_bootstrap_queue.queue:
prefill_q.add(len(req.origin_input_ids))
for req in self.scheduler.waiting_queue:
prefill_q.add(len(req.origin_input_ids))
elif self.scheduler.disaggregation_mode == DisaggregationMode.DECODE:
for req in self.scheduler.disagg_decode_prealloc_queue.queue:
decode_q.add(req.seqlen)