Skip to content

Commit b710664

Browse files
committed
refactor: ppu detect
Signed-off-by: thxCode <thxcode0824@gmail.com>
1 parent 3dd70d8 commit b710664

2 files changed

Lines changed: 5 additions & 11 deletions

File tree

gpustack_runtime/detector/iluvatar.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -241,6 +241,8 @@ def get_topology(self, devices: Devices | None = None) -> Topology | None:
241241
)
242242

243243
try:
244+
pyixml.nvmlInit()
245+
244246
for i, dev_i in enumerate(devices):
245247
dev_i_handle = pyixml.nvmlDeviceGetHandleByUUID(dev_i.uuid)
246248

gpustack_runtime/detector/thead.py

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -98,17 +98,7 @@ def detect(self) -> Devices | None:
9898

9999
sys_driver_ver = pyhgml.hgmlSystemGetDriverVersion()
100100

101-
sys_runtime_ver_original = pyhgml.hgmlSystemGetHggcDriverVersion()
102-
sys_runtime_ver_original = ".".join(
103-
map(
104-
str,
105-
[
106-
sys_runtime_ver_original // 1000,
107-
(sys_runtime_ver_original % 1000) // 10,
108-
(sys_runtime_ver_original % 10),
109-
],
110-
),
111-
)
101+
sys_runtime_ver_original = pyhgml.hgmlSystemGetHGMLVersion()
112102
sys_runtime_ver = get_brief_version(
113103
sys_runtime_ver_original,
114104
)
@@ -241,6 +231,8 @@ def get_topology(self, devices: Devices | None = None) -> Topology | None:
241231
)
242232

243233
try:
234+
pyhgml.hgmlInit()
235+
244236
for i, dev_i in enumerate(devices):
245237
dev_i_handle = pyhgml.hgmlDeviceGetHandleByUUID(dev_i.uuid)
246238

0 commit comments

Comments
 (0)