From 6bf5648c6dc0efe46760b98a2a32513904f32081 Mon Sep 17 00:00:00 2001 From: xuchen-amd Date: Thu, 21 Aug 2025 10:06:04 -0400 Subject: [PATCH] [rocprofiler-compute] Handle amd-smi rocm==6.4.2 json output issue. (#667) --- projects/rocprofiler-compute/src/utils/specs.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/projects/rocprofiler-compute/src/utils/specs.py b/projects/rocprofiler-compute/src/utils/specs.py index eb7ce7e648..ed01fb3bef 100644 --- a/projects/rocprofiler-compute/src/utils/specs.py +++ b/projects/rocprofiler-compute/src/utils/specs.py @@ -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)