Skip to content

Commit e5d7ed0

Browse files
authored
[V1] log GPU blocks num for MultiprocExecutor (#11656)
1 parent ad0d567 commit e5d7ed0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm/v1/executor/multiproc_executor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,7 @@ def initialize(self, num_gpu_blocks: int) -> None:
9595
Initialize the KV caches and begin the model execution loop of the
9696
underlying workers.
9797
"""
98+
logger.info("# GPU blocks: %d", num_gpu_blocks)
9899
self.collective_rpc("initialize_cache", args=(num_gpu_blocks, ))
99100
self.collective_rpc("compile_or_warm_up_model")
100101

0 commit comments

Comments
 (0)