adding device::virtualization_mode() wrapper.

Returns whether the current GPU device is within a virtualization mode
or bare metal.

Author: devnexen

nvml-wrapper/src/device.rs

@@ -5314,6 +5314,37 @@ impl<'nvml> Device<'nvml> {
         }
     }
 
+    /**
+    Gets the virtualization mode of `Device`
+
+    # Errors
+
+    * `Uninitialized`, if the library has not been successfully initialized
+    * `InvalidArg`, if this `Device` is invalid or `clock_type` is invalid (shouldn't occur?)
+    * `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
+
+    # Device support
+
+    Supports Kepler and newer fully supported devices.
+    */
+    // Checked against local
+    // Tested
+    #[cfg(target_os = "linux")]
+    #[doc(alias = "nvmlDeviceGetVirtualizationMode")]
+    pub fn virtualization_mode(&self) -> Result<GpuVirtualizationMode, NvmlError> {
+        let sym = nvml_sym(self.nvml.lib.nvmlDeviceGetVirtualizationMode.as_ref())?;
+
+        unsafe {
+            let mut mode: nvmlGpuVirtualizationMode_t = mem::zeroed();
+
+            nvml_try(sym(self.device, &mut mode))?;
+
+            GpuVirtualizationMode::try_from(mode)
+        }
+    }
+
     /**
     Removes this `Device` from the view of both NVML and the NVIDIA kernel driver.
 
@@ -6514,6 +6545,13 @@ mod test {
         test_with_device(3, &nvml, |device| device.is_drain_enabled(None))
     }
 
+    #[cfg(target_os = "linux")]
+    #[test]
+    fn virtualization_mode() {
+        let nvml = nvml();
+        test_with_device(3, &nvml, |device| device.virtualization_mode())
+    }
+
     #[cfg(target_os = "linux")]
     #[test]
     fn device_attributes() {

nvml-wrapper/src/enum_wrappers/device.rs

@@ -629,3 +629,24 @@ pub enum ClockLimitId {
     #[wrap(c_variant = "NVML_CLOCK_LIMIT_ID_UNLIMITED")]
     Unlimited,
 }
+
+#[derive(EnumWrapper, Debug, Clone, Eq, PartialEq, Hash)]
+#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
+#[wrap(c_enum = "nvmlGpuVirtualizationMode")]
+pub enum GpuVirtualizationMode {
+    /// Bare metal mode
+    #[wrap(c_variant = "NVML_GPU_VIRTUALIZATION_MODE_NONE")]
+    Bare,
+    /// Passthrough mode
+    #[wrap(c_variant = "NVML_GPU_VIRTUALIZATION_MODE_PASSTHROUGH")]
+    PassThrough,
+    /// vGPU inside virtual machine mode
+    #[wrap(c_variant = "NVML_GPU_VIRTUALIZATION_MODE_VGPU")]
+    Vgpu,
+    /// VGX hypervisor in vGPU mode
+    #[wrap(c_variant = "NVML_GPU_VIRTUALIZATION_MODE_HOST_VGPU")]
+    HostVgpu,
+    /// VGX hypervisor in vSGA mode
+    #[wrap(c_variant = "NVML_GPU_VIRTUALIZATION_MODE_HOST_VSGA")]
+    HostVsga,
+}

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 GpuVirtualizationMode {}
 
 #[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