[rocprofiler-compute] Handle amd-smi rocm==6.4.2 json output issue. (#667)

此提交包含在:
xuchen-amd
2025-08-21 10:06:04 -04:00
提交者 GitHub
父節點 b2857b5db9
當前提交 6bf5648c6d
+7 -3
查看文件
@@ -171,9 +171,13 @@ def generate_machine_specs(args, sysinfo: dict = None):
vbios = gpu_data.get("vbios", {}).get("part_number")
# Load amd-smi partition data for GPU 0 (amd-smi >= 26.0.0)
partition_data = json.loads(
run(["amd-smi", "partition", "--gpu=0", "--json"], exit_on_error=False)
)
try:
partition_data = json.loads(
run(["amd-smi", "partition", "--gpu=0", "--json"], exit_on_error=False)
)
except json.JSONDecodeError:
partition_data = {}
current_partition = partition_data.get("current_partition", [{}])[0]
# Extract partition values with gpu_data fallback (amd-smi < 26.0.0)