AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-455131 - Updated process APIs

Browse Source 
- Removed amdsmi_get_gpu_process_info from python API
  - Updated documentation
  - Aligned process --json output format to unit & value format

Signed-off-by: Maisam Arif <maisarif@amd.com>
Change-Id: I82bba1b6df71020b4a5995ff63b9aa62611ce4fe
This commit is contained in:
Maisam Arif
2024-04-11 05:11:54 -05:00
parent c06be55b6a
commit c551c3caed
5 changed files with 53 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py-interface/amdsmi_interface.py
+17 -30
View File
@@ -1799,12 +1799,15 @@ def amdsmi_get_gpu_bad_page_info(
num_pages = ctypes.c_uint32()
retired_page_record = ctypes.POINTER(
amdsmi_wrapper.amdsmi_retired_page_record_t)()
_check_res(
amdsmi_wrapper.amdsmi_get_gpu_bad_page_info(
processor_handle, ctypes.byref(num_pages), retired_page_record
)
)
table_records = _format_bad_page_info(retired_page_record, num_pages)
if num_pages.value == 0:
return "No bad pages found."
else:
@@ -1942,39 +1945,23 @@ def amdsmi_get_gpu_process_list(
result = []
for index in range(max_processes.value):
result.append(process_list[index])
result.append({
"name": process_list[index].name.decode("utf-8"),
"pid": process_list[index].pid,
"mem": process_list[index].mem,
"engine_usage": {
"gfx": process_list[index].engine_usage.gfx,
"enc": process_list[index].engine_usage.enc
},
"memory_usage": {
"gtt_mem": process_list[index].memory_usage.gtt_mem,
"cpu_mem": process_list[index].memory_usage.cpu_mem,
"vram_mem": process_list[index].memory_usage.vram_mem,
},
})
return result
def amdsmi_get_gpu_process_info(
processor_handle: amdsmi_wrapper.amdsmi_processor_handle,
process: amdsmi_wrapper.amdsmi_proc_info_t,
) -> Dict[str, Any]:
if not isinstance(processor_handle, amdsmi_wrapper.amdsmi_processor_handle):
raise AmdSmiParameterException(
processor_handle, amdsmi_wrapper.amdsmi_processor_handle
)
if not isinstance(process, amdsmi_wrapper.amdsmi_proc_info_t):
raise AmdSmiParameterException(
process, amdsmi_wrapper.amdsmi_proc_info_t)
return {
"name": process.name.decode("utf-8"),
"pid": process.pid,
"mem": process.mem,
"engine_usage": {
"gfx": process.engine_usage.gfx,
"enc": process.engine_usage.enc
},
"memory_usage": {
"gtt_mem": process.memory_usage.gtt_mem,
"cpu_mem": process.memory_usage.cpu_mem,
"vram_mem": process.memory_usage.vram_mem,
},
}
def amdsmi_get_gpu_device_uuid(processor_handle: amdsmi_wrapper.amdsmi_processor_handle) -> str:
if not isinstance(processor_handle, amdsmi_wrapper.amdsmi_processor_handle):
raise AmdSmiParameterException(