Skip to content

GitLab

Unverified Commit 08d5f711 authored by Ning Xie's avatar Ning Xie Committed by GitHub
Browse files

[Misc] refactor function name (#23029) 


Signed-off-by: default avatarAndy Xie <andy.xning@gmail.com>
parent b2fd0b81
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/platforms/cpu.py
View file @ 08d5f711
...@@ -268,7 +268,7 @@ class CpuPlatform(Platform): ...@@ -268,7 +268,7 @@ class CpuPlatform(Platform):
DEFAULT_MAX_NUM_BATCHED_TOKENS) DEFAULT_MAX_NUM_BATCHED_TOKENS)
@classmethod @classmethod
def get_allowed_cpu_memory_node_list( def get_allowed_cpu_core_node_list(
cls) -> tuple[list[int], list[LogicalCPUInfo]]: cls) -> tuple[list[int], list[LogicalCPUInfo]]:
assert platform.system() == "Linux" assert platform.system() == "Linux"
......
vllm/v1/worker/cpu_worker.py
View file @ 08d5f711
...@@ -132,7 +132,7 @@ class CPUWorker(Worker): ...@@ -132,7 +132,7 @@ class CPUWorker(Worker):
""" """
allowed_numa_nodes, logical_cpu_list = \ allowed_numa_nodes, logical_cpu_list = \
CpuPlatform.get_allowed_cpu_memory_node_list() CpuPlatform.get_allowed_cpu_core_node_list()
assert len(allowed_numa_nodes) >= self.parallel_config.world_size, ( assert len(allowed_numa_nodes) >= self.parallel_config.world_size, (
f"No enough allowed NUMA nodes to bind threads of " f"No enough allowed NUMA nodes to bind threads of "
f"{self.parallel_config.world_size} CPUWorkers. " f"{self.parallel_config.world_size} CPUWorkers. "
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment