From d97719e3d674384120e3771b8944e3700efa45d8 Mon Sep 17 00:00:00 2001
From: FlightlessMango <flightlessmangoyt@gmail.com>
Date: Tue, 4 Feb 2020 07:11:13 +0100
Subject: [PATCH] get nvidia vram from nvml

---
 src/loaders/loader_nvml.h | 2 +-
 src/nvidia_info.h         | 1 +
 src/nvml.cpp              | 2 ++
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/loaders/loader_nvml.h b/src/loaders/loader_nvml.h
index 40890d6..5b9522e 100644
--- a/src/loaders/loader_nvml.h
+++ b/src/loaders/loader_nvml.h
@@ -29,7 +29,7 @@ class libnvml_loader {
   decltype(&::nvmlDeviceGetCount_v2) nvmlDeviceGetCount_v2;
   decltype(&::nvmlDeviceGetHandleByIndex_v2) nvmlDeviceGetHandleByIndex_v2;
   decltype(&::nvmlDeviceGetHandleByPciBusId_v2) nvmlDeviceGetHandleByPciBusId_v2;
-
+  decltype(&::nvmlDeviceGetMemoryInfo) nvmlDeviceGetMemoryInfo;
 
  private:
   void CleanUp(bool unload);
diff --git a/src/nvidia_info.h b/src/nvidia_info.h
index 830ac57..b906c41 100644
--- a/src/nvidia_info.h
+++ b/src/nvidia_info.h
@@ -7,6 +7,7 @@ extern unsigned int nvidiaTemp, processSamplesCount, lastSeenTimeStamp, *vgpuIns
 extern nvmlValueType_t *sampleValType;
 extern nvmlDevice_t nvidiaDevice;
 extern struct nvmlUtilization_st nvidiaUtilization;
+extern struct nvmlMemory_st nvidiaMemory;
 extern bool nvmlSuccess;
 
 bool checkNvidia(void);
diff --git a/src/nvml.cpp b/src/nvml.cpp
index c161dab..213a920 100644
--- a/src/nvml.cpp
+++ b/src/nvml.cpp
@@ -8,6 +8,7 @@ nvmlDevice_t nvidiaDevice;
 bool nvmlSuccess = false;
 unsigned int nvidiaTemp;
 struct nvmlUtilization_st nvidiaUtilization;
+struct nvmlMemory_st nvidiaMemory;
 
 bool checkNvidia(){
     if (nvml.IsLoaded()){
@@ -28,4 +29,5 @@ void getNvidiaInfo(){
     nvml.nvmlDeviceGetHandleByIndex(0, &nvidiaDevice);
     nvml.nvmlDeviceGetUtilizationRates(nvidiaDevice, &nvidiaUtilization);
     nvml.nvmlDeviceGetTemperature(nvidiaDevice, NVML_TEMPERATURE_GPU, &nvidiaTemp);
+    nvml.nvmlDeviceGetMemoryInfo(nvidiaDevice, &nvidiaMemory);
 }
\ No newline at end of file