Add API for the memory type

Get the memory type from libdrm and add a new API. Change-Id: I89327bca2ef860f2e3f4f6ca20def2331eba66c0
2023-08-31 09:08:46 -05:00
parent f96c7663b5
commit b52034fed8
7 changed files with 145 additions and 13 deletions
@@ -995,7 +995,7 @@ def amdsmi_get_gpu_vram_usage(
            processor_handle, amdsmi_wrapper.amdsmi_processor_handle
        )

-    vram_info = amdsmi_wrapper.amdsmi_vram_info_t()
+    vram_info = amdsmi_wrapper.amdsmi_vram_usage_t()
    _check_res(
        amdsmi_wrapper.amdsmi_get_gpu_vram_usage(
            processor_handle, ctypes.byref(vram_info))