Add Blackwell to DeviceArchitecture enum (#70)

Adds Blackwell to DeviceArchitecture enum

Co-authored-by: Marco Morais <[email protected]>

nvml-wrapper-sys/nvml.h

@@ -1439,6 +1439,8 @@ typedef struct nvmlGridLicensableFeatures_st
 
 #define NVML_DEVICE_ARCH_HOPPER    9 // Devices based on the NVIDIA Hopper architecture
 
+#define NVML_DEVICE_ARCH_BLACKWELL 10 // Devices based on the NVIDIA Blackwell architecture
+
 #define NVML_DEVICE_ARCH_UNKNOWN   0xffffffff // Anything else, presumably something newer
 
 typedef unsigned int nvmlDeviceArchitecture_t;

nvml-wrapper-sys/src/bindings.rs

@@ -58,6 +58,7 @@ pub const NVML_DEVICE_ARCH_TURING: u32 = 6;
 pub const NVML_DEVICE_ARCH_AMPERE: u32 = 7;
 pub const NVML_DEVICE_ARCH_ADA: u32 = 8;
 pub const NVML_DEVICE_ARCH_HOPPER: u32 = 9;
+pub const NVML_DEVICE_ARCH_BLACKWELL: u32 = 10;
 pub const NVML_DEVICE_ARCH_UNKNOWN: u32 = 4294967295;
 pub const NVML_BUS_TYPE_UNKNOWN: u32 = 0;
 pub const NVML_BUS_TYPE_PCI: u32 = 1;

nvml-wrapper/src/enums/device.rs

@@ -225,6 +225,8 @@ pub enum DeviceArchitecture {
     Ada,
     /// <https://en.wikipedia.org/wiki/Hopper_(microarchitecture)>
     Hopper,
+    /// https://en.wikipedia.org/wiki/Blackwell_(microarchitecture)
+    Blackwell,
     /// Unknown device architecture (most likely something newer).
     Unknown,
 }
@@ -241,6 +243,7 @@ impl DeviceArchitecture {
             Self::Ampere => NVML_DEVICE_ARCH_AMPERE,
             Self::Ada => NVML_DEVICE_ARCH_ADA,
             Self::Hopper => NVML_DEVICE_ARCH_HOPPER,
+            Self::Blackwell => NVML_DEVICE_ARCH_BLACKWELL,
             Self::Unknown => NVML_DEVICE_ARCH_UNKNOWN,
         }
     }
@@ -259,6 +262,7 @@ impl TryFrom<nvmlDeviceArchitecture_t> for DeviceArchitecture {
             NVML_DEVICE_ARCH_AMPERE => Ok(Self::Ampere),
             NVML_DEVICE_ARCH_ADA => Ok(Self::Ada),
             NVML_DEVICE_ARCH_HOPPER => Ok(Self::Hopper),
+            NVML_DEVICE_ARCH_BLACKWELL => Ok(Self::Blackwell),
             NVML_DEVICE_ARCH_UNKNOWN => Ok(Self::Unknown),
             _ => Err(NvmlError::UnexpectedVariant(data)),
         }
@@ -276,6 +280,7 @@ impl Display for DeviceArchitecture {
             Self::Ampere => f.write_str("Ampere"),
             Self::Ada => f.write_str("Ada"),
             Self::Hopper => f.write_str("Hopper"),
+            Self::Blackwell => f.write_str("Blackwell"),
             Self::Unknown => f.write_str("Unknown"),
         }
     }