From f9bfb746fbb693a578f5dc04b1f87ca187d3f45f Mon Sep 17 00:00:00 2001 From: Maisam Arif Date: Thu, 6 Jun 2024 16:28:19 -0500 Subject: [PATCH] Update Python API README example code Signed-off-by: Maisam Arif Change-Id: I5dbb2e3cdab31b41e6f502d3257fe899eed1ee97 --- py-interface/README.md | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/py-interface/README.md b/py-interface/README.md index 93f5245248..21fb77d338 100644 --- a/py-interface/README.md +++ b/py-interface/README.md @@ -1588,8 +1588,12 @@ try: print("No GPUs on machine") else: for device in devices: - memory = amdsmi_get_gpu_memory_total(device) - print(memory) + vram_memory_total = amdsmi_get_gpu_memory_total(device, amdsmi_interface.AmdSmiMemoryType.VRAM) + print(vram_memory_total) + vis_vram_memory_total = amdsmi_get_gpu_memory_total(device, amdsmi_interface.AmdSmiMemoryType.VIS_VRAM) + print(vis_vram_memory_total) + gtt_memory_total = amdsmi_get_gpu_memory_total(device, amdsmi_interface.AmdSmiMemoryType.GTT) + print(gtt_memory_total) except AmdSmiException as e: print(e) ``` @@ -1660,8 +1664,12 @@ try: print("No GPUs on machine") else: for device in devices: - memory = amdsmi_get_gpu_memory_usage(device) - print(memory) + vram_memory_usage = amdsmi_get_gpu_memory_usage(device, amdsmi_interface.AmdSmiMemoryType.VRAM) + print(vram_memory_usage) + vis_vram_memory_usage = amdsmi_get_gpu_memory_usage(device, amdsmi_interface.AmdSmiMemoryType.VIS_VRAM) + print(vis_vram_memory_usage) + gtt_memory_usage = amdsmi_get_gpu_memory_usage(device, amdsmi_interface.AmdSmiMemoryType.GTT) + print(gtt_memory_usage) except AmdSmiException as e: print(e) ```