Skip to content

Commit ff2f172

Browse files
committed
diag(power): preflight v4 handle gpu_data wrapper
中文:raw dump 确认 AMDSMI 26.2.0 顶层为 gpu_data 数组;socket_power/total_energy_consumption 均为 value/unit 嵌套且能量单位已是 J。
1 parent 579e0de commit ff2f172

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

‎utils/mi355x_power_preflight.py‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,9 @@ def gpu_entries(doc):
3535
if isinstance(doc, list):
3636
return doc
3737
if isinstance(doc, dict):
38-
if isinstance(doc.get("gpu"), list):
39-
return doc["gpu"]
38+
for key in ("gpu_data", "gpu"):
39+
if isinstance(doc.get(key), list):
40+
return doc[key]
4041
if all(isinstance(v, dict) for v in doc.values()) and doc:
4142
return [dict(v, gpu=k) for k, v in doc.items()]
4243
return []

0 commit comments

Comments
 (0)