2 * Authors.....: Jens Steube <jens.steube@gmail.com>
3 * Gabriele Gristina <matrix@hashcat.net>
4 * magnum <john.magnum@hushmail.com>
12 #if defined(HAVE_HWMON) && defined(HAVE_NVML)
17 * Declarations from nvml.h
20 typedef struct nvmlDevice_st
* nvmlDevice_t
;
22 typedef struct nvmlPciInfo_st
{
27 unsigned int pciDeviceId
;
28 unsigned int pciSubSystemId
;
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
36 typedef enum nvmlTemperatureSensors_enum
{
37 NVML_TEMPERATURE_GPU
= 0 // Temperature sensor for the GPU die
38 } nvmlTemperatureSensors_t
;
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
55 typedef enum nvmlClockType_enum
{
56 NVML_CLOCK_GRAPHICS
= 0,
62 * End of declarations from nvml.h
65 typedef nvmlDevice_t HM_ADAPTER_NV
;
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 *);
84 NVML_ERROR_STRING nvmlErrorString
;
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
;
97 #define NVML_PTR hm_nvml_lib_t
99 int nvml_init (NVML_PTR
*lib
);
100 void nvml_close (NVML_PTR
*lib
);
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
);
113 #endif // HAVE_HWMON && HAVE_NVML