Adding device::mig_mode().

nvml-wrapper/src/device.rs

@@ -2378,6 +2378,32 @@
         }
     }
 
+    /**
+    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 mig_mode(&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".
 
@@ -3242,7 +3268,7 @@
 
     * `Uninitialized`, if the library has not been successfully initialized
     * `IncorrectBits`, if NVML returns any bits that do not correspond to flags in
     `ThrottleReasons`
     * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
     * `Unknown`, on any unexpected error
 
@@ -4289,7 +4315,7 @@
     * `Uninitialized`, if the library has not been successfully initialized
     * `InvalidArg`, if the `Device` is invalid
     * `NotSupported`, if this `Device` does not support this feature or accounting mode
     is disabled
     * `Unknown`, on any unexpected error
 
     # Device Support
@@ -4351,7 +4377,7 @@
     * `Uninitialized`, if the library has not been successfully initialized
     * `InvalidArg`, if the `Device` is invalid
     * `NotSupported`, if this `Device` does not support this feature or accounting
     mode is disabled
     * `Unknown`, on any unexpected error
     */
     // Checked against local
@@ -4408,9 +4434,9 @@
     Note:
     * Accounting mode needs to be on. See `.is_accounting_enabled()`.
     * Only compute and graphics applications stats can be queried. Monitoring
     applications can't be queried since they don't contribute to GPU utilization.
     * If a PID collision occurs, the stats of the latest process (the one that
     terminated last) will be reported.
 
     # Errors
 
@@ -4418,7 +4444,7 @@
     * `InvalidArg`, if the `Device` is invalid
     * `NotFound`, if the process stats were not found
     * `NotSupported`, if this `Device` does not support this feature or accounting
     mode is disabled
     * `Unknown`, on any unexpected error
 
     # Device Support
@@ -4452,8 +4478,8 @@
 
     Note:
     * This setting is not persistent and will default to disabled after the driver
     unloads. Enable persistence mode to be sure the setting doesn't switch off
     to disabled.
     * Enabling accounting mode has no negative impact on GPU performance.
     * Disabling accounting clears accounting information for all PIDs
 
@@ -4526,8 +4552,8 @@
     * `Uninitialized`, if the library has not been successfully initialized
     * `InvalidArg`, if the `Device` is invalid or `api_type` is invalid (shouldn't occur?)
     * `NotSupported`, if this `Device` does not support changing API restrictions or
     this `Device` does not support the feature that API restrictions are being set for
     (e.g. enabling/disabling auto boosted clocks is not supported by this `Device`).
     * `NoPermission`, if the user doesn't have permission to perform this operation
     * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
     * `Unknown`, on any unexpected error
@@ -5882,6 +5908,12 @@
         test_with_device(3, &nvml, |device| device.is_multi_gpu_board())
     }
 
+    #[test]
+    fn mig_mode() {
+        let nvml = nvml();
+        test_with_device(3, &nvml, |device| device.mig_mode())
+    }
+
     #[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