Skip to content

Commit 238dfc8

Browse files
authored
[MISC] tiny fixes (#13378)
1 parent 4518683 commit 238dfc8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

vllm/executor/executor_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def initialize_cache(self, num_gpu_blocks: int, num_cpu_blocks) -> None:
109109
"""
110110
# NOTE: This is logged in the executor because there can be >1 workers.
111111
logger.info("# %s blocks: %d, # CPU blocks: %d",
112-
vllm.platforms.current_platform.dispatch_key,
112+
vllm.platforms.current_platform.device_name,
113113
num_gpu_blocks, num_cpu_blocks)
114114
max_concurrency = (num_gpu_blocks * self.cache_config.block_size /
115115
self.model_config.max_model_len)

vllm/platforms/interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -327,7 +327,7 @@ def get_device_communicator_cls(cls) -> str:
327327
"""
328328
Get device specific communicator class for distributed communication.
329329
"""
330-
return "vllm.distributed.device_communicator.base_device_communicator.DeviceCommunicatorBase" # noqa
330+
return "vllm.distributed.device_communicators.base_device_communicator.DeviceCommunicatorBase" # noqa
331331

332332

333333
class UnspecifiedPlatform(Platform):

0 commit comments

Comments
 (0)