From 2e8dc4f2a91bfa7661f4ea289736b12153ce23c2 Mon Sep 17 00:00:00 2001
From: Divya Shikre <DivyaUday.Shikre@amd.com>
Date: Fri, 14 Aug 2020 12:12:34 -0400
Subject: [PATCH] Adding Sdma Usage to showpids

Signed-off-by: Divya Shikre <DivyaUday.Shikre@amd.com
Change-Id: I72a9e1adc61eba382f1ac17c8e50b2a8bd6d6898
---
 python_smi_tools/rocmSmiLib_cli.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/python_smi_tools/rocmSmiLib_cli.py b/python_smi_tools/rocmSmiLib_cli.py
index ba5f79132c..d1c999f93d 100755
--- a/python_smi_tools/rocmSmiLib_cli.py
+++ b/python_smi_tools/rocmSmiLib_cli.py
@@ -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()