Unverified Commit f7ee6986 authored by muzian666's avatar muzian666 Committed by GitHub
Browse files

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


Signed-off-by: default avatarmuzian666 <andylee_2001@163.com>
Co-authored-by: default avatarqingan.li <qingan.li@wizpresso.com>
parent d2a71530
...@@ -4,6 +4,7 @@ ...@@ -4,6 +4,7 @@
import json import json
import os import os
import platform import platform
import re
import subprocess import subprocess
import sys import sys
from dataclasses import dataclass from dataclasses import dataclass
...@@ -336,6 +337,7 @@ class CpuPlatform(Platform): ...@@ -336,6 +337,7 @@ class CpuPlatform(Platform):
lscpu_output = subprocess.check_output( lscpu_output = subprocess.check_output(
"lscpu -J -e=CPU,CORE,NODE", shell=True, text=True "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( logical_cpu_list: list[LogicalCPUInfo] = json.loads(
lscpu_output, object_hook=LogicalCPUInfo.json_decoder lscpu_output, object_hook=LogicalCPUInfo.json_decoder
)["cpus"] )["cpus"]
......
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