[chore]: Upgrade to stubgen-pyx 0.2.22

.pre-commit-config.yaml

@@ -88,7 +88,7 @@ repos:
         files: ^cuda_core/cuda/.*\.(pyx|pxd)$
         pass_filenames: false
         additional_dependencies:
-          - stubgen-pyx==0.2.19
+          - stubgen-pyx==0.2.22
           - Cython==3.2.9
 
   # Link checking for authored documentation files

cuda_core/cuda/core/_context.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_context.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_context.pyx
 
 from collections.abc import Sequence
 from dataclasses import dataclass

cuda_core/cuda/core/_device.pyi

@@ -1,4 +1,6 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_device.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_device.pyx
+
+import threading
 
 import cuda.core.system
 from cuda.core._context import Context, ContextOptions
@@ -13,6 +15,8 @@ from cuda.core.texture import (MipmappedArray, MipmappedArrayOptions,
                                ResourceDescriptor, SurfaceObject,
                                TextureObject, TextureObjectOptions)
 
+_tls = threading.local()
+_lock = threading.Lock()
 __all__ = ['Device']
 
 class DeviceProperties:

cuda_core/cuda/core/_device_resources.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_device_resources.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_device_resources.pyx
 
 from collections.abc import Sequence as SequenceABC
 from dataclasses import dataclass

cuda_core/cuda/core/_dlpack.pyi

@@ -1,10 +1,17 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_dlpack.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_dlpack.pyx
 
 from enum import IntEnum
-from typing import Any, Callable, TypeAlias, TypedDict
+from typing import Any, Callable, TypedDict
+
+from typing_extensions import TypeAlias
 
 _DLDeviceType: TypeAlias = int
 DLDataTypeCode: TypeAlias = int
+DLPackManagedTensorAllocator: TypeAlias = Callable[[DLTensor, DLManagedTensorVersioned, Any, Callable[[Any, bytes, bytes], None]], int]
+DLPackManagedTensorFromPyObjectNoSync: TypeAlias = Callable[[Any, DLManagedTensorVersioned], int]
+DLPackManagedTensorToPyObjectNoSync: TypeAlias = Callable[[DLManagedTensorVersioned, Any], int]
+DLPackDLTensorFromPyObjectNoSync: TypeAlias = Callable[[Any, DLTensor], int]
+DLPackCurrentWorkStream: TypeAlias = Callable[[_DLDeviceType, int, Any], int]
 
 class DLDeviceType(IntEnum):
     kDLCPU = 1
@@ -22,7 +29,7 @@ class DLDataType(TypedDict):
     lanes: int
 
 class DLTensor(TypedDict):
-    data: None
+    data: Any
     device: DLDevice
     ndim: int
     dtype: DLDataType
@@ -32,17 +39,17 @@ class DLTensor(TypedDict):
 
 class DLManagedTensor(TypedDict):
     dl_tensor: DLTensor
-    manager_ctx: None
-    deleter: Callable[..., Any]
+    manager_ctx: Any
+    deleter: Callable[[DLManagedTensor], None]
 
 class DLPackVersion(TypedDict):
     major: int
     minor: int
 
 class DLManagedTensorVersioned(TypedDict):
     version: DLPackVersion
-    manager_ctx: None
-    deleter: Callable[..., Any]
+    manager_ctx: Any
+    deleter: Callable[[DLManagedTensorVersioned], None]
     flags: int
     dl_tensor: DLTensor
 
@@ -52,11 +59,11 @@ class DLPackExchangeAPIHeader(TypedDict):
 
 class DLPackExchangeAPI(TypedDict):
     header: DLPackExchangeAPIHeader
