pyopencl 2025.2.1__cp311-cp311-macosx_11_0_arm64.whl → 2025.2.2__cp311-cp311-macosx_11_0_arm64.whl

pyopencl/compyte/array.py

@@ -20,10 +20,13 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 """
 
+from typing import Any, Protocol
+
 import numpy as np
+from typing_extensions import override
 
 
-def f_contiguous_strides(itemsize, shape):
+def f_contiguous_strides(itemsize: int, shape: tuple[int, ...]) -> tuple[int, ...]:
     if shape:
         strides = [itemsize]
         for s in shape[:-1]:
@@ -36,7 +39,7 @@ def f_contiguous_strides(itemsize, shape):
         return ()
 
 
-def c_contiguous_strides(itemsize, shape):
+def c_contiguous_strides(itemsize: int, shape: tuple[int, ...]) -> tuple[int, ...]:
     if shape:
         strides = [itemsize]
         for s in shape[:0:-1]:
@@ -49,7 +52,11 @@ def c_contiguous_strides(itemsize, shape):
         return ()
 
 
-def equal_strides(strides1, strides2, shape):
+def equal_strides(
+            strides1: tuple[int, ...],
+            strides2: tuple[int, ...],
+            shape: tuple[int, ...]
+        ) -> bool:
     if strides1 == strides2:
         return True
 
@@ -63,33 +70,53 @@ def equal_strides(strides1, strides2, shape):
     return True
 
 
-def is_f_contiguous_strides(strides, itemsize, shape):
+def is_f_contiguous_strides(
+            strides: tuple[int, ...],
+            itemsize: int,
+            shape: tuple[int, ...]
+        ) -> bool:
     from pytools import product
     return (
             equal_strides(strides, f_contiguous_strides(itemsize, shape), shape)
             or product(shape) == 0)
 
 
-def is_c_contiguous_strides(strides, itemsize, shape):
+def is_c_contiguous_strides(
+            strides: tuple[int, ...],
+            itemsize: int,
+            shape: tuple[int, ...]
+        ) -> bool:
     from pytools import product
     return (equal_strides(strides, c_contiguous_strides(itemsize, shape), shape)
             or product(shape) == 0)
 
 
+class ArrayIsh(Protocol):
+    shape: tuple[int, ...]
+    strides: tuple[int, ...]
+    dtype: np.dtype[Any]
+
+
 class ArrayFlags:
-    def __init__(self, ary):
+    f_contiguous: bool
+    c_contiguous: bool
+    forc: bool
+
+    def __init__(self, ary: ArrayIsh):
         self.f_contiguous = is_f_contiguous_strides(
             ary.strides, ary.dtype.itemsize, ary.shape)
         self.c_contiguous = is_c_contiguous_strides(
             ary.strides, ary.dtype.itemsize, ary.shape)
         self.forc = self.f_contiguous or self.c_contiguous
 
+    @override
     def __repr__(self):
         return (
                 f"  C_CONTIGUOUS : {self.c_contiguous}\n"
                 f"  F_CONTIGUOUS : {self.f_contiguous}"
                 )
 
+    @override
     def __str__(self):
         return repr(self)
 

pyopencl/compyte/dtypes.py

@@ -26,7 +26,11 @@ FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 OTHER DEALINGS IN THE SOFTWARE.
 """
 
+from collections.abc import Sequence
+from typing import Any, Callable, TypeVar
+
 import numpy as np
+from numpy.typing import DTypeLike
 
 
 class TypeNameNotKnown(RuntimeError):  # noqa: N818
@@ -36,11 +40,16 @@ class TypeNameNotKnown(RuntimeError):  # noqa: N818
 # {{{ registry
 
 class DTypeRegistry:
+    dtype_to_name: dict[np.dtype[Any] | str, str]
+    name_to_dtype: dict[str, np.dtype[Any]]
+
     def __init__(self):
         self.dtype_to_name = {}
         self.name_to_dtype = {}
 
