[chore]: Upgrade to stubgen-pyx 0.2.22 (#2763)

Author: mdboom

.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