pyopencl 2025.2.2__tar.gz → 2025.2.4__tar.gz

{pyopencl-2025.2.2 → pyopencl-2025.2.4}/PKG-INFO

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

{pyopencl-2025.2.2 → pyopencl-2025.2.4}/doc/runtime.rst

@@ -57,7 +57,7 @@ resolve them.
 
 .. class:: WaitList
 
-    A :class:`Sequence` of :class:`Event`\ s, or None.
+    A :class:`Sequence` of :class:`Event`\ s or *None*.
 
 .. class:: NDArray
 
@@ -95,11 +95,11 @@ resolve them.
 
 .. class:: WaitList
 
-    A :class:`Sequence` of :class:`Event`\ s, or None.
+    A :class:`Sequence` of :class:`Event`\ s or *None*.
 
 .. class:: Context
 
-    See :class:`pyopencl.CommandQueue`.
+    See :class:`pyopencl.Context`.
 
 .. class:: CommandQueue
 

{pyopencl-2025.2.2 → pyopencl-2025.2.4}/pyopencl/array.py

@@ -59,6 +59,7 @@ from pyopencl.compyte.array import (
     equal_strides as _equal_strides,
     f_contiguous_strides as _f_contiguous_strides,
 )
+from pyopencl.typing import Allocator
 
 
 if TYPE_CHECKING:
@@ -66,8 +67,6 @@ if TYPE_CHECKING:
 
     from numpy.typing import DTypeLike, NDArray
 
-    from pyopencl.typing import Allocator
-
 
 SCALAR_CLASSES = (Number, np.bool_, bool)
 
@@ -3404,4 +3403,39 @@ def cumsum(a, output_dtype=None, queue: cl.CommandQueue | None = None,
 
 # }}}
 
+
+__all__ = [
+    "Allocator",
+    "Array",
+    "all",
+    "any",
+    "arange",
+    "as_strided",
+    "concatenate",
+    "cumsum",
+    "diff",
+    "dot",
+    "empty_like",
+    "hstack",
+    "if_positive",
+    "logical_and",
+    "logical_not",
+    "logical_or",
+    "maximum",
+    "minimum",
+    "multi_put",
+    "multi_take",
+    "multi_take_put",
+    "reshape",
+    "stack",
+    "subset_dot",
+    "sum",
+    "take",
+    "to_device",
+    "transpose",
+    "vdot",
+    "zeros",
+    "zeros_like",
+]
+
 # vim: foldmethod=marker

{pyopencl-2025.2.2 → pyopencl-2025.2.4}/pyopencl/compyte/dtypes.py

@@ -48,8 +48,8 @@ class DTypeRegistry:
         self.name_to_dtype = {}
 
     def get_or_register_dtype(self,
-                c_names: str | Sequence[str],
-                dtype: DTypeLike | None = None):
+                names: str | Sequence[str],
+                dtype: DTypeLike | None = None) -> np.dtype[Any]:
         """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.
@@ -66,12 +66,12 @@ class DTypeRegistry:
         .. versionadded:: 2012.2
         """
 
-        if isinstance(c_names, str):
-            c_names = [c_names]
+        if isinstance(names, str):
+            names = [names]
 
         if dtype is None:
             from pytools import single_valued
-            return single_valued(self.name_to_dtype[name] for name in c_names)
+            return single_valued(self.name_to_dtype[name] for name in names)
 
         dtype = np.dtype(dtype)
 
@@ -86,7 +86,7 @@ class DTypeRegistry:
             assert existing_dtype == dtype
             dtype = existing_dtype
 
-        for nm in c_names:
+        for nm in names:
             try:
                 name_dtype = self.name_to_dtype[nm]
             except KeyError:
@@ -97,14 +97,14 @@ class DTypeRegistry:
                         f"name '{nm}' already registered to different dtype")
 
         if not existed:
-            self.dtype_to_name[dtype] = c_names[0]
+            self.dtype_to_name[dtype] = names[0]
         if str(dtype) not in self.dtype_to_name:
-            self.dtype_to_name[str(dtype)] = c_names[0]
+            self.dtype_to_name[str(dtype)] = names[0]
 
         return dtype
 
-    def dtype_to_ctype(self, dtype: np.dtype[Any]) -> str:
-        if dtype is None:  # pyright: ignore[reportUnnecessaryComparison]
+    def dtype_to_ctype(self, dtype: DTypeLike) -> str:
+        if dtype is None:
             raise ValueError("dtype may not be None")
 
         dtype = np.dtype(dtype)
@@ -160,12 +160,6 @@ def fill_registry_with_c_types(
                 f"{i64_name} unsigned int"],
                 np.uint64)
 
-    # https://github.com/numpy/numpy/issues/2610
-    if is_64_bit:
-        reg.get_or_register_dtype([f"unsigned {i64_name}"], np.uintp)
-    else:
-        reg.get_or_register_dtype(["unsigned"], np.uintp)
-
     reg.get_or_register_dtype("float", np.float32)
     reg.get_or_register_dtype("double", np.float64)
 

{pyopencl-2025.2.2 → pyopencl-2025.2.4}/pyopencl/compyte/pyproject.toml

@@ -40,6 +40,9 @@ reportUnnecessaryIsInstance = "none"
 reportUnusedCallResult = "none"
 reportExplicitAny = "none"
 reportUnreachable = "hint"
+# array.py looks like stdlib array, but pyright doesn't know this
+# won't ever be a top-level anything.
+reportShadowedImports = "none"
 
 # This reports even cycles that are qualified by 'if TYPE_CHECKING'. Not what
 # we care about at this moment.

{pyopencl-2025.2.2 → pyopencl-2025.2.4}/pyopencl/typing.py

@@ -55,3 +55,7 @@ KernelArg: TypeAlias = """
     | None"""
 
 Allocator: TypeAlias = "Callable[[int], _cl.MemoryObjectHolder | _cl.SVMPointer]"
+
+
+__all__ = [
+]

{pyopencl-2025.2.2 → pyopencl-2025.2.4}/pyproject.toml

@@ -11,7 +11,7 @@ requires = [
 
 [project]
 name = "pyopencl"
-version = "2025.2.2"
+version = "2025.2.4"
 description = "Python wrapper for OpenCL"
 readme = "README.rst"
 license = "MIT"
@@ -65,6 +65,7 @@ Repository = "https://github.com/inducer/pyopencl"
 
 [tool.scikit-build]
 sdist.exclude = [
+    ".basedpyright",
     ".mypy_cache",
     ".ci",
     ".github",