[Logging] Guard log_prefill_stats against idle batches in disagg prefill (#19361)

This commit is contained in:
Liangsheng Yin
2026-02-25 13:31:52 -08:00
committed by GitHub
parent 08957c88ea
commit c60dcc40bb

View File

@@ -535,7 +535,7 @@ class SchedulerDisaggregationPrefillMixin:
self.maybe_send_health_check_signal()
if self.current_scheduler_metrics_enabled:
if self.current_scheduler_metrics_enabled and batch.prefill_stats is not None:
can_run_cuda_graph = getattr(result, "can_run_cuda_graph", False)
self.log_prefill_stats(
prefill_stats=batch.prefill_stats,