-    def get_or_register_dtype(self, c_names, dtype=None):
+    def get_or_register_dtype(self,
+                c_names: str | Sequence[str],
+                dtype: DTypeLike | None = None):
         """Get or register a :class:`numpy.dtype` associated with the C type names
         in the string list *c_names*. If *dtype* is `None`, no registration is
         performed, and the :class:`numpy.dtype` must already have been registered.
@@ -94,8 +103,8 @@ class DTypeRegistry:
 
         return dtype
 
-    def dtype_to_ctype(self, dtype):
-        if dtype is None:
+    def dtype_to_ctype(self, dtype: np.dtype[Any]) -> str:
+        if dtype is None:  # pyright: ignore[reportUnnecessaryComparison]
             raise ValueError("dtype may not be None")
 
         dtype = np.dtype(dtype)
@@ -110,7 +119,11 @@ class DTypeRegistry:
 
 # {{{ C types
 
-def fill_registry_with_c_types(reg, respect_windows, include_bool=True):
+def fill_registry_with_c_types(
+            reg: DTypeRegistry,
+            respect_windows: bool,
+            include_bool: bool = True
+        ) -> None:
     import struct
     from sys import platform
 
@@ -157,7 +170,7 @@ def fill_registry_with_c_types(reg, respect_windows, include_bool=True):
     reg.get_or_register_dtype("double", np.float64)
 
 
-def fill_registry_with_opencl_c_types(reg):
+def fill_registry_with_opencl_c_types(reg: DTypeRegistry) -> None:
     reg.get_or_register_dtype(["char", "signed char"], np.int8)
     reg.get_or_register_dtype(["uchar", "unsigned char"], np.uint8)
     reg.get_or_register_dtype(["short", "signed short",
@@ -183,7 +196,7 @@ def fill_registry_with_opencl_c_types(reg):
     reg.get_or_register_dtype("double", np.float64)
 
 
-def fill_registry_with_c99_stdint_types(reg):
+def fill_registry_with_c99_stdint_types(reg: DTypeRegistry) -> None:
     reg.get_or_register_dtype("bool", np.bool_)
 
     reg.get_or_register_dtype("int8_t", np.int8)
@@ -200,7 +213,7 @@ def fill_registry_with_c99_stdint_types(reg):
     reg.get_or_register_dtype("double", np.float64)
 
 
-def fill_registry_with_c99_complex_types(reg):
+def fill_registry_with_c99_complex_types(reg: DTypeRegistry) -> None:
     reg.get_or_register_dtype("float complex", np.complex64)
     reg.get_or_register_dtype("double complex", np.complex128)
     reg.get_or_register_dtype("long double complex", np.clongdouble)
@@ -229,12 +242,21 @@ def _fill_dtype_registry(respect_windows, include_bool=True):
 
 # {{{ c declarator parsing
 
-def parse_c_arg_backend(c_arg, scalar_arg_factory, vec_arg_factory,
-        name_to_dtype=None):
+ArgTypeT = TypeVar("ArgTypeT")
+
+
+def parse_c_arg_backend(
+            c_arg: str,
+            scalar_arg_factory: Callable[[np.dtype[Any], str], ArgTypeT],
+            vec_arg_factory: Callable[[np.dtype[Any], str], ArgTypeT],
+            name_to_dtype: Callable[[str], np.dtype[Any]] | DTypeRegistry | None = None,
+        ):
     if isinstance(name_to_dtype, DTypeRegistry):
-        name_to_dtype = name_to_dtype.name_to_dtype__getitem__
+        name_to_dtype_clbl = name_to_dtype.name_to_dtype.__getitem__
     elif name_to_dtype is None:
-        name_to_dtype = NAME_TO_DTYPE.__getitem__
+        name_to_dtype_clbl = NAME_TO_DTYPE.__getitem__
+    else:
+        name_to_dtype_clbl = name_to_dtype
 
     c_arg = (c_arg
             .replace("const", "")
@@ -261,7 +283,7 @@ def parse_c_arg_backend(c_arg, scalar_arg_factory, vec_arg_factory,
     tp = " ".join(tp.split())
 
     try:
-        dtype = name_to_dtype(tp)
+        dtype = name_to_dtype_clbl(tp)
     except KeyError:
         raise ValueError(f"unknown type '{tp}'") from None
 
@@ -270,7 +292,11 @@ def parse_c_arg_backend(c_arg, scalar_arg_factory, vec_arg_factory,
 # }}}
 
 
-def register_dtype(dtype, c_names, alias_ok=False):
+def register_dtype(
+            dtype: DTypeLike,
+            c_names: Sequence[str] | str,
+            alias_ok: bool = False
+        ) -> None:
     from warnings import warn
     warn("register_dtype is deprecated. Use get_or_register_dtype instead.",
             DeprecationWarning, stacklevel=2)

pyopencl/compyte/pyproject.toml

@@ -32,3 +32,18 @@ known-first-party = [
     "pytools",
 ]
 lines-after-imports = 2
+
+
+[tool.basedpyright]
+reportImplicitStringConcatenation = "none"
+reportUnnecessaryIsInstance = "none"
+reportUnusedCallResult = "none"
+reportExplicitAny = "none"
+reportUnreachable = "hint"
+
+# This reports even cycles that are qualified by 'if TYPE_CHECKING'. Not what
+# we care about at this moment.
+# https://github.com/microsoft/pyright/issues/746
+reportImportCycles = "none"
+pythonVersion = "3.10"
+pythonPlatform = "All"

pyopencl/reduction.py

@@ -30,12 +30,14 @@ Based on code/ideas by Mark Harris <mharris@nvidia.com>.
 None of the original source code remains.
 """
 
+import builtins
 from dataclasses import dataclass
-from typing import Any
+from typing import TYPE_CHECKING, Any, Literal, cast, overload
 
 import numpy as np
 
 import pyopencl as cl
+import pyopencl.array as cl_array
 from pyopencl.tools import (
     DtypedArgument,
     KernelTemplateBase,
@@ -45,6 +47,10 @@ from pyopencl.tools import (
 )
 
 
+if TYPE_CHECKING:
+    from pyopencl.typing import Allocator
+
+
 # {{{ kernel source
 
 KERNEL = r"""//CL//
@@ -351,7 +357,40 @@ class ReductionKernel:
                 name=f"{name}_stage2", options=options, preamble=preamble,
                 max_group_size=max_group_size)
 
-    def __call__(self, *args: Any, **kwargs: Any) -> cl.Event:
+    @overload
+    def __call__(self,
+                *args: object,
+                return_event: Literal[True],
+                queue: cl.CommandQueue | None = None,
+                allocator: Allocator | None = None,
+                wait_for: cl.WaitList = None,
+                out: cl_array.Array | None = None,
+                range: slice | None = None,
+                slice: slice | None = None
+            ) ->  tuple[cl_array.Array, cl.Event]: ...
+
+    @overload
+    def __call__(self,
+                *args: object,
+                return_event: Literal[False],
+                queue: cl.CommandQueue | None = None,
+                allocator: Allocator | None = None,
+                wait_for: cl.WaitList = None,
+                out: cl_array.Array | None = None,
+                range: slice | None = None,
+                slice: slice | None = None
+            ) -> cl_array.Array: ...
+
+    def __call__(self,
+                *args: object,
+                return_event: bool = False,
+                queue: cl.CommandQueue | None = None,
+                allocator: Allocator | None = None,
+                wait_for: cl.WaitList = None,
+                out: cl_array.Array | None = None,
+                range: slice | None = None,
+                slice: slice | None = None
+            ) -> cl_array.Array | tuple[cl_array.Array, cl.Event]:
         """Invoke the generated kernel.
 
         |explain-waitfor|
@@ -390,18 +429,6 @@ class ReductionKernel:
             ``(scalar_array, event)``.
         """
 
-        queue = kwargs.pop("queue", None)
-        allocator = kwargs.pop("allocator", None)
-        wait_for = kwargs.pop("wait_for", None)
-        return_event = kwargs.pop("return_event", False)
-        out = kwargs.pop("out", None)
-
-        range_ = kwargs.pop("range", None)
-        slice_ = kwargs.pop("slice", None)
-
-        if kwargs:
-            raise TypeError("invalid keyword argument to reduction kernel")
-
         if wait_for is None:
             wait_for = []
         else:
@@ -415,13 +442,14 @@ class ReductionKernel:
 
         while True:
             invocation_args = []
-            vectors = []
+            vectors: list[cl_array.Array] = []
 
             array_empty = empty
 
             from pyopencl.tools import VectorArg
             for arg, arg_tp in zip(args, stage_inf.arg_types, strict=True):
                 if isinstance(arg_tp, VectorArg):
+                    assert isinstance(arg, cl_array.Array)
                     array_empty = arg.__class__
                     if not arg.flags.forc:
                         raise RuntimeError(
@@ -443,31 +471,30 @@ class ReductionKernel:
 
             # {{{ range/slice processing
 
-            if range_ is not None:
-                if slice_ is not None:
+            if range is not None:
+                if slice is not None:
                     raise TypeError("may not specify both range and slice "
                             "keyword arguments")
 
             else:
-                if slice_ is None:
-                    slice_ = slice(None)
+                if slice is None:
+                    slice = builtins.slice(None)
 
                 if repr_vec is None:
                     raise TypeError(
                             "must have vector argument when range is not specified")
 
-                range_ = slice(*slice_.indices(repr_vec.size))
+                range = builtins.slice(*slice.indices(repr_vec.size))
 
-            assert range_ is not None
+            assert range is not None
 
-            start = range_.start
+            start = cast("int | None", range.start)
             if start is None:
                 start = 0
-            if range_.step is None:
+            step = cast("int | None", range.step)
+            if step is None:
                 step = 1
-            else:
-                step = range_.step
-            sz = abs(range_.stop - start)//step
+            sz = abs(cast("int", range.stop) - start) //step
 
             # }}}
 
@@ -504,7 +531,7 @@ class ReductionKernel:
                 macrogroup_size = group_count*stage_inf.group_size
                 seq_count = (sz + macrogroup_size - 1) // macrogroup_size
 
-            size_args = [start, step, range_.stop, seq_count, sz]
+            size_args = [start, step, range.stop, seq_count, sz]
 
             if group_count == 1 and out is not None:
                 result = out
@@ -536,7 +563,7 @@ class ReductionKernel:
                 stage_inf = self.stage_2_inf
                 args = (result, *stage1_args)
 
-                range_ = slice_ = None
+                range = slice = None
 
 # }}}
 

pyopencl/typing.py

@@ -23,7 +23,7 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 """
 
-from collections.abc import Sequence
+from collections.abc import Callable, Sequence
 from typing import TYPE_CHECKING, Any, TypeAlias, TypeVar
 
 import numpy as np
@@ -33,6 +33,7 @@ from typing_extensions import Buffer as abc_Buffer
 
 if TYPE_CHECKING:
     import pyopencl as _cl
+    import pyopencl.array as _cl_array
 
 
 DTypeT = TypeVar("DTypeT", bound=np.dtype[Any])
@@ -49,4 +50,8 @@ KernelArg: TypeAlias = """
     | _cl.Buffer
     | _cl.Image
     | _cl.Sampler
-    | _cl.SVMPointer"""
+    | _cl.SVMPointer
+    | _cl_array.Array
+    | None"""
+
+Allocator: TypeAlias = "Callable[[int], _cl.MemoryObjectHolder | _cl.SVMPointer]"

{pyopencl-2025.2.1.dist-info → pyopencl-2025.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyopencl
-Version: 2025.2.1
+Version: 2025.2.2
 Summary: Python wrapper for OpenCL
 Author-Email: Andreas Kloeckner <inform@tiker.net>
 License-Expression: MIT

{pyopencl-2025.2.1.dist-info → pyopencl-2025.2.2.dist-info}/RECORD

@@ -1,16 +1,12 @@
-pyopencl-2025.2.1.dist-info/RECORD,,
-pyopencl-2025.2.1.dist-info/WHEEL,sha256=7CikrUOKtiy-2ivl-sZ5UBq05y8ywDmf9_HO2E_GAYY,141
-pyopencl-2025.2.1.dist-info/METADATA,sha256=ZWuBJXxGy78MiQ68EB8rPStHxAZ7PNgk5LJqDH2OFbg,4756
-pyopencl-2025.2.1.dist-info/licenses/LICENSE,sha256=wiBvs-UC54bB5DswWuvB66B96b4hkYw_VLt8IR0cBPI,15284
 pyopencl/algorithm.py,sha256=NuBpzVlfY8V6nKcGcQe5uXGnVka7LV1hHCi-uBCAxVc,51370
 pyopencl/clmath.py,sha256=d5aHPy0WQMb01anNAX2nk7t-yRBIEif7sW3aTXEiEUo,8258
 pyopencl/version.py,sha256=53bqHNFDyobSVMAdQ47kiOx1UdhKXDLkTtzNwfLEvY4,304
 pyopencl/capture_call.py,sha256=c1X3HKwtikqgkSuMSvMNKezYCrAtCc8Wt60cvGfFJTQ,6479
-pyopencl/reduction.py,sha256=Lt0Lwnt34RVxyojZ9CiTFj2E_23_vrK5mbj4TZ9l6V8,25507
+pyopencl/reduction.py,sha256=2Gf91Rw0mUu61SVSqhYWhReV5Ru28Qw60CXmXLhSDO8,26644
 pyopencl/tools.py,sha256=bKX4ND-DcKXr7dCFJoJ6C3SWPNFNKA490oDofTaRuCI,46596
 pyopencl/elementwise.py,sha256=uc8aXWv60yUtlwwKhCOVVeo5o7ylPoQPTRLsnRF6gTY,38678
 pyopencl/cache.py,sha256=RiSW4K8FynlV8ibwyDngzSFOa3k-Zi16GSSK05yrdZk,16017
-pyopencl/__init__.py,sha256=_Q1cnbWmRGkZRENKLJwmNXGDbYDmoLuA5T7X7W3nVIU,62740
+pyopencl/__init__.py,sha256=70i8Bo6qCxXT61loiSs4MfHBk_MDzgDN9eVMnR8Kab4,63178
 pyopencl/_cl.pyi,sha256=VmAi42KFEFL5UdCMuX7I45j16mve3RBGXFfkbSwBICw,56235
 pyopencl/cltypes.py,sha256=hDpwt24YFtng0Z4qNJB5D0NxkOIGwnLRUQKoaWcMcnE,6045
 pyopencl/invoker.py,sha256=ws1kdpzIkc7vc7s4jxtJ7IcdCa5qOT5DiYW6rMxjVVg,13706
@@ -21,16 +17,17 @@ pyopencl/_cluda.py,sha256=kjI6pW9M-zGCVDsautsKsYMs1U9v73_W5riUjdzmKA0,2111
 pyopencl/_cl.cpython-311-darwin.so,sha256=QoKLIpHEtXulXf1-d_lE37FBk1lWnpzSVt0UGcFfFHg,539736
 pyopencl/_mymako.py,sha256=1xp0pWcE57hpIDKw915CvOGQpC-Jral02Inykx_k198,657
 pyopencl/_monkeypatch.py,sha256=bBJjQYW6XdiurbG_jgTzJg5BtDteuNVqRQ3Fuzx-Gbk,34845
-pyopencl/typing.py,sha256=sKeC1VCKI57C5SQDNr3yGtJk1QLtPbhlE_bRL4JmAQ4,1805
-pyopencl/array.py,sha256=mVJtRU7Zfgf9aVv9PW1Y_UIz6er_GTBkYLmzbu8kvgo,109596
+pyopencl/typing.py,sha256=DbU4i2J3fMXRqK49rbsaOpMz7xsaeTMY66iMsc6ZOmo,1970
+pyopencl/array.py,sha256=_6HR45dN7bH9uiTDPXqbco8ZDKlddkOIzLMV14j_LWg,111551
 pyopencl/bitonic_sort.py,sha256=FpTfCf_s22O-SgPWBoXZjCG_k1afm7uUwnO9N7jubgg,8052
 pyopencl/bitonic_sort_templates.py,sha256=316nsWdr7jpg9g8SmZk8KCoHmHKabjvOdBaufxQPw8Y,16188
 pyopencl/clrandom.py,sha256=Ox_Eebq5fcjsBXpa68C1kt8Ca6CPi5tXwXhEhLY_yP8,13081
-pyopencl/compyte/pyproject.toml,sha256=XuQgiCiOLu7OsrAdhBzbHcM_tLnU9-28KNVDNn9qbZw,722
+pyopencl/compyte/pyproject.toml,sha256=yISxFxhotloFnx_Adffvnr0w2samJwGonr64Eq2YDQo,1145
 pyopencl/compyte/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pyopencl/compyte/dtypes.py,sha256=C303fA_dumU9WyfXaYh53DerGgODJGfDkSEJ8d1_dAk,9905
+pyopencl/compyte/dtypes.py,sha256=NIv-oOc-DO6YLvX34grMUdefqWVl6QbxztWsGcvnuvA,10803
 pyopencl/compyte/.gitignore,sha256=HVYtbKRtOCPWZavrgYqO2u7UKFiE7g7ympmuQzKbBBw,168
-pyopencl/compyte/array.py,sha256=LQID4Wi04aktqgxCqfQXed9ouxhDuovlBE-fKHLrv9E,7380
+pyopencl/compyte/array.py,sha256=t1XORyCQ4zRe_KUnoo4LH1W5XOcGXaf5xvGtgOPDzyg,8035
+pyopencl/compyte/.basedpyright/baseline.json,sha256=3HQHk-43Nuqf3UVXVCQli6nXbrqnrQvQtsZPjuLR61g,37643
 pyopencl/characterize/__init__.py,sha256=AaDNS1p2RtXqGf6tA--GB9FRp-fzavbpgulrAW_BE2Y,14565
 pyopencl/characterize/performance.py,sha256=i8X1jWlyL1bUD7wuDoSlk1CckJkT6zAVPefziQwmNXU,6903
 pyopencl/cl/pyopencl-bessel-y.cl,sha256=VDy8l4lVxO8VcJR_maeGu_Qjnw27j28zBwhaTKDhBBg,12297
@@ -44,3 +41,7 @@ pyopencl/cl/pyopencl-random123/threefry.cl,sha256=2WmQGxx5gPSv22UL9_MlXv0eMug91k
 pyopencl/cl/pyopencl-random123/philox.cl,sha256=vYcQH7Vw13Q3qkW5Nhy1HTUDbWLGKoloE1YP0VWk6vU,21740
 pyopencl/cl/pyopencl-random123/array.h,sha256=nIV0zDWYuybldNgtsh79icNtDXHYdDsSpFaWIvDTyw4,17088
 pyopencl/cl/pyopencl-random123/openclfeatures.h,sha256=pAPbl7JkQgJxulSuGGevpaI43P7PwiH2mYxtNfHq59M,2881
+pyopencl-2025.2.2.dist-info/RECORD,,
+pyopencl-2025.2.2.dist-info/WHEEL,sha256=7CikrUOKtiy-2ivl-sZ5UBq05y8ywDmf9_HO2E_GAYY,141
+pyopencl-2025.2.2.dist-info/METADATA,sha256=PtweID2xOHA7p_vJBtyUejmEOn9Mw0_s1mlTWubMAH8,4756
+pyopencl-2025.2.2.dist-info/licenses/LICENSE,sha256=wiBvs-UC54bB5DswWuvB66B96b4hkYw_VLt8IR0cBPI,15284