Add nvmlDeviceGetPowerUsage import

pull/197/head
jackun 4 years ago
parent 9b595cef1a
commit 5c3c8a5ac9
No known key found for this signature in database
GPG Key ID: 119DB3F1D05A9ED3

@ -174,6 +174,19 @@ bool libnvml_loader::Load(const std::string& library_name) {
return false;
}
#if defined(LIBRARY_LOADER_NVML_H_DLOPEN)
nvmlDeviceGetPowerUsage =
reinterpret_cast<decltype(this->nvmlDeviceGetPowerUsage)>(
dlsym(library_, "nvmlDeviceGetPowerUsage"));
#endif
#if defined(LIBRARY_LOADER_NVML_H_DT_NEEDED)
nvmlDeviceGetPowerUsage = &::nvmlDeviceGetPowerUsage;
#endif
if (!nvmlDeviceGetPowerUsage) {
CleanUp(true);
return false;
}
loaded_ = true;
return true;
}

@ -36,6 +36,7 @@ class libnvml_loader {
decltype(&::nvmlDeviceGetMemoryInfo) nvmlDeviceGetMemoryInfo;
decltype(&::nvmlDeviceGetClockInfo) nvmlDeviceGetClockInfo;
decltype(&::nvmlErrorString) nvmlErrorString;
decltype(&::nvmlDeviceGetPowerUsage) nvmlDeviceGetPowerUsage;
private:
void CleanUp(bool unload);

Loading…
Cancel
Save