From fde20cac81af8abfe2bac7e0696171880910e586 Mon Sep 17 00:00:00 2001
From: FlightlessMango <flightlessmangoyt@gmail.com>
Date: Fri, 6 Nov 2020 22:04:10 +0100
Subject: [PATCH] Collect total vram for Nvidia

---
 src/gpu.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/gpu.cpp b/src/gpu.cpp
index 57724f38..0c8119d0 100644
--- a/src/gpu.cpp
+++ b/src/gpu.cpp
@@ -34,6 +34,7 @@ void getNvidiaGpuInfo(){
         gpu_info.CoreClock = nvidiaCoreClock;
         gpu_info.MemClock = nvidiaMemClock;
         gpu_info.powerUsage = nvidiaPowerUsage / 1000;
+        gpu_info.memoryTotal = nvidiaMemory.total / (1024.f * 1024.f * 1024.f);
         return;
     }
 #endif
@@ -46,6 +47,7 @@ void getNvidiaGpuInfo(){
         gpu_info.CoreClock = nvctrl_info.CoreClock;
         gpu_info.MemClock = nvctrl_info.MemClock;
         gpu_info.powerUsage = 0;
+        gpu_info.memoryTotal = nvctrl_info.memoryTotal;
         return;
     }
 #endif