pyopencl 2025.2.6__tar.gz → 2025.2.7__tar.gz

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/PKG-INFO

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

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/doc/array.rst

@@ -132,6 +132,8 @@ never directly.
 The :class:`Array` Class
 ------------------------
 
+.. autoclass:: ArrayT
+
 .. autoclass:: Array
 
 .. autoexception:: ArrayHasOffsetError

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/doc/conf.py

@@ -1,5 +1,7 @@
 from urllib.request import urlopen
 
+from sphinx.application import Sphinx
+
 
 _conf_url = \
         "https://raw.githubusercontent.com/inducer/sphinxconfig/main/sphinxconfig.py"
@@ -31,3 +33,11 @@ intersphinx_mapping = {
     "mako": ("https://docs.makotemplates.org/en/latest", None),
     "pytools": ("https://documen.tician.de/pytools", None),
 }
+
+sphinxconfig_missing_reference_aliases: dict[str, str] = {
+    "np.dtype": "class:numpy.dtype",
+}
+
+
+def setup(app: Sphinx) -> None:
+    app.connect("missing-reference", process_autodoc_missing_reference)  # noqa: F821

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/doc/index.rst

@@ -61,8 +61,8 @@ Tutorials
   `Problem set 2 <https://tiker.net/pub/simula-pyopencl-probset2.pdf>`__
 * Ian Johnson's `PyOpenCL tutorial <https://web.archive.org/web/20170907175053/http://enja.org:80/2011/02/22/adventures-in-pyopencl-part-1-getting-started-with-python>`__.
 
-Software that works with or enhances PyOpenCL
-=============================================
+Software related to PyOpenCL
+============================
 
 * Jon Roose's `pyclblas <https://pyclblas.readthedocs.io/en/latest/index.html>`__
   (`code <https://github.com/jroose/pyclblas>`__)
@@ -97,6 +97,10 @@ Software that works with or enhances PyOpenCL
 * Vincent Favre-Nicolin's `pyvkfft <https://github.com/vincefn/pyvkfft/>`__
   makes `vkfft <https://github.com/DTolm/VkFFT>`__ accessible from PyOpenCL.
 
+* Want something like PyOpenCL but for Apple's `Metal <https://en.wikipedia.org/wiki/Metal_(API)>`__?
+  Sean True's `PyMetallic <https://github.com/seantrue/pymetallic/>`__ may be
+  worth considering.
+
 If you know of a piece of software you feel that should be on this list, please
 let me know, or, even better, send a patch!
 

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/pyopencl/array.py

@@ -40,6 +40,7 @@ from typing import (
     Literal,
     ParamSpec,
     TypeAlias,
+    TypeVar,
     cast,
 )
 from warnings import warn
@@ -75,6 +76,9 @@ else:
     _SVMPointer_or_nothing = ()
 
 
+ArrayT = TypeVar("ArrayT", bound="Array")
+
+
 class _NoValue:
     pass
 
@@ -2424,7 +2428,7 @@ def empty_like(
             allocator=allocator)
 
 
-def zeros_like(ary):
+def zeros_like(ary: ArrayT) -> ArrayT:
     """Make a new, zero-initialized :class:`Array` having the same properties
     as *other_ary*.
     """

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/pyopencl/compyte/array.py

