From fb2ed24372aa3965ef65326724acf52cf1decd35 Mon Sep 17 00:00:00 2001
From: Mukul Joshi <mukul.joshi@amd.com>
Date: Mon, 21 Sep 2020 12:28:22 -0400
Subject: [PATCH] Use correct string conversion function for VRAM and SDMA
 usage

VRAM and SDMA usage can be 64-bit long numbers. Use stoull()
instead of stoi() to convert the VRAM and SDMA usage strings to
numbers.

Change-Id: Ifadbada9f33320fc67666036ce8439823c1d1fb7
---
 src/rocm_smi_kfd.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/rocm_smi_kfd.cc b/src/rocm_smi_kfd.cc
index b70f22c25b..fdb189ed20 100755
--- a/src/rocm_smi_kfd.cc
+++ b/src/rocm_smi_kfd.cc
@@ -451,7 +451,7 @@ int GetProcessInfoForPID(uint32_t pid, rsmi_process_info_t *proc,
       return EINVAL;
     }
 
-    proc->vram_usage += std::stoi(tmp);
+    proc->vram_usage += std::stoull(tmp);
 
     std::string sdma_str_path = proc_str_path;
     sdma_str_path += "/sdma_";
@@ -466,7 +466,7 @@ int GetProcessInfoForPID(uint32_t pid, rsmi_process_info_t *proc,
       return EINVAL;
     }
 
-    proc->sdma_usage += std::stoi(tmp);
+    proc->sdma_usage += std::stoull(tmp);
   }
 
   return 0;