Fix new enums

Author: mdboom

cuda_core/cuda/core/system/_clock.pxi

@@ -9,18 +9,6 @@ _CLOCK_ID_MAPPING = {
 }
 
 
-_CLOCK_EVENT_REASON_BOARD_LIMIT = getattr(
-    nvml.ClocksEventReasons,
-    "EVENT_REASON_BOARD_LIMIT",
-    0x0000000000000200,
-)
-_CLOCK_EVENT_REASON_RELIABILITY = getattr(
-    nvml.ClocksEventReasons,
-    "EVENT_REASON_RELIABILITY",
-    0x0000000000000400,
-)
-
-
 _CLOCKS_EVENT_REASONS_MAPPING = {
     nvml.ClocksEventReasons.EVENT_REASON_NONE: ClocksEventReasons.NONE,
     nvml.ClocksEventReasons.EVENT_REASON_GPU_IDLE: ClocksEventReasons.GPU_IDLE,
@@ -32,11 +20,16 @@ _CLOCKS_EVENT_REASONS_MAPPING = {
     nvml.ClocksEventReasons.THROTTLE_REASON_HW_THERMAL_SLOWDOWN: ClocksEventReasons.HW_THERMAL_SLOWDOWN,
     nvml.ClocksEventReasons.THROTTLE_REASON_HW_POWER_BRAKE_SLOWDOWN: ClocksEventReasons.HW_POWER_BRAKE_SLOWDOWN,
     nvml.ClocksEventReasons.EVENT_REASON_DISPLAY_CLOCK_SETTING: ClocksEventReasons.DISPLAY_CLOCK_SETTING,
-    _CLOCK_EVENT_REASON_BOARD_LIMIT: ClocksEventReasons.BOARD_LIMIT,
-    _CLOCK_EVENT_REASON_RELIABILITY: ClocksEventReasons.RELIABILITY,
 }
 
 
+if hasattr(nvml.ClocksEventReasons, "EVENT_REASON_BOARD_LIMIT"):
+    _CLOCKS_EVENT_REASONS_MAPPING.update({
+        nvml.ClocksEventReasons.EVENT_REASON_BOARD_LIMIT: ClocksEventReasons.BOARD_LIMIT,
+        nvml.ClocksEventReasons.EVENT_REASON_RELIABILITY: ClocksEventReasons.RELIABILITY,
+    })
+
+
 _CLOCK_TYPE_MAPPING = {
     ClockType.GRAPHICS: nvml.ClockType.CLOCK_GRAPHICS,
     ClockType.SM: nvml.ClockType.CLOCK_SM,

cuda_core/cuda/core/system/_device.pyx

@@ -103,10 +103,15 @@ _BRAND_TYPE_MAPPING = {
     nvml.BrandType.BRAND_NVIDIA: "NVIDIA",
     nvml.BrandType.BRAND_GEFORCE_RTX: "GeForce RTX",
     nvml.BrandType.BRAND_TITAN_RTX: "Titan RTX",
-    nvml.BrandType.BRAND_NVIDIA_DLA: "NVIDIA DLA",
-    nvml.BrandType.BRAND_NVIDIA_VGAMEDEV: "NVIDIA vGameDev",
-    nvml.BrandType.BRAND_NVIDIA_NPU: "NVIDIA NPU",
 }
+    
+
+if hasattr(nvml.BrandType, "BRAND_NVIDIA_DLA"):
+    _BRAND_TYPE_MAPPING.update({
+        nvml.BrandType.BRAND_NVIDIA_DLA: "NVIDIA DLA",
+        nvml.BrandType.BRAND_NVIDIA_VGAMEDEV: "NVIDIA vGameDev",
+        nvml.BrandType.BRAND_NVIDIA_NPU: "NVIDIA NPU",
+    })
 
 
 _GPU_P2P_CAPS_INDEX_MAPPING = {

cuda_core/tests/test_enum_coverage.py

@@ -132,15 +132,6 @@
 
     _MODULES.append(system_typing)
 
-    _CLOCKS_EVENT_REASONS_STR_UNMAPPED = {
-        core_member
-        for binding_member, core_member in (
-            ("EVENT_REASON_BOARD_LIMIT", "BOARD_LIMIT"),
-            ("EVENT_REASON_RELIABILITY", "RELIABILITY"),
-        )
-        if binding_member not in nvml.ClocksEventReasons.__members__
-    }
-
     _CASES.extend(
         [
             (
@@ -173,7 +164,7 @@
                 system_typing.ClocksEventReasons,
                 _device._CLOCKS_EVENT_REASONS_MAPPING,
                 set(),
-                _CLOCKS_EVENT_REASONS_STR_UNMAPPED,
+                set(),
             ),
             (
                 nvml.EventType,