adding device::gsp_firmware_mode/gsp_firmware_version() (#113)

Author: devnexen

nvml-wrapper/src/device.rs

@@ -6455,6 +6455,57 @@ impl<'nvml> Device<'nvml> {
         }
     }
 
+    /**
+     Get GSP firmware mode. Whether it is enabled and if it is in default mode.
+
+    # Errors
+
+    * `Uninitialized`, if the library has not been successfully initialized
+    * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
+    * `Unknown`, on any unexpected error
+    * `NotSupported`, if the platform does not support this feature
+    */
+    #[doc(alias = "nvmlDeviceGetGspFirmwareMode")]
+    pub fn gsp_firmware_mode(&self) -> Result<GspFirmwareMode, NvmlError> {
+        let sym = nvml_sym(self.nvml.lib.nvmlDeviceGetGspFirmwareMode.as_ref())?;
+
+        unsafe {
+            let mut enabled: c_uint = 0;
+            let mut default: c_uint = 0;
+
+            nvml_try(sym(self.device, &mut enabled, &mut default))?;
+
+            Ok(GspFirmwareMode {
+                enabled: enabled != 0,
+                default: default != 0,
+            })
+        }
+    }
+
+    /**
+     Get GSP firmware version.
+
+    # Errors
+
+    * `Uninitialized`, if the library has not been successfully initialized
+    * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
+    * `Unknown`, on any unexpected error
+    * `NotSupported`, if the platform does not support this feature
+    */
+    #[doc(alias = "nvmlDeviceGetGspFirmwareVersion")]
+    pub fn gsp_firmware_version(&self) -> Result<String, NvmlError> {
+        let sym = nvml_sym(self.nvml.lib.nvmlDeviceGetGspFirmwareVersion.as_ref())?;
+
+        unsafe {
+            let mut version = vec![0; 80];
+
+            nvml_try(sym(self.device, version.as_mut_ptr()))?;
+            let raw = CStr::from_ptr(version.as_ptr());
+
+            Ok(raw.to_str()?.into())
+        }
+    }
+
     // NvLink
 
     /**
@@ -7213,6 +7264,18 @@ mod test {
         })
     }
 
+    #[test]
+    fn gsp_firmware_mode() {
+        let nvml = nvml();
+        test_with_device(3, &nvml, |device| device.gsp_firmware_mode())
+    }
+
+    #[test]
+    fn gsp_firmware_version() {
+        let nvml = nvml();
+        test_with_device(3, &nvml, |device| device.gsp_firmware_version())
+    }
+
     #[test]
     fn field_values_for() {
         let nvml = nvml();

nvml-wrapper/src/structs/device.rs

@@ -177,3 +177,11 @@ pub struct MigMode {
     /// Mode set after reboot.
     pub pending: u32,
 }
+
+/// Returned from `Device.gsp_firmware_mode()`
+#[derive(Debug, Clone, Eq, PartialEq, Hash)]
+#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
+pub struct GspFirmwareMode {
+    pub enabled: bool,
+    pub default: bool,
+}

nvml-wrapper/src/test_utils.rs

@@ -113,6 +113,7 @@ impl ShouldPrint for MigMode {}
 impl ShouldPrint for Vec<GpuInstancePlacement> {}
 impl ShouldPrint for (VgpuVersion, VgpuVersion) {}
 impl ShouldPrint for ProfileInfo {}
+impl ShouldPrint for GspFirmwareMode {}
 
 #[cfg(target_os = "windows")]
 impl ShouldPrint for DriverModelState {}