Adding device::multi_gpu_instances().

To check if the device supports the MultiInstancesGpu feature and its
state.

Author: devnexen

nvml-wrapper/src/device.rs

@@ -2378,6 +2378,32 @@ impl<'nvml> Device<'nvml> {
         }
     }
 
+    /**
+    Checks if the `Device`supports multi partitioned GPU feature and if enabled.
+    Not to confuse with `is_multi_gpu_board`, MIG is a single GPU
+    being able to be split into isolated instances, a sort of "NUMA" for GPU.
+    If the `Device` supports MIG, we can have its current mode (enabled/disabled)
+    and, if set, its pending mode for the next system reboot.
+    # Errors
+
+    * `Uninitialized`, if the library has not been successfully initialized
+    * `InvalidArg`, if this `Device` is invalid
+    * `NotSupported`, if this `Device` does not support this feature
+    * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
+    * `Unknown`, on any unexpected error
+    */
+    #[doc(alias = "nvmlDeviceGetMigMode")]
+    pub fn multi_gpu_instances(&self) -> Result<MigMode, NvmlError> {
+        let sym = nvml_sym(self.nvml.lib.nvmlDeviceGetMigMode.as_ref())?;
+
+        unsafe {
+            let mut mode: MigMode = mem::zeroed();
+            nvml_try(sym(self.device, &mut mode.current, &mut mode.pending))?;
+
+            Ok(mode)
+        }
+    }
+
     /**
     The name of this `Device`, e.g. "Tesla C2070".
 
@@ -5882,6 +5908,12 @@ mod test {
         test_with_device(3, &nvml, |device| device.is_multi_gpu_board())
     }
 
+    #[test]
+    fn multi_gpu_instances() {
+        let nvml = nvml();
+        test_with_device(3, &nvml, |device| device.multi_gpu_instances())
+    }
+
     #[test]
     fn name() {
         let nvml = nvml();

nvml-wrapper/src/structs/device.rs

@@ -135,3 +135,13 @@ pub struct RetiredPage {
 #[derive(Debug, Clone, Eq, PartialEq, Hash)]
 #[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
 pub struct FieldId(pub u32);
+
+/// Returned from `Device.mig_mode()`
+#[derive(Debug, Clone, Eq, PartialEq, Hash)]
+#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
+pub struct MigMode {
+    /// Whether MIG mode is enabled/disabled.
+    pub current: u32,
+    /// Mode set after reboot.
+    pub pending: u32,
+}

nvml-wrapper/src/test_utils.rs

@@ -105,6 +105,7 @@ impl ShouldPrint for PowerSource {}
 impl ShouldPrint for DeviceArchitecture {}
 impl ShouldPrint for PcieLinkMaxSpeed {}
 impl ShouldPrint for DeviceAttributes {}
+impl ShouldPrint for MigMode {}
 
 #[cfg(target_os = "windows")]
 impl ShouldPrint for DriverModelState {}

nvml-wrapper/unwrapped_functions.txt

@@ -12,11 +12,9 @@ nvmlDeviceGetCapabilities
 nvmlDeviceGetClkMonStatus
 nvmlDeviceGetClockOffsets
 nvmlDeviceGetComputeInstanceId
-nvmlDeviceGetConfComputeGpuCertificate
 nvmlDeviceGetConfComputeMemSizeInfo
 nvmlDeviceGetConfComputeProtectedMemoryUsage
 nvmlDeviceGetCoolerInfo
-nvmlDeviceGetCpuAffinityWithinScope
 nvmlDeviceGetCreatableVgpus
 nvmlDeviceGetCurrentClockFreqs
 nvmlDeviceGetCurrentClocksEventReasons