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)