Signed-off-by: Robert Shaw <robshaw@redhat.com>
This commit is contained in:
Robert Shaw 2025-07-20 13:34:38 +00:00
parent 471fa4ae68
commit 6569facd3b
2 changed files with 3 additions and 0 deletions

View File

@ -410,6 +410,8 @@ class AsyncLLM(EngineClient):
# 4) Logging.
# TODO(rob): make into a coroutine and launch it in
# background thread once Prometheus overhead is non-trivial.
logger.info(f"{outputs.engine_index=}")
logger.info(f"{stat_loggers[outputs.engine_index]=}")
if stat_loggers:
AsyncLLM._record_stats(
stat_loggers[outputs.engine_index],

View File

@ -422,6 +422,7 @@ class PrometheusStatLogger(StatLoggerBase):
def record(self, scheduler_stats: Optional[SchedulerStats],
iteration_stats: Optional[IterationStats]):
logger.info(f"PROM STAT LOGGER: {self.engine_index=}")
"""Log to prometheus."""
if scheduler_stats is not None:
self.gauge_scheduler_running.set(scheduler_stats.num_running_reqs)