include/ext_nvml.h

   1 /**
   2  * Authors.....: Jens Steube <jens.steube@gmail.com>
   3  *               Gabriele Gristina <matrix@hashcat.net>
   4  *               magnum <john.magnum@hushmail.com>
   5  *
   6  * License.....: MIT
   7  */
   8
   9 #ifndef EXT_NVML_H
  10 #define EXT_NVML_H
  11
  12 #if defined(HAVE_HWMON) && defined(HAVE_NVML)
  13
  14 #include <common.h>
  15
  16 /**
  17  * Declarations from nvml.h
  18  */
  19
  20 typedef struct nvmlDevice_st* nvmlDevice_t;
  21
  22 typedef struct nvmlPciInfo_st {
  23   char busId[16];
  24   unsigned int domain;
  25   unsigned int bus;
  26   unsigned int device;
  27   unsigned int pciDeviceId;
  28   unsigned int pciSubSystemId;
  29 } nvmlPciInfo_t;
  30
  31 typedef struct nvmlUtilization_st {
  32   unsigned int gpu;    // GPU kernel execution last second, percent
  33   unsigned int memory; // GPU memory read/write last second, percent
  34 } nvmlUtilization_t;
  35
  36 typedef enum nvmlTemperatureSensors_enum {
  37   NVML_TEMPERATURE_GPU = 0     // Temperature sensor for the GPU die
  38 } nvmlTemperatureSensors_t;
  39
  40 typedef enum nvmlReturn_enum {
  41   NVML_SUCCESS = 0,                   // The operation was successful
  42   NVML_ERROR_UNINITIALIZED = 1,       // NVML was not first initialized with nvmlInit()
  43   NVML_ERROR_INVALID_ARGUMENT = 2,    // A supplied argument is invalid
  44   NVML_ERROR_NOT_SUPPORTED = 3,       // The requested operation is not available on target device
  45   NVML_ERROR_NO_PERMISSION = 4,       // The current user does not have permission for operation
  46   NVML_ERROR_ALREADY_INITIALIZED = 5, // Deprecated: Multiple initializations are now allowed through ref counting
  47   NVML_ERROR_NOT_FOUND = 6,           // A query to find an object was unsuccessful
  48   NVML_ERROR_INSUFFICIENT_SIZE = 7,   // An input argument is not large enough
  49   NVML_ERROR_INSUFFICIENT_POWER = 8,  // A device's external power cables are not properly attached
  50   NVML_ERROR_DRIVER_NOT_LOADED = 9,   // NVIDIA driver is not loaded
  51   NVML_ERROR_TIMEOUT = 10,            // User provided timeout passed
  52   NVML_ERROR_UNKNOWN = 999            // An internal driver error occurred
  53 } nvmlReturn_t;
  54
  55 typedef enum nvmlClockType_enum {
  56         NVML_CLOCK_GRAPHICS = 0,
  57         NVML_CLOCK_SM = 1,
  58         NVML_CLOCK_MEM = 2
  59 } nvmlClockType_t;
  60
  61 /*
  62  * End of declarations from nvml.h
  63  **/
  64
  65 typedef nvmlDevice_t HM_ADAPTER_NV;
  66
  67 #include <shared.h>
  68
  69 typedef const char * (*NVML_ERROR_STRING) (nvmlReturn_t);
  70 typedef int (*NVML_INIT) (void);
  71 typedef int (*NVML_SHUTDOWN) (void);
  72 typedef nvmlReturn_t (*NVML_DEVICE_GET_NAME) (nvmlDevice_t, char *, unsigned int);
  73 typedef nvmlReturn_t (*NVML_DEVICE_GET_HANDLE_BY_INDEX) (unsigned int, nvmlDevice_t *);
  74 typedef nvmlReturn_t (*NVML_DEVICE_GET_TEMPERATURE) (nvmlDevice_t, nvmlTemperatureSensors_t, unsigned int *);
  75 typedef nvmlReturn_t (*NVML_DEVICE_GET_FAN_SPEED) (nvmlDevice_t, unsigned int *);
  76 typedef nvmlReturn_t (*NVML_DEVICE_GET_POWER_USAGE) (nvmlDevice_t, unsigned int *);
  77 typedef nvmlReturn_t (*NVML_DEVICE_GET_UTILIZATION_RATES) (nvmlDevice_t, nvmlUtilization_t *);
  78 typedef nvmlReturn_t (*NVML_DEVICE_GET_CLOCKINFO) (nvmlDevice_t, nvmlClockType_t, unsigned int *);
  79
  80 typedef struct
  81 {
  82   NV_LIB lib;
  83
  84   NVML_ERROR_STRING nvmlErrorString;
  85   NVML_INIT nvmlInit;
  86   NVML_SHUTDOWN nvmlShutdown;
  87   NVML_DEVICE_GET_NAME nvmlDeviceGetName;
  88   NVML_DEVICE_GET_HANDLE_BY_INDEX nvmlDeviceGetHandleByIndex;
  89   NVML_DEVICE_GET_TEMPERATURE nvmlDeviceGetTemperature;
  90   NVML_DEVICE_GET_FAN_SPEED nvmlDeviceGetFanSpeed;
  91   NVML_DEVICE_GET_POWER_USAGE nvmlDeviceGetPowerUsage;
  92   NVML_DEVICE_GET_UTILIZATION_RATES nvmlDeviceGetUtilizationRates;
  93   NVML_DEVICE_GET_CLOCKINFO nvmlDeviceGetClockInfo;
  94
  95 } hm_nvml_lib_t;
  96
  97 #define NVML_PTR hm_nvml_lib_t
  98
  99 int nvml_init (NVML_PTR *lib);
 100 void nvml_close (NVML_PTR *lib);
 101
 102 const char * hm_NVML_nvmlErrorString (NVML_PTR *nvml, nvmlReturn_t nvml_rc);
 103 nvmlReturn_t hm_NVML_nvmlInit (NVML_PTR *nvml);
 104 nvmlReturn_t hm_NVML_nvmlShutdown (NVML_PTR *nvml);
 105 nvmlReturn_t hm_NVML_nvmlDeviceGetName (NVML_PTR *nvml, nvmlDevice_t device, char *name, unsigned int length);
 106 nvmlReturn_t hm_NVML_nvmlDeviceGetHandleByIndex (NVML_PTR *nvml, int, unsigned int index, nvmlDevice_t *device);
 107 nvmlReturn_t hm_NVML_nvmlDeviceGetTemperature (NVML_PTR *nvml, nvmlDevice_t device, nvmlTemperatureSensors_t sensorType, unsigned int *temp);
 108 nvmlReturn_t hm_NVML_nvmlDeviceGetFanSpeed (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned int *speed);
 109 nvmlReturn_t hm_NVML_nvmlDeviceGetPowerUsage (NVML_PTR *nvml, nvmlDevice_t device, unsigned int *power);
 110 nvmlReturn_t hm_NVML_nvmlDeviceGetUtilizationRates (NVML_PTR *nvml, nvmlDevice_t device, nvmlUtilization_t *utilization);
 111 nvmlReturn_t hm_NVML_nvmlDeviceGetClockInfo (NVML_PTR *nvml, nvmlDevice_t device, nvmlClockType_t type, unsigned int *clock);
 112
 113 #endif // HAVE_HWMON && HAVE_NVML
 114
 115 #endif // EXT_NVML_H