xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-01-26 19:54:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CPU] fix the issue when the node is '-' cause json decode error. (#26562)

Browse Source 
Signed-off-by: muzian666 <andylee_2001@163.com>
Co-authored-by: qingan.li <qingan.li@wizpresso.com>
This commit is contained in:
muzian666 2025-10-11 20:04:04 +08:00 committed by GitHub
parent d2a71530c1
commit f7ee69868a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/platforms/cpu.py
View File

@ -4,6 +4,7 @@
import json
import os
import platform
import re
import subprocess
import sys
from dataclasses import dataclass
@ -336,6 +337,7 @@ class CpuPlatform(Platform):
lscpu_output = subprocess.check_output(
"lscpu -J -e=CPU,CORE,NODE", shell=True, text=True
)
lscpu_output = re.sub(r'"node":\s*-\s*(,|\n)', r'"node": 0\1', lscpu_output)
logical_cpu_list: list[LogicalCPUInfo] = json.loads(
lscpu_output, object_hook=LogicalCPUInfo.json_decoder
)["cpus"]