IntenseWebs/tuxclocker
3
0
Fork 0
mirror of https://github.com/Lurkki14/tuxclocker.git synced 2024-11-25 01:30:18 -06:00
Code Issues Packages Projects Releases Wiki Activity

add NVIDIA VRAM nodes

Browse Source
This commit is contained in:
Jussi Kuokkanen 2023-11-15 11:17:52 +02:00
parent 7d26a0a2e9
commit be14f08a66
1 changed files with 69 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
src/plugins/Nvidia.cpp
View File

@ -289,6 +289,70 @@ std::vector<TreeNode<DeviceNode>> getMemClockRead(NvidiaGPUData data) {
return {};
}
std::vector<TreeNode<DeviceNode>> getTotalVram(NvidiaGPUData data) {
nvmlMemory_t mem;
if (nvmlDeviceGetMemoryInfo(data.devHandle, &mem) != NVML_SUCCESS)
return {};
// B -> MB
auto total = static_cast<uint>(mem.total / 1000000ull);
StaticReadable sr{total, _("MB")};
return {DeviceNode{
.name = _("Total Memory"),
.interface = sr,
.hash = md5(data.uuid + "Total VRAM"),
}};
}
std::vector<TreeNode<DeviceNode>> getReservedVram(NvidiaGPUData data) {
auto func = [=]() -> ReadResult {
nvmlMemory_v2_t mem;
if (nvmlDeviceGetMemoryInfo_v2(data.devHandle, &mem) != NVML_SUCCESS)
return ReadError::UnknownError;
// B -> MB
return static_cast<uint>(mem.reserved / 1000000ull);
};
DynamicReadable dr{func, _("MB")};
if (hasReadableValue(func()))
return {DeviceNode{
.name = _("Reserved Memory"),
.interface = dr,
.hash = md5(data.uuid + "Reserved VRAM"),
}};
return {};
}
std::vector<TreeNode<DeviceNode>> getUsedVram(NvidiaGPUData data) {
auto func = [=]() -> ReadResult {
nvmlMemory_t mem;
if (nvmlDeviceGetMemoryInfo(data.devHandle, &mem) != NVML_SUCCESS)
return ReadError::UnknownError;
// B -> MB
return static_cast<uint>(mem.used / 1000000ull);
};
DynamicReadable dr{func, _("MB")};
if (hasReadableValue(func()))
return {DeviceNode{
.name = _("Used Memory"),
.interface = dr,
.hash = md5(data.uuid + "Used VRAM"),
}};
return {};
}
std::vector<TreeNode<DeviceNode>> getVramRoot(NvidiaGPUData data) {
return {DeviceNode{
.name = _("Video Memory"),
.interface = std::nullopt,
.hash = md5(data.uuid + "VRAM Root"),
}};
}
std::vector<TreeNode<DeviceNode>> getClocksRoot(NvidiaGPUData data) {
// TODO: this gets added unconditionally
// If leaf nodes without an interface become a problem, we can remove them
@ -775,6 +839,11 @@ auto gpuTree = TreeConstructor<NvidiaGPUData, DeviceNode>{
{getSingleFanSpeedWrite, {}},
{getSingleFanMode, {}}
}},
{getVramRoot, {
{getTotalVram, {}},
{getUsedVram, {}},
{getReservedVram, {}}
}},
{getPowerUsage, {}},
{getPowerLimit, {}},
{getVoltage, {}},