@@ -63,7 +63,7 @@ def equal_strides(
     if len(strides1) != len(strides2) or len(strides2) != len(shape):
         return False
 
-    for s, st1, st2 in zip(shape, strides1, strides2):
+    for s, st1, st2 in zip(shape, strides1, strides2, strict=True):
         if s != 1 and st1 != st2:
             return False
 
@@ -98,30 +98,27 @@ class ArrayIsh(Protocol):
 
 
 class ArrayFlags:
-    f_contiguous: bool
-    c_contiguous: bool
-    forc: bool
-
-    def __init__(self, ary: ArrayIsh):
-        self.f_contiguous = is_f_contiguous_strides(
+    def __init__(self, ary: ArrayIsh) -> None:
+        self.f_contiguous: bool = is_f_contiguous_strides(
             ary.strides, ary.dtype.itemsize, ary.shape)
-        self.c_contiguous = is_c_contiguous_strides(
+        self.c_contiguous: bool = is_c_contiguous_strides(
             ary.strides, ary.dtype.itemsize, ary.shape)
-        self.forc = self.f_contiguous or self.c_contiguous
+        self.forc: bool = self.f_contiguous or self.c_contiguous
 
     @override
-    def __repr__(self):
+    def __repr__(self) -> str:
         return (
                 f"  C_CONTIGUOUS : {self.c_contiguous}\n"
                 f"  F_CONTIGUOUS : {self.f_contiguous}"
                 )
 
     @override
-    def __str__(self):
+    def __str__(self) -> str:
         return repr(self)
 
 
-def get_common_dtype(obj1, obj2, allow_double):
+def get_common_dtype(obj1: ArrayIsh, obj2: ArrayIsh,
+                     allow_double: bool) -> np.dtype[Any]:
     # Yes, numpy behaves differently depending on whether
     # we're dealing with arrays or scalars.
 
@@ -143,33 +140,6 @@ def get_common_dtype(obj1, obj2, allow_double):
     return result
 
 
-def bound(a):
-    high = a.bytes
-    low = a.bytes
-
-    for stri, shp in zip(a.strides, a.shape):
-        if stri < 0:
-            low += (stri)*(shp-1)
-        else:
-            high += (stri)*(shp-1)
-    return low, high
-
-
-def may_share_memory(a, b):
-    # When this is called with a an ndarray and b
-    # a sparse matrix, numpy.may_share_memory fails.
-    if a is b:
-        return True
-    if a.__class__ is b.__class__:
-        a_l, a_h = bound(a)
-        b_l, b_h = bound(b)
-        if b_l >= a_h or a_l >= b_h:
-            return False
-        return True
-    else:
-        return False
-
-
 # {{{ as_strided implementation
 
 try:

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/pyopencl/compyte/dtypes.py

@@ -26,8 +26,8 @@ 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
+from collections.abc import Callable, Sequence
+from typing import Any, TypeVar
 
 import numpy as np
 from numpy.typing import DTypeLike
@@ -40,14 +40,12 @@ 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) -> None:
+        self.dtype_to_name: dict[np.dtype[Any] | str, str] = {}
+        self.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,
+    def get_or_register_dtype(
+                self,
                 names: str | Sequence[str],
                 dtype: DTypeLike | None = None) -> np.dtype[Any]:
         """Get or register a :class:`numpy.dtype` associated with the C type names
@@ -227,7 +225,7 @@ dtype_to_ctype = TYPE_REGISTRY.dtype_to_ctype
 get_or_register_dtype = TYPE_REGISTRY.get_or_register_dtype
 
 
-def _fill_dtype_registry(respect_windows, include_bool=True):
+def _fill_dtype_registry(respect_windows: bool, include_bool: bool = True) -> None:
     fill_registry_with_c_types(
             TYPE_REGISTRY, respect_windows, include_bool)
 
@@ -244,7 +242,7 @@ def parse_c_arg_backend(
             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,
-        ):
+        ) -> ArgTypeT:
     if isinstance(name_to_dtype, DTypeRegistry):
         name_to_dtype_clbl = name_to_dtype.name_to_dtype.__getitem__
     elif name_to_dtype is None:

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/pyopencl/compyte/pyproject.toml

@@ -40,9 +40,6 @@ 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.6 → pyopencl-2025.2.7}/pyopencl/tools.py

@@ -772,7 +772,7 @@ def pytest_generate_tests_for_pyopencl(metafunc: pytest.Metafunc) -> None:
     functions, and they will automatically be run for each OpenCL device/platform
     in the system, as appropriate.
 
-    The following two environment variabls is also supported to control
+    The following two environment variables is also supported to control
     device/platform choice::
 
         PYOPENCL_TEST=0:0,1;intel=i5,i7

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/pyproject.toml

@@ -11,7 +11,7 @@ requires = [
 
 [project]
 name = "pyopencl"
-version = "2025.2.6"
+version = "2025.2.7"
 description = "Python wrapper for OpenCL"
 readme = "README.rst"
 license = "MIT"
@@ -148,6 +148,9 @@ environment-pass = [
 test-skip = [
     "*-macosx_arm64",
 ]
+skip = [
+    "cp31?t-*",
+]
 
 [tool.cibuildwheel.linux]
 skip = [
@@ -172,9 +175,6 @@ before-all = [
 repair-wheel-command = "auditwheel repair -w {dest_dir} --lib-sdir=/.libs {wheel}"
 
 [tool.cibuildwheel.macos]
-skip = [
-    "pp*",
-]
 before-all = "bash {package}/scripts/build-ocl-macos.sh"
 test-command = "pytest {project}/test/test_array.py" # same limitation as conda-forge
 
@@ -187,7 +187,6 @@ MACOSX_DEPLOYMENT_TARGET = "10.14"
 [tool.cibuildwheel.windows]
 skip = [
     "*-win32",
-    "pp*",
 ]
 test-command = ""
 before-all = "bash {package}/scripts/build-ocl-windows.sh"

{pyopencl-2025.2.6 → pyopencl-2025.2.7}/test/test_array.py

@@ -462,7 +462,7 @@ def test_pow_number(ctx_factory: cl.CtxFactory):
 # {{{ test_multiply
 
 def test_multiply(ctx_factory: cl.CtxFactory):
-    """Test the muliplication of an array with a scalar. """
+    """Test the multiplication of an array with a scalar. """
 
     context = ctx_factory()
     queue = cl.CommandQueue(context)