pyopencl 2025.2.5__tar.gz → 2025.2.7__tar.gz

{pyopencl-2025.2.5 → pyopencl-2025.2.7}/.gitignore

@@ -10,7 +10,6 @@ _skbuild
 *.pyc
 build
 *.prof
-doc/hedge-notes.pdf
 *.vtk
 *.silo
 *.session
@@ -73,7 +72,4 @@ wheelhouse
 memray-*.bin
 memray-*.html
 
-.pylintrc.yml
-.run-pylint.py
-
 pyopencl/_cl_gen.pyi

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyopencl
-Version: 2025.2.5
+Version: 2025.2.7
 Summary: Python wrapper for OpenCL
 Author-Email: Andreas Kloeckner <inform@tiker.net>
 License-Expression: MIT
@@ -30,11 +30,10 @@ Requires-Dist: oclgrind-binary-distribution>=18.3; extra == "oclgrind"
 Provides-Extra: pocl
 Requires-Dist: pocl-binary-distribution>=1.2; extra == "pocl"
 Provides-Extra: test
-Requires-Dist: ruff; extra == "test"
+Requires-Dist: basedpyright; extra == "test"
 Requires-Dist: mako; extra == "test"
-Requires-Dist: mypy; extra == "test"
-Requires-Dist: pylint; extra == "test"
 Requires-Dist: pytest>=7; extra == "test"
+Requires-Dist: ruff; extra == "test"
 Description-Content-Type: text/x-rst
 
 PyOpenCL: Pythonic Access to OpenCL, with Arrays and Algorithms

{pyopencl-2025.2.5 → pyopencl-2025.2.7}/doc/algorithm.rst

@@ -24,6 +24,13 @@ Here's a usage example:
 :download:`examples/demo_elementwise.py <../examples/demo_elementwise.py>`
 in the PyOpenCL distribution.)
 
+References
+^^^^^^^^^^
+
+.. class:: WaitList
+
+    See :class:`pyopencl.WaitList`.
+
 .. _custom-reductions:
 
 Sums and counts ("reduce")

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

@@ -132,6 +132,8 @@ never directly.
 The :class:`Array` Class
 ------------------------
 
+.. autoclass:: ArrayT
+
 .. autoclass:: Array
 
 .. autoexception:: ArrayHasOffsetError
@@ -308,3 +310,10 @@ Generating Arrays of Random Numbers
 -----------------------------------
 
 .. automodule:: pyopencl.clrandom
+
+Type Aliases
+------------
+
+.. class:: cl.Device
+
+    See :class:`pyopencl.Device`.

