From 8a0e65d911ecf34a669b33f5e2b7f07bd2233c3e Mon Sep 17 00:00:00 2001 From: gabrpham Date: Thu, 12 Jun 2025 16:44:47 -0500 Subject: [PATCH] Added GTT Memory to default command and adjusted table format Signed-off-by: gabrpham --- amdsmi_cli/amdsmi_commands.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/amdsmi_cli/amdsmi_commands.py b/amdsmi_cli/amdsmi_commands.py index f3896bb388..86de9bb3f8 100644 --- a/amdsmi_cli/amdsmi_commands.py +++ b/amdsmi_cli/amdsmi_commands.py @@ -6660,9 +6660,14 @@ class AMDSMICommands(): total_vram = amdsmi_interface.amdsmi_get_gpu_memory_total(processor, amdsmi_interface.AmdSmiMemoryType.VRAM) // (1024*1024) used_vram = amdsmi_interface.amdsmi_get_gpu_memory_usage(processor, amdsmi_interface.AmdSmiMemoryType.VRAM) // (1024*1024) mem_usage = {"used_vram": used_vram, "total_vram": total_vram} + total_gtt =amdsmi_interface.amdsmi_get_gpu_memory_total(processor, amdsmi_interface.AmdSmiMemoryType.GTT) // (1024*1024) + used_gtt = amdsmi_interface.amdsmi_get_gpu_memory_usage(processor, amdsmi_interface.AmdSmiMemoryType.GTT) // (1024*1024) + gtt_usage = {"used_gtt": used_gtt, "total_gtt": total_gtt} except amdsmi_exception.AmdSmiLibraryException as e: mem_usage = "N/A" + gtt_usage = "N/A" gpu_info_dict.update({"mem_usage": mem_usage}) + gpu_info_dict.update({"gtt_usage": gtt_usage}) # uncorrectable ECC errors try: