Signed-off-by: Divya Shikre <DivyaUday.Shikre@amd.com
Change-Id: I72a9e1adc61eba382f1ac17c8e50b2a8bd6d6898


[ROCm/amdsmi commit: 2e8dc4f2a9]
Этот коммит содержится в:
Divya Shikre
2020-08-14 12:12:34 -04:00
родитель 8acb8b76cf
Коммит bdda71d7a7
+4 -2
Просмотреть файл
@@ -1308,7 +1308,7 @@ def showPids():
""" Show Information for PIDs created in a KFD (Compute) context """
printLogSpacer(' KFD Processes ')
dataArray = []
dataArray.append(['PID', 'PROCESS NAME', 'GPU(s)', 'VRAM USED'])
dataArray.append(['PID', 'PROCESS NAME', 'GPU(s)', 'VRAM USED', 'SDMA USED'])
pidList = getPidList()
if not pidList:
printLog(None, 'No KFD PIDs currently running', None)
@@ -1320,6 +1320,7 @@ def showPids():
for pid in pidList:
gpuNumber = 'UNKNOWN'
vramUsage = 'UNKNOWN'
sdmaUsage = 'UNKNOWN'
ret = rocmsmi.rsmi_compute_process_gpus_get(int(pid), None, byref(num_devices))
if rsmi_ret_ok(ret):
dv_indices = (c_uint32 * num_devices.value)()
@@ -1331,9 +1332,10 @@ def showPids():
ret = rocmsmi.rsmi_compute_process_info_by_pid_get(int(pid), byref(proc))
if rsmi_ret_ok(ret):
vramUsage = proc.vram_usage
sdmaUsage = proc.sdma_usage
else:
logging.debug('Unable to fetch PID VRAM information')
dataArray.append([pid, getName(pid), str(gpuNumber), str(vramUsage)])
dataArray.append([pid, getName(pid), str(gpuNumber), str(vramUsage), str(sdmaUsage)])
printLog(None, 'KFD process information:', None)
print2DArray(dataArray)
printLogSpacer()