AI/rocm-systems
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

Added GTT Memory to process table of default command

Telusuri Sumber 
Signed-off-by: gabrpham <Gabriel.Pham@amd.com>
This commit is contained in:
gabrpham
2025-06-12 16:44:47 -05:00
committed by Arif, Maisam
orang tua 8a0e65d911
melakukan 9e221a3f09
1 mengubah file dengan 0 tambahan dan 5 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
amdsmi_cli/amdsmi_commands.py
-5
Melihat File
@@ -6660,14 +6660,9 @@ 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: