--versions: Add support for dumping NVIDIA graphics info

Trying it out with thunderbolt eGPUs

Signed-off-by: Daniel Schaefer <dhs@frame.work>

Author: JohnAZoidberg

Cargo.lock

@@ -52,6 +52,7 @@ clap-num = { version = "1.2.0" }
 clap-verbosity-flag = { version = "2.2.1" }
 windows-version = "0.1.4"
 winreg = "0.55.0"
+nvml-wrapper = "0.11.0"
 
 [target.'cfg(unix)'.dependencies]
 libc = "0.2.155"
@@ -62,6 +63,7 @@ env_logger = "0.11"
 clap = { version = "4.5", features = ["derive", "cargo"] }
 clap-num = { version = "1.2.0" }
 clap-verbosity-flag = { version = "2.2.1" }
+nvml-wrapper = "0.11.0"
 
 [target.'cfg(windows)'.dependencies.windows]
 version = "0.59.0"

framework_lib/Cargo.toml

@@ -72,6 +72,9 @@ use sha2::{Digest, Sha256, Sha384, Sha512};
 //use smbioslib::*;
 use smbioslib::{DefinedStruct, SMBiosInformation};
 
+#[cfg(any(target_os = "linux", target_os = "windows"))]
+use nvml_wrapper::{enum_wrappers::device::TemperatureSensor, Nvml};
+
 use crate::chromium_ec::{CrosEc, CrosEcDriverType, HardwareDeviceType};
 
 #[cfg(feature = "uefi")]
@@ -363,6 +366,8 @@ fn print_pd_details(ec: &CrosEc) {
 #[cfg(feature = "hidapi")]
 const NOT_SET: &str = "NOT SET";
 
+const UNKNOWN: &str = "Unknown";
+
 #[cfg(feature = "rusb")]
 fn print_audio_card_details() {
     check_synaptics_fw_version();
@@ -767,6 +772,88 @@ fn print_versions(ec: &CrosEc) {
             }
         }
     }
+
+    #[cfg(any(target_os = "linux", target_os = "windows"))]
+    print_nvidia_details();
+}
+
+#[cfg(any(target_os = "linux", target_os = "windows"))]
+fn print_nvidia_details() {
+    let nvml = match Nvml::init() {
+        Ok(nvml) => nvml,
+        Err(err) => {
+            debug!("Nvidia, library init fail: {:?}", err);
+            return;
+        }
+    };
+    // Get the first `Device` (GPU) in the system
+    let device = match nvml.device_by_index(0) {
+        Ok(device) => device,
+        Err(err) => {
+            debug!("Nvidia, device not found: {:?}", err);
+            return;
+        }
+    };
+
+    println!("NVIDIA GPU");
+    // GeForce
+    info!("  BRAND:          {:?}", device.brand());
+    println!(
+        "  Name:             {}",
+        device.name().unwrap_or(UNKNOWN.to_string())
+    );
+    println!("  Architecture:     {:?}", device.architecture());
+    println!(
+        "  VBIOS Version:    {}",
+        device.vbios_version().unwrap_or(UNKNOWN.to_string())
+    );
+    println!(
+        "  INFO ROM Ver:     {}",
+        device
+            .info_rom_image_version()
+            .unwrap_or(UNKNOWN.to_string())
+    );
+    println!("  PCI Info:         {:X?}", device.pci_info());
+    println!("  Performance State:{:?}", device.performance_state());
+    println!(
+        "  Pwr Mgmt Limit Df:{:?}mW",
+        device.power_management_limit_default()
+    );
+    // NotSupported
+    println!(
+        "  Pwr Mgmt Limit:   {:?}mW",
+        device.power_management_limit()
+    );
+    println!(
+        "  Pwr Mgmt Limit Cs:{:?}",
+        device.power_management_limit_constraints()
+    );
+    println!("  Pwr Usage:        {:?}mW", device.power_usage());
+    println!(
+        "  Total Energy:     {:?}mJ",
+        device.total_energy_consumption()
+    );
+    // 0/NotSupported right now
+    println!("  Serialnum:        {:?}", device.serial());
+    println!(
+        "  Throttle Reason:  {:?}",
+        device.current_throttle_reasons()
+    );
+    println!(
+        "  Temperature:      {:?}C",
+        device.temperature(TemperatureSensor::Gpu)
+    );
+    //println!("  Temperature Thres:{:?}C", device.temperature_threshold());
+    println!("  Util Rate:        {:?}", device.utilization_rates());
+    println!("  Memory Info:      {:?}", device.memory_info());
+    // Not supported
+    println!("  Part Number:      {:?}", device.board_part_number());
+    println!("  Board ID:         {:?}", device.board_id());
+    // 0
+    println!("  Num Fans:         {:?}", device.num_fans());
+    // Works
+    println!("  Display Active?:  {:?}", device.is_display_active());
+    println!("  Display Conn?:    {:?}", device.is_display_connected());
 }
 
 fn print_esrt() {