Fixes when compiling with both NVML and DCGM.

NVML temp code updated (driver version ~555 and up)
This commit is contained in:
2026-09-30 01:18:05 -07:00
parent f6efb12fdb
commit fdf90ef790
6 changed files with 22 additions and 25 deletions
+1
View File
@@ -3,6 +3,7 @@
#include <dirent.h> #include <dirent.h>
#include <stdint.h> #include <stdint.h>
#include <math.h> #include <math.h>
#include <stdbool.h>
#include "blackwell-sensors.h" #include "blackwell-sensors.h"
+6
View File
@@ -36,6 +36,12 @@ struct card_info {
void *nvml_device; /* Pointer to nvmlDevice_t for use in NVML calls */ void *nvml_device; /* Pointer to nvmlDevice_t for use in NVML calls */
}; };
struct clock_reason {
unsigned long long mask;
char *short_name;
char *long_name;
};
unsigned short read_pci_id(char *device_path, char *field); unsigned short read_pci_id(char *device_path, char *field);
char *read_nvidia_pci_address(char *device_path); char *read_nvidia_pci_address(char *device_path);
void mmap_bar0(struct card_info *card); void mmap_bar0(struct card_info *card);
+4 -4
View File
@@ -78,8 +78,8 @@ void print_dcgm(int compact, int overwrite)
else else
printf("Clock reasons: "); printf("Clock reasons: ");
for (int i = 0; i < (sizeof(clock_reason_names) / sizeof(struct clock_reason)); i++) { for (int i = 0; i < (sizeof(dcgm_clock_reason_names) / sizeof(struct clock_reason)); i++) {
if (clock_reasons & clock_reason_names[i].mask) { if (clock_reasons & dcgm_clock_reason_names[i].mask) {
if (!single_reason) { if (!single_reason) {
if (compact) if (compact)
printf(","); printf(",");
@@ -89,9 +89,9 @@ void print_dcgm(int compact, int overwrite)
single_reason = 0; single_reason = 0;
if (compact) if (compact)
printf("%s", clock_reason_names[i].short_name); printf("%s", dcgm_clock_reason_names[i].short_name);
else else
printf("%s", clock_reason_names[i].long_name); printf("%s", dcgm_clock_reason_names[i].long_name);
} }
} }
+1 -7
View File
@@ -26,13 +26,7 @@ static struct metric_reading metric_readings[] =
{DCGM_FI_PROF_DRAM_ACTIVE, "DRAM", "Memory BW Utilization"} {DCGM_FI_PROF_DRAM_ACTIVE, "DRAM", "Memory BW Utilization"}
}; };
struct clock_reason { static struct clock_reason dcgm_clock_reason_names[] =
unsigned long long mask;
char *short_name;
char *long_name;
};
static struct clock_reason clock_reason_names[] =
{ {
{DCGM_CLOCKS_EVENT_REASON_GPU_IDLE, "Idle", "GPU idle"}, {DCGM_CLOCKS_EVENT_REASON_GPU_IDLE, "Idle", "GPU idle"},
{DCGM_CLOCKS_EVENT_REASON_CLOCKS_SETTING, "AppClk", "Application clocks"}, {DCGM_CLOCKS_EVENT_REASON_CLOCKS_SETTING, "AppClk", "Application clocks"},
+9 -7
View File
@@ -25,8 +25,8 @@ void print_nvml_clock_reason(int compact, struct card_info *card)
unsigned long long reasons = get_nvml_clock_reasons(card); unsigned long long reasons = get_nvml_clock_reasons(card);
int single_reason = 1; int single_reason = 1;
for (int i = 0; i < (sizeof(clock_reason_names) / sizeof(struct clock_reason)); i++) { for (int i = 0; i < (sizeof(nvml_clock_reason_names) / sizeof(struct clock_reason)); i++) {
if (reasons & clock_reason_names[i].mask) { if (reasons & nvml_clock_reason_names[i].mask) {
if (!single_reason) { if (!single_reason) {
if (compact) if (compact)
printf(","); printf(",");
@@ -36,9 +36,9 @@ void print_nvml_clock_reason(int compact, struct card_info *card)
single_reason = 0; single_reason = 0;
if (compact) if (compact)
printf("%s", clock_reason_names[i].short_name); printf("%s", nvml_clock_reason_names[i].short_name);
else else
printf("%s", clock_reason_names[i].long_name); printf("%s", nvml_clock_reason_names[i].long_name);
} }
} }
@@ -56,13 +56,15 @@ unsigned int get_nvml_temp(struct card_info *card)
if (card->nvml_device == NULL) if (card->nvml_device == NULL)
return 0; return 0;
unsigned int temp; nvmlTemperature_t temp;
nvmlReturn_t result = nvmlDeviceGetTemperature(*(nvmlDevice_t*)(card->nvml_device), NVML_TEMPERATURE_GPU, &temp); temp.version = nvmlTemperature_v1;
temp.sensorType = NVML_TEMPERATURE_GPU;
nvmlReturn_t result = nvmlDeviceGetTemperatureV(*(nvmlDevice_t*)(card->nvml_device), &temp);
if (result != NVML_SUCCESS) { if (result != NVML_SUCCESS) {
printf("Failed to get temperature for card at %s: %s\n", card->pci_id, nvmlErrorString(result)); printf("Failed to get temperature for card at %s: %s\n", card->pci_id, nvmlErrorString(result));
return 0; return 0;
} }
return temp; return temp.temperature;
} }
unsigned long long get_nvml_clock_reasons(struct card_info *card) unsigned long long get_nvml_clock_reasons(struct card_info *card)
+1 -7
View File
@@ -2,13 +2,7 @@
#include "evga-card.h" #include "evga-card.h"
struct clock_reason { static struct clock_reason nvml_clock_reason_names[] =
unsigned long long mask;
char *short_name;
char *long_name;
};
static struct clock_reason clock_reason_names[] =
{ {
{nvmlClocksEventReasonGpuIdle, "Idle", "GPU idle"}, {nvmlClocksEventReasonGpuIdle, "Idle", "GPU idle"},
{nvmlClocksEventReasonApplicationsClocksSetting, "AppClk", "Application clocks"}, {nvmlClocksEventReasonApplicationsClocksSetting, "AppClk", "Application clocks"},