Small fixes

Author: mdboom

cuda_bindings/cuda/bindings/_nvml.pxd

@@ -41,7 +41,6 @@ ctypedef nvmlComputeInstanceProfileInfo_t ComputeInstanceProfileInfo
 ctypedef nvmlMask255_t Mask255
 ctypedef nvmlHostname_v1_t Hostname_v1
 ctypedef nvmlNvLinkInfo_v1_t NvLinkInfo_v1
-ctypedef nvmlPRMCounterInput_v1_t PRMCounterInput_v1
 ctypedef nvmlPowerValue_v2_t PowerValue_v2
 ctypedef nvmlVgpuProcessUtilizationSample_t VgpuProcessUtilizationSample
 ctypedef nvmlGpuFabricInfo_t GpuFabricInfo

cuda_bindings/cuda/bindings/_nvml.pyx

@@ -1176,7 +1176,7 @@ NVLINK_MAX_LINKS = 18
 
 
 class RUSD(_IntEnum):
-    POLL_NONE - 0x0        # Disable RUSD polling on all metric groups
+    POLL_NONE = 0x0        # Disable RUSD polling on all metric groups
     POLL_CLOCK = 0x1       # Enable RUSD polling on clock group
     POLL_PERF = 0x2        # Enable RUSD polling on performance group
     POLL_MEMORY = 0x4      # Enable RUSD polling on memory group
@@ -15297,6 +15297,126 @@ cdef class RusdSettings_v1:
         return obj
 
 
+cdef _get_prm_counter_input_v1_dtype_offsets():
+    cdef nvmlPRMCounterInput_v1_t pod = nvmlPRMCounterInput_v1_t()
+    return _numpy.dtype({
+        'names': ['local_port'],
+        'formats': [_numpy.uint32],
+        'offsets': [
+            (<intptr_t>&(pod.localPort)) - (<intptr_t>&pod),
+        ],
+        'itemsize': sizeof(nvmlPRMCounterInput_v1_t),
+    })
+
+prm_counter_input_v1_dtype = _get_prm_counter_input_v1_dtype_offsets()
+
+cdef class PRMCounterInput_v1:
+    """Empty-initialize an instance of `nvmlPRMCounterInput_v1_t`.
+
+
+    .. seealso:: `nvmlPRMCounterInput_v1_t`
+    """
+    cdef:
+        nvmlPRMCounterInput_v1_t *_ptr
+        object _owner
+        bint _owned
+        bint _readonly
+
+    def __init__(self):
+        self._ptr = <nvmlPRMCounterInput_v1_t *>calloc(1, sizeof(nvmlPRMCounterInput_v1_t))
+        if self._ptr == NULL:
+            raise MemoryError("Error allocating PRMCounterInput_v1")
+        self._owner = None
+        self._owned = True
+        self._readonly = False
+
+    def __dealloc__(self):
+        cdef nvmlPRMCounterInput_v1_t *ptr
+        if self._owned and self._ptr != NULL:
+            ptr = self._ptr
+            self._ptr = NULL
+            free(ptr)
+
+    def __repr__(self):
+        return f"<{__name__}.PRMCounterInput_v1 object at {hex(id(self))}>"
+
+    @property
+    def ptr(self):
+        """Get the pointer address to the data as Python :class:`int`."""
+        return <intptr_t>(self._ptr)
+
+    cdef intptr_t _get_ptr(self):
+        return <intptr_t>(self._ptr)
+
+    def __int__(self):
+        return <intptr_t>(self._ptr)
+
+    def __eq__(self, other):
+        cdef PRMCounterInput_v1 other_
+        if not isinstance(other, PRMCounterInput_v1):
+            return False
+        other_ = other
+        return (memcmp(<void *><intptr_t>(self._ptr), <void *><intptr_t>(other_._ptr), sizeof(nvmlPRMCounterInput_v1_t)) == 0)
+
+    def __setitem__(self, key, val):
+        if key == 0 and isinstance(val, _numpy.ndarray):
+            self._ptr = <nvmlPRMCounterInput_v1_t *>malloc(sizeof(nvmlPRMCounterInput_v1_t))
+            if self._ptr == NULL:
+                raise MemoryError("Error allocating PRMCounterInput_v1")
+            memcpy(<void*>self._ptr, <void*><intptr_t>val.ctypes.data, sizeof(nvmlPRMCounterInput_v1_t))
+            self._owner = None
+            self._owned = True
+            self._readonly = not val.flags.writeable
+        else:
+            setattr(self, key, val)
+
+    @property
+    def local_port(self):
+        """int: Local port number."""
+        return self._ptr[0].localPort
+
+    @local_port.setter
+    def local_port(self, val):
+        if self._readonly:
+            raise ValueError("This PRMCounterInput_v1 instance is read-only")
+        self._ptr[0].localPort = val
+
+    @staticmethod
+    def from_data(data):
+        """Create an PRMCounterInput_v1 instance wrapping the given NumPy array.
+
+        Args:
+            data (_numpy.ndarray): a single-element array of dtype `prm_counter_input_v1_dtype` holding the data.
+        """
+        return __from_data(data, "prm_counter_input_v1_dtype", prm_counter_input_v1_dtype, PRMCounterInput_v1)
+
+    @staticmethod
+    def from_ptr(intptr_t ptr, bint readonly=False, object owner=None):
+        """Create an PRMCounterInput_v1 instance wrapping the given pointer.
+
+        Args:
+            ptr (intptr_t): pointer address as Python :class:`int` to the data.
+            owner (object): The Python object that owns the pointer. If not provided, data will be copied.
+            readonly (bool): whether the data is read-only (to the user). default is `False`.
+        """
+        if ptr == 0:
+            raise ValueError("ptr must not be null (0)")
+        cdef PRMCounterInput_v1 obj = PRMCounterInput_v1.__new__(PRMCounterInput_v1)
+        if owner is None:
+            obj._ptr = <nvmlPRMCounterInput_v1_t *>malloc(sizeof(nvmlPRMCounterInput_v1_t))
+            if obj._ptr == NULL:
+                raise MemoryError("Error allocating PRMCounterInput_v1")
+            memcpy(<void*>(obj._ptr), <void*>ptr, sizeof(nvmlPRMCounterInput_v1_t))
+            obj._owner = None
+            obj._owned = True
+        else:
+            obj._ptr = <nvmlPRMCounterInput_v1_t *>ptr
+            obj._owner = owner
+            obj._owned = False
+        obj._readonly = readonly
+        return obj
+
+
 cdef _get_excluded_device_info_dtype_offsets():
     cdef nvmlExcludedDeviceInfo_t pod = nvmlExcludedDeviceInfo_t()
     return _numpy.dtype({