adding device::performance_modes() (#88)

Adds device::performance_modes()

Author: devnexen

nvml-wrapper/src/device.rs

@@ -5793,6 +5793,44 @@ impl<'nvml> Device<'nvml> {
         }
     }
 
+    /**
+    Get the list of performance modes for `Device`
+
+    Originally, NVML returns as a list in a form of a single string separated by comma.
+
+    # Errors
+
+    * `Uninitialized`, if the library has not been successfully initialized
+    * `NotSupported`, if the platform does not support this feature or some of the
+    requested event types.
+    * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
+    * `Unknown`, on any unexpected error. **If this error is returned, the `set` you
+
+    # Platform Support
+
+    Only supports Linux.
+    */
+    // Checked against local
+    // Tested
+    #[cfg(target_os = "linux")]
+    #[doc(alias = "nvmlDeviceGetPerformanceModes")]
+    pub fn performance_modes(&self) -> Result<(Vec<String>, u32), NvmlError> {
+        let sym = nvml_sym(self.nvml.lib.nvmlDeviceGetPerformanceModes.as_ref())?;
+
+        unsafe {
+            let mut pmodes: nvmlDevicePerfModes_t = mem::zeroed();
+
+            nvml_try(sym(self.device, &mut pmodes))?;
+
+            let modes_str = CStr::from_ptr(pmodes.str_.as_ptr());
+            let modes = modes_str.to_str()?;
+            Ok((
+                modes.split(';').map(str::to_string).collect(),
+                pmodes.version,
+            ))
+        }
+    }
+
     /**
     Gets the active vGPU instances for `Device`
 
@@ -7112,6 +7150,13 @@ mod test {
         test_with_device(3, &nvml, |device| device.is_drain_enabled(None))
     }
 
+    #[cfg(target_os = "linux")]
+    #[test]
+    fn performance_modes() {
+        let nvml = nvml();
+        test_with_device(3, &nvml, |device| device.performance_modes())
+    }
+
     #[cfg(target_os = "linux")]
     #[test]
     fn active_vgpus() {

nvml-wrapper/src/test_utils.rs

@@ -106,6 +106,7 @@ impl ShouldPrint for PowerSource {}
 impl ShouldPrint for DeviceArchitecture {}
 impl ShouldPrint for PcieLinkMaxSpeed {}
 impl ShouldPrint for DeviceAttributes {}
+impl ShouldPrint for (Vec<String>, u32) {}
 impl ShouldPrint for GpuVirtualizationMode {}
 impl ShouldPrint for ClockOffset {}
 impl ShouldPrint for MigMode {}