From f82063d19f82c398ca4d41c1582ddbee1ba650a2 Mon Sep 17 00:00:00 2001 From: Maisam Arif Date: Fri, 22 Sep 2023 06:49:58 -0500 Subject: [PATCH] Updated metric --usage labels & error handling Signed-off-by: Maisam Arif Change-Id: Ie376785deeae2df562962c18f4395db87025181b --- amdsmi_cli/amdsmi_commands.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/amdsmi_cli/amdsmi_commands.py b/amdsmi_cli/amdsmi_commands.py index 00bddc532a..65d61ba873 100644 --- a/amdsmi_cli/amdsmi_commands.py +++ b/amdsmi_cli/amdsmi_commands.py @@ -829,16 +829,18 @@ class AMDSMICommands(): if args.usage: try: engine_usage = amdsmi_interface.amdsmi_get_gpu_activity(args.gpu) + engine_usage['gfx_usage'] = engine_usage.pop('gfx_activity') + engine_usage['mem_usage'] = engine_usage.pop('umc_activity') + engine_usage['mm_ip_usage'] = engine_usage.pop('mm_activity') - if self.logger.is_gpuvsmi_compatibility(): - engine_usage['gfx_usage'] = engine_usage.pop('gfx_activity') - engine_usage['mem_usage'] = engine_usage.pop('umc_activity') - engine_usage['mm_usage_list'] = engine_usage.pop('mm_activity') + for key, value in engine_usage.items(): + if value == 65535: + engine_usage[key] = "N/A" - if self.logger.is_human_readable_format(): - unit = '%' - for usage_name, usage_value in engine_usage.items(): - engine_usage[usage_name] = f"{usage_value} {unit}" + if self.logger.is_human_readable_format(): + if engine_usage[key] != "N/A": + unit = '%' + engine_usage[key] = f"{value} {unit}" values_dict['usage'] = engine_usage except amdsmi_exception.AmdSmiLibraryException as e: