You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.6 KiB
C++
86 lines
2.6 KiB
C++
#include "nvidia_info.h"
|
|
#include "memory.h"
|
|
#include "gpu.h"
|
|
|
|
struct gpuInfo gpu_info;
|
|
FILE *amdGpuFile = nullptr, *amdTempFile = nullptr, *amdGpuVramTotalFile = nullptr, *amdGpuVramUsedFile = nullptr, *amdGpuCoreClockFile = nullptr, *amdGpuMemoryClockFile = nullptr;
|
|
|
|
void getNvidiaGpuInfo(){
|
|
if (nvmlSuccess){
|
|
getNVMLInfo();
|
|
gpu_info.load = nvidiaUtilization.gpu;
|
|
gpu_info.temp = nvidiaTemp;
|
|
gpu_info.memoryUsed = nvidiaMemory.used / (1024.f * 1024.f * 1024.f);
|
|
gpu_info.CoreClock = nvidiaCoreClock;
|
|
gpu_info.MemClock = nvidiaMemClock;
|
|
}
|
|
#ifdef HAVE_XNVCTRL
|
|
else if (nvctrlSuccess) {
|
|
getNvctrlInfo();
|
|
gpu_info.load = nvctrl_info.load;
|
|
gpu_info.temp = nvctrl_info.temp;
|
|
gpu_info.memoryUsed = nvctrl_info.memoryUsed / (1024.f);
|
|
gpu_info.CoreClock = nvctrl_info.CoreClock;
|
|
gpu_info.MemClock = nvctrl_info.MemClock;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void getAmdGpuUsage(){
|
|
int64_t value = 0;
|
|
|
|
if (amdGpuFile) {
|
|
rewind(amdGpuFile);
|
|
fflush(amdGpuFile);
|
|
if (fscanf(amdGpuFile, "%d", &amdgpu.load) != 1)
|
|
amdgpu.load = 0;
|
|
gpu_info.load = amdgpu.load;
|
|
}
|
|
|
|
if (amdTempFile) {
|
|
rewind(amdTempFile);
|
|
fflush(amdTempFile);
|
|
if (fscanf(amdTempFile, "%d", &amdgpu.temp) != 1)
|
|
amdgpu.temp = 0;
|
|
amdgpu.temp /= 1000;
|
|
gpu_info.temp = amdgpu.temp;
|
|
}
|
|
|
|
if (amdGpuVramTotalFile) {
|
|
rewind(amdGpuVramTotalFile);
|
|
fflush(amdGpuVramTotalFile);
|
|
if (fscanf(amdGpuVramTotalFile, "%" PRId64, &value) != 1)
|
|
value = 0;
|
|
amdgpu.memoryTotal = float(value) / (1024 * 1024 * 1024);
|
|
gpu_info.memoryTotal = amdgpu.memoryTotal;
|
|
}
|
|
|
|
if (amdGpuVramUsedFile) {
|
|
rewind(amdGpuVramUsedFile);
|
|
fflush(amdGpuVramUsedFile);
|
|
if (fscanf(amdGpuVramUsedFile, "%" PRId64, &value) != 1)
|
|
value = 0;
|
|
amdgpu.memoryUsed = float(value) / (1024 * 1024 * 1024);
|
|
gpu_info.memoryUsed = amdgpu.memoryUsed;
|
|
}
|
|
|
|
if (amdGpuCoreClockFile) {
|
|
rewind(amdGpuCoreClockFile);
|
|
fflush(amdGpuCoreClockFile);
|
|
if (fscanf(amdGpuCoreClockFile, "%" PRId64, &value) != 1)
|
|
value = 0;
|
|
|
|
amdgpu.CoreClock = value / 1000000;
|
|
gpu_info.CoreClock = amdgpu.CoreClock;
|
|
}
|
|
|
|
if (amdGpuMemoryClockFile) {
|
|
rewind(amdGpuMemoryClockFile);
|
|
fflush(amdGpuMemoryClockFile);
|
|
if (fscanf(amdGpuMemoryClockFile, "%" PRId64, &value) != 1)
|
|
value = 0;
|
|
|
|
amdgpu.MemClock = value / 1000000;
|
|
gpu_info.MemClock = amdgpu.MemClock;
|
|
}
|
|
} |