diff options
Diffstat (limited to 'src/pmdas/nvidia/localnvml.c')
| -rw-r--r-- | src/pmdas/nvidia/localnvml.c | 270 |
1 files changed, 270 insertions, 0 deletions
diff --git a/src/pmdas/nvidia/localnvml.c b/src/pmdas/nvidia/localnvml.c new file mode 100644 index 0000000..2cadeb9 --- /dev/null +++ b/src/pmdas/nvidia/localnvml.c @@ -0,0 +1,270 @@ +/* + * Copyright (c) 2014 Red Hat. + * + * This program is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License as published by the + * Free Software Foundation; either version 2 of the License, or (at your + * option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY + * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * for more details. + */ +#include "pmapi.h" +#include "impl.h" +#if defined(HAVE_DLFCN_H) +#include <dlfcn.h> +#endif +#include "localnvml.h" + +/* + * Implements NVML interfaces based on: + * http://docs.nvidia.com/deploy/nvml-api/index.html + * ... using either a dlopen'd 3rd party or "no values available". + */ + +struct { + const char *symbol; + void *handle; +} nvml_symtab[] = { + { .symbol = "nvmlInit" }, + { .symbol = "nvmlShutdown" }, + { .symbol = "nvmlDeviceGetCount" }, + { .symbol = "nvmlDeviceGetHandleByIndex" }, + { .symbol = "nvmlDeviceGetName" }, + { .symbol = "nvmlDeviceGetPciInfo" }, + { .symbol = "nvmlDeviceGetFanSpeed" }, + { .symbol = "nvmlDeviceGetTemperature" }, + { .symbol = "nvmlDeviceGetUtilizationRates" }, + { .symbol = "nvmlDeviceGetMemoryInfo" }, + { .symbol = "nvmlDeviceGetPerformanceState" }, +}; +enum { + NVML_INIT, + NVML_SHUTDOWN, + NVML_DEVICE_GET_COUNT, + NVML_DEVICE_GET_HANDLEBYINDEX, + NVML_DEVICE_GET_NAME, + NVML_DEVICE_GET_PCIINFO, + NVML_DEVICE_GET_FANSPEED, + NVML_DEVICE_GET_TEMPERATURE, + NVML_DEVICE_GET_UTILIZATIONRATES, + NVML_DEVICE_GET_MEMORYINFO, + NVML_DEVICE_GET_PERFORMANCESTATE, + NVML_SYMBOL_COUNT +}; +typedef int (*local_init_t)(void); +typedef int (*local_shutdown_t)(void); +typedef int (*local_dev_get_count_t)(unsigned int *); +typedef int (*local_dev_get_handlebyindex_t)(unsigned int, nvmlDevice_t *); +typedef int (*local_dev_get_name_t)(nvmlDevice_t, char *, unsigned int); +typedef int (*local_dev_get_pciinfo_t)(nvmlDevice_t, nvmlPciInfo_t *); +typedef int (*local_dev_get_fanspeed_t)(nvmlDevice_t, unsigned int *); +typedef int (*local_dev_get_temperature_t)(nvmlDevice_t, nvmlTemperatureSensors_t, unsigned int *); +typedef int (*local_dev_get_utilizationrates_t)(nvmlDevice_t, nvmlUtilization_t *); +typedef int (*local_dev_get_memoryinfo_t)(nvmlDevice_t, nvmlMemory_t *); +typedef int (*local_dev_get_performancestate_t)(nvmlDevice_t, nvmlPstates_t *); + +static int +resolve_symbols(void) +{ + static void *nvml_dso; + int i; + + if (nvml_dso != NULL) + return 0; + if ((nvml_dso = dlopen("libnvidia-ml." DSOSUFFIX, RTLD_NOW)) == NULL) + return NVML_ERROR_LIBRARY_NOT_FOUND; + __pmNotifyErr(LOG_INFO, "Successfully loaded NVIDIA NVML library"); + for (i = 0; i < NVML_SYMBOL_COUNT; i++) + nvml_symtab[i].handle = dlsym(nvml_dso, nvml_symtab[i].symbol); + return 0; +} + +int +localNvmlInit(void) +{ + local_init_t init; + void *func; + int sts = resolve_symbols(); + + if (sts != 0) + return sts; + if ((func = nvml_symtab[NVML_INIT].handle) == NULL) + return NVML_ERROR_FUNCTION_NOT_FOUND; + init = (local_init_t)func; + return init(); +} + +int +localNvmlShutdown(void) +{ + local_shutdown_t shutdown; + void *func = nvml_symtab[NVML_SHUTDOWN].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + shutdown = (local_shutdown_t)func; + return shutdown(); +} + +int +localNvmlDeviceGetCount(unsigned int *count) +{ + local_dev_get_count_t dev_get_count; + void *func = nvml_symtab[NVML_DEVICE_GET_COUNT].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_count = (local_dev_get_count_t)func; + return dev_get_count(count); +} + +int +localNvmlDeviceGetHandleByIndex(unsigned int index, nvmlDevice_t *device) +{ + local_dev_get_handlebyindex_t dev_get_handlebyindex; + void *func = nvml_symtab[NVML_DEVICE_GET_HANDLEBYINDEX].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_handlebyindex = (local_dev_get_handlebyindex_t)func; + return dev_get_handlebyindex(index, device); +} + +int +localNvmlDeviceGetName(nvmlDevice_t device, char *name, unsigned int size) +{ + local_dev_get_name_t dev_get_name; + void *func = nvml_symtab[NVML_DEVICE_GET_NAME].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_name = (local_dev_get_name_t)func; + return dev_get_name(device, name, size); +} + +int +localNvmlDeviceGetPciInfo(nvmlDevice_t device, nvmlPciInfo_t *info) +{ + local_dev_get_pciinfo_t dev_get_pciinfo; + void *func = nvml_symtab[NVML_DEVICE_GET_PCIINFO].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_pciinfo = (local_dev_get_pciinfo_t)func; + return dev_get_pciinfo(device, info); +} + +int +localNvmlDeviceGetFanSpeed(nvmlDevice_t device, unsigned int *speed) +{ + local_dev_get_fanspeed_t dev_get_fanspeed; + void *func = nvml_symtab[NVML_DEVICE_GET_FANSPEED].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_fanspeed = (local_dev_get_fanspeed_t)func; + return dev_get_fanspeed(device, speed); +} + +int +localNvmlDeviceGetTemperature(nvmlDevice_t device, nvmlTemperatureSensors_t code, unsigned int *temp) +{ + local_dev_get_temperature_t dev_get_temperature; + void *func = nvml_symtab[NVML_DEVICE_GET_TEMPERATURE].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_temperature = (local_dev_get_temperature_t)func; + return dev_get_temperature(device, code, temp); +} + +int +localNvmlDeviceGetUtilizationRates(nvmlDevice_t device, nvmlUtilization_t *util) +{ + local_dev_get_utilizationrates_t dev_get_utilizationrates; + void *func = nvml_symtab[NVML_DEVICE_GET_UTILIZATIONRATES].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_utilizationrates = (local_dev_get_utilizationrates_t)func; + return dev_get_utilizationrates(device, util); +} + +int +localNvmlDeviceGetMemoryInfo(nvmlDevice_t device, nvmlMemory_t *memory) +{ + local_dev_get_memoryinfo_t dev_get_memoryinfo; + void *func = nvml_symtab[NVML_DEVICE_GET_MEMORYINFO].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_memoryinfo = (local_dev_get_memoryinfo_t)func; + return dev_get_memoryinfo(device, memory); +} + +int +localNvmlDeviceGetPerformanceState(nvmlDevice_t device, nvmlPstates_t *state) +{ + local_dev_get_performancestate_t dev_get_performancestate; + void *func = nvml_symtab[NVML_DEVICE_GET_PERFORMANCESTATE].handle; + + if (!func) + return NVML_ERROR_FUNCTION_NOT_FOUND; + dev_get_performancestate = (local_dev_get_performancestate_t)func; + return dev_get_performancestate(device, state); +} + +const char * +localNvmlErrStr(nvmlReturn_t sts) +{ + int i; + static const char *unknown = "No such error code"; + static struct { + int code; + const char *msg; + } table[] = { { + NVML_SUCCESS, +"The operation was successful" }, { + NVML_ERROR_UNINITIALIZED, +"NVML was not first initialized with nvmlInit()" }, { + NVML_ERROR_INVALID_ARGUMENT, +"A supplied argument is invalid" }, { + NVML_ERROR_NOT_SUPPORTED, +"The requested operation is not available on target device" }, { + NVML_ERROR_NO_PERMISSION, +"The current user does not have permission for operation" }, { + NVML_ERROR_ALREADY_INITIALIZED, +"Deprecated error code (5)" }, { + NVML_ERROR_NOT_FOUND, +"A query to find an object was unsuccessful" }, { + NVML_ERROR_INSUFFICIENT_SIZE, +"An input argument is not large enough" }, { + NVML_ERROR_INSUFFICIENT_POWER, +"A device's external power cables are not properly attached" }, { + NVML_ERROR_DRIVER_NOT_LOADED, +"NVIDIA driver is not loaded" }, { + NVML_ERROR_TIMEOUT, +"User provided timeout passed" }, { + NVML_ERROR_IRQ_ISSUE, +"NVIDIA Kernel detected an interrupt issue with a GPU" }, { + NVML_ERROR_LIBRARY_NOT_FOUND, +"NVML Shared Library couldn't be found or loaded" }, { + NVML_ERROR_FUNCTION_NOT_FOUND, +"Local version of NVML doesn't implement this function" }, { + NVML_ERROR_CORRUPTED_INFOROM, +"infoROM is corrupted" }, { + NVML_ERROR_GPU_IS_LOST, +"The GPU has fallen off the bus or has otherwise become inaccessible" }, { + NVML_ERROR_UNKNOWN, +"An internal driver error occurred" + } }; + + for (i = 0; i < (sizeof(table)/sizeof(table[0])); i++) { + if (table[i].code == sts) + return table[i].msg; + } + return unknown; +} |