Skip to content

Commit 7042cc9

Browse files
authored
[V1][Spec Decoding] Log accumulated metrics after system goes idle (#17913)
Signed-off-by: Mark McLoughlin <[email protected]>
1 parent 0c0fdae commit 7042cc9

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

vllm/v1/metrics/loggers.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,9 +128,7 @@ def log(self):
128128
scheduler_stats.gpu_cache_usage * 100,
129129
self.prefix_caching_metrics.hit_rate * 100,
130130
)
131-
132-
if scheduler_stats.spec_decoding_stats is not None:
133-
self.spec_decoding_logging.log(log_fn=log_fn)
131+
self.spec_decoding_logging.log(log_fn=log_fn)
134132

135133
def log_engine_initialized(self):
136134
logger.info(

vllm/v1/spec_decode/metrics.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,8 @@ def observe(self, spec_decoding_stats: SpecDecodingStats):
6767
spec_decoding_stats.num_accepted_tokens_per_pos)
6868

6969
def log(self, log_fn=logger.info):
70+
if not self.num_drafts:
71+
return
7072
num_drafts = np.sum(self.num_drafts)
7173
num_draft_tokens = np.sum(self.num_draft_tokens)
7274
num_accepted_tokens = np.sum(self.num_accepted_tokens)

0 commit comments

Comments
 (0)