-    managed_tensor_allocator: ...
-    managed_tensor_from_py_object_no_sync: ...
-    managed_tensor_to_py_object_no_sync: ...
-    dltensor_from_py_object_no_sync: ...
-    current_work_stream: ...
+    managed_tensor_allocator: DLPackManagedTensorAllocator
+    managed_tensor_from_py_object_no_sync: DLPackManagedTensorFromPyObjectNoSync
+    managed_tensor_to_py_object_no_sync: DLPackManagedTensorToPyObjectNoSync
+    dltensor_from_py_object_no_sync: DLPackDLTensorFromPyObjectNoSync
+    current_work_stream: DLPackCurrentWorkStream
 
 def classify_dl_device(buf: object) -> tuple[int, int]:
     """Classify a buffer into a DLPack (device_type, device_id) pair.

cuda_core/cuda/core/_event.pyi

@@ -1,9 +1,9 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_event.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_event.pyx
 
 from dataclasses import dataclass
-from typing import Any
 
 import cuda.bindings.driver
+from _typeshed import Incomplete
 from cuda.core._context import Context
 from cuda.core._device import Device
 
@@ -146,7 +146,7 @@ class IPCEventDescriptor:
     """Serializable object describing an event that can be shared between processes."""
     def __init__(self, *arg, **kwargs) -> None: ...
     @staticmethod
-    def _init(reserved: bytes, is_blocking_sync: Any) -> IPCEventDescriptor: ...
+    def _init(reserved: bytes, is_blocking_sync: Incomplete) -> IPCEventDescriptor: ...
     def __eq__(self, other: object) -> bool: ...
     def __reduce__(self) -> tuple[object, ...]: ...
 

cuda_core/cuda/core/_graphics.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_graphics.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_graphics.pyx
 
 from typing import Sequence
 

cuda_core/cuda/core/_kernel_arg_handler.pyi

@@ -1,9 +1,13 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_kernel_arg_handler.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_kernel_arg_handler.pyx
 
-from typing import Any, Sequence, TypeAlias, TypedDict
+from typing import Any, Sequence, TypedDict
 
-cpp_single_complex: TypeAlias = Any
-cpp_double_complex: TypeAlias = Any
+from _typeshed import Incomplete
+from typing_extensions import TypeAlias
+
+cpp_single_complex: TypeAlias = Incomplete
+cpp_double_complex: TypeAlias = Incomplete
+voidptr: TypeAlias = Any
 
 class __half_raw(TypedDict):
     x: int

cuda_core/cuda/core/_launch_config.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_launch_config.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_launch_config.pyx
 
 from typing import Any
 

cuda_core/cuda/core/_launcher.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_launcher.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_launcher.pyx
 
 from cuda.core._launch_config import LaunchConfig
 from cuda.core._module import Kernel

cuda_core/cuda/core/_layout.pyi

@@ -1,6 +1,9 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_layout.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_layout.pyx
 
-from typing import Any, TypeAlias, TypedDict
+from typing import TypedDict
+
+from _typeshed import Incomplete
+from typing_extensions import TypeAlias
 
 OrderFlag: TypeAlias = int
 Property: TypeAlias = int
@@ -9,8 +12,8 @@ stride_t: TypeAlias = int
 axis_t: TypeAlias = int
 axes_mask_t: TypeAlias = int
 property_mask_t: TypeAlias = int
-extents_strides_t: TypeAlias = Any
-axis_vec_t: TypeAlias = Any
+extents_strides_t: TypeAlias = Incomplete
+axis_vec_t: TypeAlias = Incomplete
 
 class _StridedLayout:
     """

cuda_core/cuda/core/_linker.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_linker.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_linker.pyx
 
 """Linking machinery for combining object codes.
 
@@ -13,9 +13,11 @@ import cuda.bindings.driver
 import cuda.bindings.nvjitlink
 from cuda.core._module import ObjectCode
 from cuda.core.typing import CompilerBackendType, ObjectCodeFormatType
+from typing_extensions import TypeAlias
 
 _keep_driver_in_stub: cuda.bindings.driver.CUlinkState
 _keep_nvjitlink_in_stub: cuda.bindings.nvjitlink.nvJitLinkHandle
+const_char_ptr: TypeAlias = bytes
 __all__ = ['Linker', 'LinkerOptions']
 LinkerHandleT = Union['cuda.bindings.nvjitlink.nvJitLinkHandle', 'cuda.bindings.driver.CUlinkState']
 _driver = None

cuda_core/cuda/core/_memory/_buffer.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_buffer.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_buffer.pyx
 
 from typing import TypedDict
 

cuda_core/cuda/core/_memory/_copy_attributes.pyi

@@ -1 +1 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_copy_attributes.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_copy_attributes.pyx

cuda_core/cuda/core/_memory/_copy_ops.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_copy_ops.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_copy_ops.pyx
 
 from collections.abc import Sequence
 

cuda_core/cuda/core/_memory/_device_memory_resource.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_device_memory_resource.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_device_memory_resource.pyx
 
 import uuid
 from dataclasses import dataclass

cuda_core/cuda/core/_memory/_graph_memory_resource.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_graph_memory_resource.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_graph_memory_resource.pyx
 
 from cuda.core._device import Device
 from cuda.core._memory._buffer import Buffer, MemoryResource

cuda_core/cuda/core/_memory/_ipc.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_ipc.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_ipc.pyx
 
 import uuid
 

cuda_core/cuda/core/_memory/_location.pyi

@@ -1 +1 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_location.pxd
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_location.pxd

cuda_core/cuda/core/_memory/_managed_memory_ops.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_managed_memory_ops.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_managed_memory_ops.pyx
 
 from collections.abc import Sequence
 

cuda_core/cuda/core/_memory/_managed_memory_resource.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_managed_memory_resource.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_managed_memory_resource.pyx
 
 from dataclasses import dataclass
 

cuda_core/cuda/core/_memory/_memory_pool.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_memory_pool.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_memory_pool.pyx
 
 import uuid
 

cuda_core/cuda/core/_memory/_peer_access_utils.pyi

@@ -1,12 +1,13 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_peer_access_utils.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_peer_access_utils.pyx
 
-from collections.abc import Iterable, Iterator, Set
+from collections.abc import Callable, Iterable, Iterator, MutableSet, Set
 from dataclasses import dataclass
-from typing import Any, Callable
+from typing import Any, TypeVar
 
 from cuda.core._device import Device
 from cuda.core._memory._device_memory_resource import DeviceMemoryResource
 
+_S = TypeVar('_S')
 
 @dataclass(frozen=True)
 class PeerAccessPlan:
@@ -15,7 +16,7 @@ class PeerAccessPlan:
     to_add: tuple[int, ...]
     to_remove: tuple[int, ...]
 
-class PeerAccessibleBySetProxy:
+class PeerAccessibleBySetProxy(MutableSet['Device']):
     """Live driver-backed view of the peer devices granted access to a memory pool.
 
     Reads (``__contains__``, ``__iter__``, ``len(...)``) call ``cuMemPoolGetAccess``;
