Add DeviceAttributes and device::attributes().

nvml-wrapper/src/device.rs

@@ -73,8 +73,8 @@
     nvml: &'nvml Nvml,
 }
 
 unsafe impl<'nvml> Send for Device<'nvml> {}
 unsafe impl<'nvml> Sync for Device<'nvml> {}
 
 assert_impl_all!(Device: Send, Sync);
 
@@ -155,10 +155,10 @@
 
     * `Uninitialized`, if the library has not been successfully initialized
     * `InvalidArg`, if this `Device` is invalid or the apiType is invalid (may occur if
     the C lib changes dramatically?)
     * `NotSupported`, if this query is not supported by this `Device` or this `Device`
     does not support the feature that is being queried (e.g. enabling/disabling auto
     boosted clocks is not supported by this `Device`).
     * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
     * `UnexpectedVariant`, for which you can read the docs for
     * `Unknown`, on any unexpected error
@@ -192,7 +192,7 @@
 
     * `Uninitialized`, if the library has not been successfully initialized
     * `InvalidArg`, if this `Device` is invalid or the clockType is invalid (may occur
     if the C lib changes dramatically?)
     * `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
@@ -432,7 +432,7 @@
     * `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` or the `clock_type` on 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
 
@@ -926,6 +926,31 @@
         }
     }
 
+    /**
+    Gets GPU device hardware attributes
+
+    DeviceAttributes represents compute capabilities, Streaming MultiProcessor
+    capacity, slices allocated to a given GPU, decoding/encoding supported, 
+    available memory for these GPU operations
+
+    # Errors
+    * `Uninitialized`, if the library has not been successfully initialized
+    * `InvalidArg`, if this `Device` is invalid
+    * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
+    * `Unknown`, on any unexpected error
+    */
+    #[doc(alias = "nvmlDeviceGetAttributes_v2")]
+    pub fn attributes(&self) -> Result<DeviceAttributes, NvmlError> {
+        let sym = nvml_sym(self.nvml.lib.nvmlDeviceGetAttributes_v2.as_ref())?;
+
+        unsafe {
+            let mut attrs: nvmlDeviceAttributes_t = mem::zeroed();
+            nvml_try(sym(self.device, &mut attrs))?;
+
+            Ok(attrs.into())
+        }
+    }
+
     /**
     Gets the default applications clock that this `Device` boots with or defaults to after
     `reset_applications_clocks()`.
@@ -1236,7 +1261,7 @@
     * `Uninitialized`, if the library has not been successfully initialized
     * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
     * `UnexpectedVariant`, if an enum variant not defined in this wrapper gets
     returned in a field of an `EncoderSessionInfo` struct
     * `Unknown`, on any unexpected error
 
     # Device Support
@@ -1899,7 +1924,7 @@
     * `Uninitialized`, if the library has not been successfully initialized
     * `InvalidArg`, if `error_type`, `counter_type`, or `location` is invalid (shouldn't occur?)
     * `NotSupported`, if this `Device` does not support ECC error reporting for the specified
     memory
     * `GpuLost`, if this `Device` has fallen off the bus or is otherwise inaccessible
     * `Unknown`, on any unexpected error
 
@@ -2821,7 +2846,7 @@
 
     * `Uninitialized`, if the library has not been successfully initialized
     * `IncorrectBits`, if NVML returns any bits that do not correspond to flags in
     `ThrottleReasons`
     * `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
@@ -6075,4 +6100,11 @@
         let nvml = nvml();
         test_with_device(3, &nvml, |device| device.is_drain_enabled(None))
     }
+
+    #[cfg(target_os = "linux")]
+    #[test]
+    fn device_attributes() {
+        let nvml = nvml();
+        test_with_device(3, &nvml, |device| device.attributes())
+    }
 }

nvml-wrapper/src/struct_wrappers/device.rs

@@ -656,6 +656,46 @@ impl TryFrom<nvmlFBCSessionInfo_t> for FbcSessionInfo {
     }
 }
 
+/// Hardware level attributes from a GPU device
+#[derive(Debug, Clone, Eq, PartialEq, Hash)]
+#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
+pub struct DeviceAttributes {
+    /// Streaming MultiProcessor Count
+    pub multiprocessor_count: u32,
+    /// Shared Copy Engine Count
+    pub shared_copy_engine_count: u32,
+    /// Shared Decoder Count
+    pub shared_decoder_count: u32,
+    /// Shared Encoder Count
+    pub shared_encoder_count: u32,
+    /// Shared JPEG Count
+    pub shared_jpeg_count: u32,
+    /// Shared OFA Count
+    pub shared_ofa_count: u32,
+    /// GPU instance slice Count
+    pub gpu_instance_slice_count: u32,
+    /// Compute Instance slice count
+    pub compute_instance_slice_count: u32,
+    /// Device memory size in MB
+    pub memory_size_mb: u64,
+}
+
+impl From<nvmlDeviceAttributes_t> for DeviceAttributes {
+    fn from(struct_: nvmlDeviceAttributes_t) -> Self {
+        Self {
+            multiprocessor_count: struct_.multiprocessorCount,
+            shared_copy_engine_count: struct_.sharedCopyEngineCount,
+            shared_decoder_count: struct_.sharedDecoderCount,
+            shared_encoder_count: struct_.sharedEncoderCount,
+            shared_jpeg_count: struct_.sharedJpegCount,
+            shared_ofa_count: struct_.sharedOfaCount,
+            gpu_instance_slice_count: struct_.gpuInstanceSliceCount,
+            compute_instance_slice_count: struct_.computeInstanceSliceCount,
+            memory_size_mb: struct_.memorySizeMB,
+        }
+    }
+}
+
 #[cfg(test)]
 #[allow(unused_variables, unused_imports)]
 mod tests {

nvml-wrapper/src/test_utils.rs

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