From fe1dc23ade70383dc11660bb468c9d77fd3bc0a0 Mon Sep 17 00:00:00 2001 From: gabrpham Date: Thu, 1 Aug 2024 11:30:23 -0500 Subject: [PATCH] Fixed 'amd-smi process -G' Issue linked here: https://github.com/ROCm/amdsmi/issues/23 Signed-off-by: gabrpham Change-Id: I73c2dede8634b21a5dfe0245a202e883fa856de2 --- amdsmi_cli/amdsmi_commands.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/amdsmi_cli/amdsmi_commands.py b/amdsmi_cli/amdsmi_commands.py index a4583e091c..9d4e7ebc07 100644 --- a/amdsmi_cli/amdsmi_commands.py +++ b/amdsmi_cli/amdsmi_commands.py @@ -2679,14 +2679,15 @@ class AMDSMICommands(): # General and Engine to expose process_info values if args.general or args.engine: for process_info in filtered_process_values: - if args.general and args.engine: - del process_info['process_info']['memory_usage'] - elif args.general: - del process_info['process_info']['memory_usage'] - del process_info['process_info']['usage'] # Used in engine - elif args.engine: - del process_info['process_info']['memory_usage'] - del process_info['process_info']['mem_usage'] # Used in general + if not process_info['process_info'] == "N/A": + if args.general and args.engine: + del process_info['process_info']['memory_usage'] + elif args.general: + del process_info['process_info']['memory_usage'] + del process_info['process_info']['usage'] # Used in engine + elif args.engine: + del process_info['process_info']['memory_usage'] + del process_info['process_info']['mem_usage'] # Used in general # Filter out non specified pids if args.pid: