Add support to retrieve process VRAM usage information.

Change-Id: I60843a99207a658022a26aa346b79f91863833cf

include/rocm_smi/rocm_smi.h

@@ -712,6 +712,7 @@ typedef struct {
 typedef struct {
     uint32_t process_id;  //!< Process ID
     uint32_t pasid;    //!< PASID
+    uint64_t vram_usage;  //!< VRAM usage
 } rsmi_process_info_t;
 
 

include/rocm_smi/rocm_smi_kfd.h

@@ -98,7 +98,8 @@ int
 GetProcessInfo(rsmi_process_info_t *procs, uint32_t num_allocated,
                                                    uint32_t *num_procs_found);
 int
-GetProcessInfoForPID(uint32_t pid, rsmi_process_info_t *proc);
+GetProcessInfoForPID(uint32_t pid, rsmi_process_info_t *proc,
+                     std::unordered_set<uint64_t> *gpu_set);
 
 int
 GetProcessGPUs(uint32_t pid, std::unordered_set<uint64_t> *gpu_count);