Skip to content

Commit bc41720

Browse files
committed
refactor(amd): enhance uuid getting
Signed-off-by: thxCode <[email protected]>
1 parent 2be423e commit bc41720

File tree

1 file changed

+5
-1
lines changed
  • gpustack_runtime/detector

1 file changed

+5
-1
lines changed

gpustack_runtime/detector/amd.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,11 @@ def detect(self) -> Devices | None:
107107
asic_serial = dev_gpu_asic_info.get("asic_serial")
108108
dev_uuid = f"GPU-{(asic_serial[2:]).lower()}"
109109
else:
110-
dev_uuid = f"GPU-{pyrocmsmi.rsmi_dev_unique_id_get(dev_idx)[2:]}"
110+
dev_uuid = ""
111+
with contextlib.suppress(pyrocmsmi.ROCMSMIError):
112+
dev_uuid = (
113+
f"GPU-{pyrocmsmi.rsmi_dev_unique_id_get(dev_idx)[2:]}"
114+
)
111115
dev_hsa_agent = hsa_agents.get(dev_uuid, pyhsa.Agent())
112116

113117
dev_gpu_driver_info = pyamdsmi.amdsmi_get_gpu_driver_info(dev)

0 commit comments

Comments
 (0)