@@ -33,11 +34,11 @@ class PeerAccessibleBySetProxy:
     is updated, so coalescing into a single driver call lets the toolkit handle
     the mappings in parallel.
     """
-    __slots__ = '_mr'
+    __slots__ = ('_mr',)
 
     def __init__(self, mr: DeviceMemoryResource) -> None: ...
     @classmethod
-    def _from_iterable(cls, it: Iterable[Device]) -> set[Device]: ...
+    def _from_iterable(cls, it: Iterable[_S]) -> set[_S]: ...
     def __contains__(self, value: object) -> bool: ...
     def __iter__(self) -> Iterator[Device]: ...
     def __len__(self) -> int: ...
@@ -55,10 +56,12 @@ class PeerAccessibleBySetProxy:
         """Restrict peer access to the intersection in a single driver call."""
     def symmetric_difference_update(self, other: Iterable[Device | int]) -> None:
         """Toggle peer access for every device in ``other`` in one driver call."""
-    def __ior__(self, other: Set[Any]) -> PeerAccessibleBySetProxy: ...
+    def __ior__(self, other: Set[Any]) -> PeerAccessibleBySetProxy: # type: ignore[misc]
+        ...
     def __iand__(self, other: Set[Any]) -> PeerAccessibleBySetProxy: ...
     def __isub__(self, other: Set[Any]) -> PeerAccessibleBySetProxy: ...
-    def __ixor__(self, other: Set[Any]) -> PeerAccessibleBySetProxy: ...
+    def __ixor__(self, other: Set[Any]) -> PeerAccessibleBySetProxy: # type: ignore[misc]
+        ...
     def __repr__(self) -> str: ...
     def _apply(self, additions, removals) -> None:
         """Compute the diff and issue a single ``cuMemPoolSetAccess``.

cuda_core/cuda/core/_memory/_peer_access_utils.pyx

@@ -6,7 +6,9 @@ from __future__ import annotations
 
 from collections.abc import Callable, Iterable, Iterator, MutableSet, Set
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, TypeVar
+
+_S = TypeVar("_S")
 
 from cuda.bindings cimport cydriver
 from cuda.core._memory._device_memory_resource cimport DeviceMemoryResource
@@ -235,7 +237,7 @@ class PeerAccessibleBySetProxy(MutableSet["Device"]):
         self._mr = mr
 
     @classmethod
-    def _from_iterable(cls, it: Iterable[Device]) -> set[Device]:  # type: ignore[override]
+    def _from_iterable(cls, it: Iterable[_S]) -> set[_S]:
         # Binary set operators (&, |, -, ^) collect their result through
         # _from_iterable. Returning a plain set lets the user reason about
         # the result independently of any pool's driver state.
@@ -354,7 +356,7 @@ class PeerAccessibleBySetProxy(MutableSet["Device"]):
         if to_add or to_remove:
             self._apply(to_add, to_remove)
 
-    def __ior__(self, other: Set[Any]) -> PeerAccessibleBySetProxy:
+    def __ior__(self, other: Set[Any]) -> PeerAccessibleBySetProxy:  # type: ignore[misc]
         self.update(other)
         return self
 
@@ -369,7 +371,7 @@ class PeerAccessibleBySetProxy(MutableSet["Device"]):
             self.difference_update(other)
         return self
 
-    def __ixor__(self, other: Set[Any]) -> PeerAccessibleBySetProxy:
+    def __ixor__(self, other: Set[Any]) -> PeerAccessibleBySetProxy:  # type: ignore[misc]
         self.symmetric_difference_update(other)
         return self
 

cuda_core/cuda/core/_memory/_pinned_memory_resource.pyi

@@ -1,4 +1,4 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memory/_pinned_memory_resource.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memory/_pinned_memory_resource.pyx
 
 import uuid
 from dataclasses import dataclass

cuda_core/cuda/core/_memoryview.pyi

@@ -1,7 +1,8 @@
-# This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_memoryview.pyx
+# This file was generated by stubgen-pyx v0.2.22 from cuda_core/cuda/core/_memoryview.pyx
 
 import functools
-from typing import Any, Callable, TypedDict
+from collections.abc import Callable
+from typing import Any, TypedDict
 
 import numpy
 from cuda.core._layout import _StridedLayout
@@ -11,10 +12,9 @@ from cuda.core._tensor_map import TensorMapDescriptorOptions
 
 from ._dlpack import *
 
-_SMV_DLPACK_EXCHANGE_API_CAPSULE = ...
 
 class PyTypeObject(TypedDict):
-    tp_dict: None
+    tp_dict: Any
 
 class StridedMemoryView:
     """A class holding metadata of a strided dense array/tensor.