{pyopencl-2025.2.5 → 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"
@@ -18,10 +20,11 @@ version = ".".join(str(x) for x in ver_dic["VERSION"])
 # The full version, including alpha/beta/rc tags.
 release = ver_dic["VERSION_TEXT"]
 
-
 nitpick_ignore = [
+    ("py:class", r"builtins.slice"),
     ("py:class", r"numpy._typing._dtype_like._SupportsDType"),
     ("py:class", r"numpy._typing._dtype_like._DTypeDict"),
+    ("py:class", r"pytest.Metafunc"),
 ]
 
 intersphinx_mapping = {
@@ -30,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.5 → 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.5 → pyopencl-2025.2.7}/doc/runtime.rst

@@ -55,10 +55,6 @@ References
 These are only here because Sphinx, our documentation tool, struggles to
 resolve them.
 
-.. class:: WaitList
-
-    A :class:`Sequence` of :class:`Event`\ s or *None*.
-
 .. class:: NDArray
 
     See :data:`numpy.typing.NDArray`.
@@ -75,6 +71,10 @@ resolve them.
 
     A generic type variable, used for a return type.
 
+.. class:: HashableT
+
+   An invariant type variable that is bound to :class:`Hashable`.
+
 .. class:: P
 
     A :class:`~typing.ParamSpec`.
@@ -95,7 +95,7 @@ resolve them.
 
 .. class:: WaitList
 
-    A :class:`Sequence` of :class:`Event`\ s or *None*.
+    See :class:`pyopencl.WaitList`.
 
 .. class:: Context
 
@@ -109,6 +109,10 @@ resolve them.
 
     See :class:`pyopencl.Event`.
 
+.. class:: KernelArg
+
+    A union of allowed kernel arguments.
+
 .. currentmodule:: cl_tools
 ..
 .. class:: AllocatorBase

{pyopencl-2025.2.5 → pyopencl-2025.2.7}/doc/runtime_queue.rst

@@ -116,6 +116,15 @@ Event
 
     |comparable|
 
+.. autodata:: WaitList
+    :noindex:
+
+    A :class:`Sequence` of :class:`Event`\ s or *None*.
+
+.. class:: WaitList
+
+    See above.
+
 .. class:: ProfilingInfoGetter
 
    .. attribute:: info

{pyopencl-2025.2.5 → pyopencl-2025.2.7}/pyopencl/_cl.pyi

@@ -20,7 +20,7 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 """
 
-from collections.abc import Callable, Sequence
+from collections.abc import Callable, Hashable, Sequence
 from enum import IntEnum, auto
 from typing import TYPE_CHECKING, Generic, Literal, overload
 
@@ -591,12 +591,12 @@ class kernel_arg_type_qualifier(IntEnum):  # noqa: N801
     to_string = classmethod(pyopencl._monkeypatch.to_string)
 
 class kernel_work_group_info(IntEnum):  # noqa: N801
-    WORK_GROUP_SIZE = auto()
-    COMPILE_WORK_GROUP_SIZE = auto()
-    LOCAL_MEM_SIZE = auto()
-    PREFERRED_WORK_GROUP_SIZE_MULTIPLE = auto()
-    PRIVATE_MEM_SIZE = auto()
-    GLOBAL_WORK_SIZE = auto()
+    WORK_GROUP_SIZE = 0x11B0
+    COMPILE_WORK_GROUP_SIZE = 0x11B1
+    LOCAL_MEM_SIZE = 0x11B2
+    PREFERRED_WORK_GROUP_SIZE_MULTIPLE = 0x11B3
+    PRIVATE_MEM_SIZE = 0x11B4
+    GLOBAL_WORK_SIZE = 0x11B5
     to_string = classmethod(pyopencl._monkeypatch.to_string)
 
 class kernel_sub_group_info(IntEnum):  # noqa: N801
@@ -837,6 +837,9 @@ class Device:
 
     __repr__ = pyopencl._monkeypatch.device_repr
 
+    @property
+    def hashable_model_and_version_identifier(self) -> Hashable: ...
+
     type: device_type
     vendor_id: int
     max_compute_units: int

{pyopencl-2025.2.5 → pyopencl-2025.2.7}/pyopencl/_monkeypatch.py

@@ -27,9 +27,11 @@ from sys import intern
 from typing import (
     TYPE_CHECKING,
     Any,
+    Literal,
     TextIO,
     TypeVar,
     cast,
+    overload,
 )
 from warnings import warn
 
@@ -221,7 +223,7 @@ kernel_old_get_info = _cl.Kernel.get_info
 kernel_old_get_work_group_info = _cl.Kernel.get_work_group_info
 
 
-def kernel_set_arg_types(self: _cl.Kernel, arg_types):
+def kernel_set_arg_types(self: _cl.Kernel, arg_types) -> None:
     arg_types = tuple(arg_types)
 
     # {{{ arg counting bug handling
@@ -259,7 +261,42 @@ def kernel_set_arg_types(self: _cl.Kernel, arg_types):
                     devs=self.context.devices))
 
 
-def kernel_get_work_group_info(self: _cl.Kernel, param: int, device: _cl.Device):
+@overload
+def kernel_get_work_group_info(
+            self: _cl.Kernel,
+            param: Literal[
+                    _cl.kernel_work_group_info.WORK_GROUP_SIZE,
+                    _cl.kernel_work_group_info.PREFERRED_WORK_GROUP_SIZE_MULTIPLE,
+                    _cl.kernel_work_group_info.LOCAL_MEM_SIZE,
+                    _cl.kernel_work_group_info.PRIVATE_MEM_SIZE,
+                ],
+            device: _cl.Device
+        ) -> int: ...
+
+@overload
+def kernel_get_work_group_info(
+            self: _cl.Kernel,
+            param: Literal[
+                    _cl.kernel_work_group_info.COMPILE_WORK_GROUP_SIZE,
+                    _cl.kernel_work_group_info.GLOBAL_WORK_SIZE,
+                ],
+            device: _cl.Device
+        ) -> Sequence[int]: ...
+
+
+@overload
+def kernel_get_work_group_info(
+            self: _cl.Kernel,
+            param: int,
+            device: _cl.Device
+        ) -> object: ...
+
+
+def kernel_get_work_group_info(
+            self: _cl.Kernel,
+            param: int,
+            device: _cl.Device
+        ) -> object:
     try:
         wg_info_cache = self._wg_info_cache
     except AttributeError:
@@ -411,9 +448,7 @@ def image_init(self: _cl.Image,
         else:
             raise ValueError("images cannot have more than three dimensions")
 
-        desc = _cl.ImageDescriptor() \
-            # pylint: disable=possibly-used-before-assignment
-
+        desc = _cl.ImageDescriptor()
         desc.image_type = image_type
         desc.shape = shape  # also sets desc.array_size
 

{pyopencl-2025.2.5 → pyopencl-2025.2.7}/pyopencl/algorithm.py

@@ -1235,7 +1235,7 @@ class ListOfListsBuilder:
                 queue, (n_objects + 1,), index_dtype, allocator=allocator)
             info_record.compressed_indices[0] = 0
 
-            compress_events[name] = compress_kernel(  # pylint: disable=possibly-used-before-assignment
+            compress_events[name] = compress_kernel(
                 info_record.starts,
                 compressed_counts,
                 info_record.nonempty_indices,