[rocprofiler-compute] Handle amd-smi rocm==6.4.2 json output issue. (#667)
此提交包含在:
@@ -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)
|
||||
|
||||
新增問題並參考
封鎖使用者