pyopencl 2024.2.6__cp311-cp311-win_amd64.whl → 2024.3__cp311-cp311-win_amd64.whl

pyopencl/scan.py

@@ -28,6 +28,7 @@ from dataclasses import dataclass
 from typing import Any, Dict, List, Optional, Set, Tuple, Union
 
 import numpy as np
+
 from pytools.persistent_dict import WriteOncePersistentDict
 
 import pyopencl as cl
@@ -35,9 +36,16 @@ import pyopencl._mymako as mako
 import pyopencl.array
 from pyopencl._cluda import CLUDA_PREAMBLE
 from pyopencl.tools import (
-    DtypedArgument, KernelTemplateBase, _NumpyTypesKeyBuilder,
-    _process_code_for_macro, bitlog2, context_dependent_memoize, dtype_to_ctype,
-    get_arg_list_scalar_arg_dtypes, get_arg_offset_adjuster_code)
+    DtypedArgument,
+    KernelTemplateBase,
+    _NumpyTypesKeyBuilder,
+    _process_code_for_macro,
+    bitlog2,
+    context_dependent_memoize,
+    dtype_to_ctype,
+    get_arg_list_scalar_arg_dtypes,
+    get_arg_offset_adjuster_code,
+)
 
 
 logger = logging.getLogger(__name__)
@@ -1047,9 +1055,9 @@ class GenericScanKernelBase(ABC):
         from pyopencl.tools import parse_arg_list
         self.parsed_args = parse_arg_list(arguments)
         from pyopencl.tools import VectorArg
-        self.first_array_idx = [
+        self.first_array_idx = next(
                 i for i, arg in enumerate(self.parsed_args)
-                if isinstance(arg, VectorArg)][0]
+                if isinstance(arg, VectorArg))
 
         self.input_expr = input_expr
 
@@ -1178,7 +1186,7 @@ class GenericScanKernel(GenericScanKernelBase):
                 result = generic_scan_kernel_cache[cache_key]
                 from_cache = True
                 logger.debug(
-                    "cache hit for generated scan kernel '%s'" % self.name_prefix)
+                    "cache hit for generated scan kernel '%s'", self.name_prefix)
                 (
                     self.first_level_scan_gen_info,
                     self.second_level_scan_gen_info,
@@ -1188,7 +1196,7 @@ class GenericScanKernel(GenericScanKernelBase):
 
         if not from_cache:
             logger.debug(
-                    "cache miss for generated scan kernel '%s'" % self.name_prefix)
+                    "cache miss for generated scan kernel '%s'", self.name_prefix)
             self._finish_setup_impl()
 
             result = (self.first_level_scan_gen_info,
@@ -1316,8 +1324,10 @@ class GenericScanKernel(GenericScanKernelBase):
         # {{{ build second-level scan
 
         from pyopencl.tools import VectorArg
-        second_level_arguments = self.parsed_args + [
-                VectorArg(self.dtype, "interval_sums")]
+        second_level_arguments = [
+            *self.parsed_args,
+            VectorArg(self.dtype, "interval_sums"),
+            ]
 
         second_level_build_kwargs: Dict[str, Optional[str]] = {}
         if self.is_segmented:
@@ -1364,9 +1374,10 @@ class GenericScanKernel(GenericScanKernelBase):
             use_lookbehind_update=self.use_lookbehind_update,
             **self.code_variables))
 
-        update_scalar_arg_dtypes = (
-                get_arg_list_scalar_arg_dtypes(self.parsed_args)
-                + [self.index_dtype, self.index_dtype, None, None])
+        update_scalar_arg_dtypes = [
+                *get_arg_list_scalar_arg_dtypes(self.parsed_args),
+                self.index_dtype, self.index_dtype, None, None]
+
         if self.is_segmented:
             # g_first_segment_start_in_interval
             update_scalar_arg_dtypes.append(None)
@@ -1558,7 +1569,8 @@ class GenericScanKernel(GenericScanKernelBase):
 
         # {{{ first level scan of interval (one interval per block)
 
-        scan1_args = data_args + [
+        scan1_args = [
+                *data_args,
                 partial_scan_buffer.data, n, interval_size, interval_results.data,
                 ]
 
@@ -1582,12 +1594,15 @@ class GenericScanKernel(GenericScanKernelBase):
         # can scan at most one interval
         assert interval_size >= num_intervals
 
-        scan2_args = data_args + [
+        scan2_args = [
+                *data_args,
                 interval_results.data,  # interval_sums
                 ]
+
         if self.is_segmented:
             scan2_args.append(first_segment_start_in_interval.data)
-        scan2_args = scan2_args + [
+        scan2_args = [
+                *scan2_args,
                 interval_results.data,  # partial_scan_buffer
                 num_intervals, interval_size]
 
@@ -1599,7 +1614,8 @@ class GenericScanKernel(GenericScanKernelBase):
 
         # {{{ update intervals with result of interval scan
 
-        upd_args = data_args + [
+        upd_args = [
+                *data_args,
                 n, interval_size, interval_results.data, partial_scan_buffer.data]
         if self.is_segmented:
             upd_args.append(first_segment_start_in_interval.data)
@@ -1705,10 +1721,11 @@ class GenericDebugScanKernel(GenericScanKernelBase):
 
         scan_prg = cl.Program(self.context, scan_src).build(self.options)
         self.kernel = getattr(scan_prg, f"{self.name_prefix}_debug_scan")
-        scalar_arg_dtypes = (
-                [None]
-                + get_arg_list_scalar_arg_dtypes(self.parsed_args)
-                + [self.index_dtype])
+        scalar_arg_dtypes = [
+                None,
+                *get_arg_list_scalar_arg_dtypes(self.parsed_args),
+                self.index_dtype,
+                ]
         self.kernel.set_scalar_arg_dtypes(scalar_arg_dtypes)
 
     def __call__(self, *args: Any, **kwargs: Any) -> cl.Event:
@@ -1755,8 +1772,7 @@ class GenericDebugScanKernel(GenericScanKernelBase):
 
         # }}}
 
-        return self.kernel(queue, (1,), (1,),
-                *(data_args + [n]), wait_for=wait_for)
+        return self.kernel(queue, (1,), (1,), *([*data_args, n]), wait_for=wait_for)
 
 # }}}
 

pyopencl/tools.py

@@ -132,21 +132,24 @@ from sys import intern
 from typing import Any, List, Optional, Union
 
 import numpy as np
+
 from pytools import memoize, memoize_method
 from pytools.persistent_dict import KeyBuilder as KeyBuilderBase
 
 from pyopencl._cl import bitlog2, get_cl_header_version  # noqa: F401
-from pyopencl.compyte.dtypes import TypeNameNotKnown  # noqa: F401
 from pyopencl.compyte.dtypes import (  # noqa: F401
-    dtype_to_ctype, get_or_register_dtype, register_dtype)
+    TypeNameNotKnown,
+    dtype_to_ctype,
+    get_or_register_dtype,
+    register_dtype,
+)
 
 
 # Do not add a pyopencl import here: This will add an import cycle.
 
 
 def _register_types():
-    from pyopencl.compyte.dtypes import (
-        TYPE_REGISTRY, fill_registry_with_opencl_c_types)
+    from pyopencl.compyte.dtypes import TYPE_REGISTRY, fill_registry_with_opencl_c_types
 
     fill_registry_with_opencl_c_types(TYPE_REGISTRY)
 
@@ -159,12 +162,17 @@ _register_types()
 
 # {{{ imported names
 
-from pyopencl._cl import (  # noqa: F401
-    AllocatorBase, DeferredAllocator, ImmediateAllocator, MemoryPool, PooledBuffer)
+from pyopencl._cl import (
+    AllocatorBase,
+    DeferredAllocator,
+    ImmediateAllocator,
+    MemoryPool,
+    PooledBuffer,
+)
 
 
 if get_cl_header_version() >= (2, 0):
-    from pyopencl._cl import PooledSVM, SVMAllocator, SVMPool  # noqa: F401
+    from pyopencl._cl import PooledSVM, SVMAllocator, SVMPool
 
 # }}}
 
@@ -521,7 +529,7 @@ def first_arg_dependent_memoize_nested(nested_func):
         cache_context = caller_frame.f_globals[
                 caller_frame.f_code.co_name]
     finally:
-        #del caller_frame
+        # del caller_frame
         pass
 
     try:
@@ -653,7 +661,8 @@ def get_pyopencl_fixture_arg_names(metafunc, extra_arg_names=None):
     supported_arg_names = [
             "platform", "device",
             "ctx_factory", "ctx_getter",
-            ] + extra_arg_names
+            *extra_arg_names
+            ]
 
     arg_names = []
     for arg in supported_arg_names:
@@ -978,7 +987,7 @@ class _CDeclList:
             return
 
         for _name, field_data in sorted(dtype.fields.items()):
-            field_dtype, offset = field_data[:2]
+            field_dtype, _offset = field_data[:2]
             self.add_dtype(field_dtype)
 
         _, cdecl = match_dtype_to_c_struct(
@@ -1062,7 +1071,7 @@ def match_dtype_to_c_struct(device, name, dtype, context=None):
 
     c_fields = []
     for field_name, dtype_and_offset in fields:
-        field_dtype, offset = dtype_and_offset[:2]
+        field_dtype, _offset = dtype_and_offset[:2]
         if hasattr(field_dtype, "subdtype") and field_dtype.subdtype is not None:
             array_dtype = field_dtype.subdtype[0]
             if hasattr(array_dtype, "subdtype") and array_dtype.subdtype is not None:
@@ -1087,7 +1096,7 @@ def match_dtype_to_c_struct(device, name, dtype, context=None):
 
     cdl = _CDeclList(device)
     for _field_name, dtype_and_offset in fields:
-        field_dtype, offset = dtype_and_offset[:2]
+        field_dtype, _offset = dtype_and_offset[:2]
         cdl.add_dtype(field_dtype)
 
     pre_decls = cdl.get_declarations()
@@ -1121,7 +1130,8 @@ def match_dtype_to_c_struct(device, name, dtype, context=None):
     prg = cl.Program(context, src)
     knl = prg.build(devices=[device]).get_size_and_offsets
 
-    import pyopencl.array  # noqa: F401
+    import pyopencl.array
+
     result_buf = cl.array.empty(queue, 1+len(fields), np.uint32)
     knl(queue, (1,), (1,), result_buf.data)
     queue.finish()

{pyopencl-2024.2.6.dist-info → pyopencl-2024.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyopencl
-Version: 2024.2.6
+Version: 2024.3
 Summary: Python wrapper for OpenCL
 Author-Email: Andreas Kloeckner <inform@tiker.net>
 Classifier: Development Status :: 5 - Production/Stable
@@ -12,22 +12,25 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Natural Language :: English
 Classifier: Programming Language :: C++
 Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Scientific/Engineering :: Mathematics
 Classifier: Topic :: Scientific/Engineering :: Physics
-Project-URL: Homepage, https://mathema.tician.de/software/pyopencl
 Project-URL: Documentation, https://documen.tician.de/pyopencl
+Project-URL: Homepage, https://mathema.tician.de/software/pyopencl
 Project-URL: Repository, https://github.com/inducer/pyopencl
 Requires-Python: ~=3.8
 Requires-Dist: importlib-resources; python_version < "3.9"
 Requires-Dist: numpy
-Requires-Dist: platformdirs>=2.2.0
+Requires-Dist: platformdirs>=2.2
 Requires-Dist: pytools>=2024.1.5
-Requires-Dist: oclgrind_binary_distribution>=18.3; extra == "oclgrind"
-Requires-Dist: pocl_binary_distribution>=1.2; extra == "pocl"
-Requires-Dist: Mako; extra == "test"
-Requires-Dist: pytest>=7.0.0; extra == "test"
+Requires-Dist: oclgrind-binary-distribution>=18.3; extra == "oclgrind"
+Requires-Dist: pocl-binary-distribution>=1.2; extra == "pocl"
+Requires-Dist: ruff; extra == "test"
+Requires-Dist: mako; extra == "test"
+Requires-Dist: mypy; extra == "test"
+Requires-Dist: pylint; extra == "test"
+Requires-Dist: pytest>=7; extra == "test"
 Provides-Extra: oclgrind
 Provides-Extra: pocl
 Provides-Extra: test

pyopencl-2024.3.dist-info/RECORD

@@ -0,0 +1,42 @@
+pyopencl/__init__.py,sha256=wCR09HsGaQ532Dc5vdU8WiyZGZ0qtHBbJQrPQMEGpYI,83853
+pyopencl/_cl.cp311-win_amd64.pyd,sha256=UXxsO8CNEkppDzva8VK98YUSbskdMmkil0FZCnSbUUE,580096
+pyopencl/_cluda.py,sha256=5coLt4eR5BOLgfUesgnaTx_nUhZ6BBhYi9dodvl7cbc,2128
+pyopencl/_mymako.py,sha256=55mRxOnOH9N7ltPWPND19GD4cszQrB2dhiMUlouoWKU,634
+pyopencl/algorithm.py,sha256=7ilnIG3c4YMDUBP4w4e10UMDHj9nbFITg1oLnEEsej0,52742
+pyopencl/array.py,sha256=Ij8euQy7zxxyxyjw7IM841K0MirNV5Zgr690_wMWmAA,115210
+pyopencl/bitonic_sort.py,sha256=r5Jy-C7LE5ml-Qt0KwD4E0g-Cf4DwpmO_3d25KLNUwY,8263
+pyopencl/bitonic_sort_templates.py,sha256=XQjUiHS9KjFIUYIil1Ls9293hy751AOsJmszlNS-IFk,16745
+pyopencl/cache.py,sha256=HMS-dEURWulvFOzEY33LzNSw3notDY_Rsv_L16OiSuo,16535
+pyopencl/capture_call.py,sha256=3zldPy-ssAchA9Q0ntDDcao2zMPWK9wDPdjdYj4CrIk,5871
+pyopencl/characterize/__init__.py,sha256=rBhH9M6wkp2fYs51vx9BHgNVl-N1Vz_2ofV0uhPsGk0,14802
+pyopencl/characterize/performance.py,sha256=xwyt4SsABo4LipXzjRF7TKz8LVBcLmZZ5os6f7hp3Go,7103
+pyopencl/cl/pyopencl-airy.cl,sha256=HSUEWbUN2MNzuhvDP3LB9dVY_3dhjFSWXhvGQsrA8VA,8446
+pyopencl/cl/pyopencl-bessel-j-complex.cl,sha256=XOUBUZFxgra9nUAymnbJugcOa1lQyj-fwiZA9ly3rdI,6264
+pyopencl/cl/pyopencl-bessel-j.cl,sha256=V16uWa8t1b0oxNHRCLhCqxzR-28yr-5ZslqHkw2Ved0,24358
+pyopencl/cl/pyopencl-bessel-y.cl,sha256=kD6u2qtgmrNNq75w5uaxHdrvb6oeeTPgrGpF29ERsuE,12732
+pyopencl/cl/pyopencl-complex.h,sha256=DHFUM2sHCv50m4LJvvIpF8sMWc5yOO6kYu9nWlScl6A,8847
+pyopencl/cl/pyopencl-eval-tbl.cl,sha256=SDYB_RkW7jtN_Y_xhCbwGjBSNmBpMg8T4ZXlrP_2q9U,2736
+pyopencl/cl/pyopencl-hankel-complex.cl,sha256=yrxPF4wgr9bTzG4SyJsVbij5SODoJvWaDRDDuo4zrs4,32005
+pyopencl/cl/pyopencl-random123/array.h,sha256=oTYPJfU7s4IXy8xRc3H0rqMq4mgyVxd7UuLSf3liUPY,17413
+pyopencl/cl/pyopencl-random123/openclfeatures.h,sha256=jauJ1WEspr-YNefAuUwPhapl_JQDVa6my_h4fLl-p4o,2974
+pyopencl/cl/pyopencl-random123/philox.cl,sha256=KqPbLt54UwrHPvBe7v4ZMzqI2oGrzp0c0QzqVhydNCY,22226
+pyopencl/cl/pyopencl-random123/threefry.cl,sha256=bC78-HJVuc4AFiNppglDQiN8cZRD45hG1PPdi4Sdt-o,55563
+pyopencl/clmath.py,sha256=GL9s0YwKpCgvsGl6ndoWXEngPFWAqRRR_HOoQm57OA8,8502
+pyopencl/clrandom.py,sha256=baRE9GgAqGcp2xy9FgI0E4oEygLqyjZqwIs6YvB35ok,13453
+pyopencl/cltypes.py,sha256=2jCvMMBtkhXxfbBFBfM4NSFKRg7kEo1tlwGJQlyiwmQ,4955
+pyopencl/compyte/.gitignore,sha256=PFMRSJycIqPtcpEn7VqbcenWJqHFDuYtwQm_tLbADt8,189
+pyopencl/compyte/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pyopencl/compyte/array.py,sha256=IJzw-dSS_injEiLVNHLlSRi0R2F1sO_8xeeK-XezsSo,7636
+pyopencl/compyte/dtypes.py,sha256=wjJy3pXU6mdYmlFeX-8VM97gH37S5VmMVE1PBNo9om8,10133
+pyopencl/compyte/pyproject.toml,sha256=pcYndN5KAcdZXgsBUiH3VyeMxrV-oesECjS7yGCaY4I,1272
+pyopencl/elementwise.py,sha256=cqUtTgSCys1Z__gOCqyOXYnzPU-jdGz5NKZ9pkaF4oE,39796
+pyopencl/invoker.py,sha256=gTnJQN4_fJ9-3_wKqXfWVpDKzstD2jgYu5qxzPKXdY4,14481
+pyopencl/ipython_ext.py,sha256=e-K-Yjd6sTP9-9d69v0k_1_ifdK77JWMhbynn1FBDH4,1983
+pyopencl/reduction.py,sha256=QoXRhi2_RjUpGDhcfWiMzHHXoBZJuRncnBTn5xwJgR4,26319
+pyopencl/scan.py,sha256=5YoKCBdIHoPha9onGna_65ne2maU0RZ7nR7X2L47YrA,67553
+pyopencl/tools.py,sha256=Pzjm7juH52NZ5fF7ppgCy-p9kR-Jhtw8xqnh-qCEjfs,47467
+pyopencl/version.py,sha256=Fp_v5G59_24Um9NujHpex3VRwdffKTwtN8tjiQl5QOQ,276
+pyopencl-2024.3.dist-info/METADATA,sha256=3Y2AW8BahKcnAPNcX0jR65xiQPsUluv8sTbrVsCtZZc,4783
+pyopencl-2024.3.dist-info/WHEEL,sha256=kXCl1J14PkmxQKXf5U_5vxmme_OmC3Ydcral7u0yA3M,106
+pyopencl-2024.3.dist-info/licenses/LICENSE,sha256=jib9h6nV8oAvCkKPKDecpLakmQi1SktCn4YXmllgySY,15566
+pyopencl-2024.3.dist-info/RECORD,,

{pyopencl-2024.2.6.dist-info → pyopencl-2024.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: scikit-build-core 0.9.5
+Generator: scikit-build-core 0.10.7
 Root-Is-Purelib: false
 Tag: cp311-cp311-win_amd64
 

pyopencl/compyte/.git

@@ -1 +0,0 @@
-gitdir: ../../.git/modules/pyopencl/compyte

pyopencl/compyte/ndarray/Makefile

@@ -1,31 +0,0 @@
-all: pygpu_ndarray.so
-
-PYTHONVERSION ?= $(shell python -c "import sys; print '%d.%d'%(sys.version_info[0], sys.version_info[1]"))
-CUDA_ROOT ?= /opt/lisa/os/cuda
-THEANO_ROOT ?= /u/bastienf/repos/Theano
-
-
-CFLAGS=-g -DDEBUG -DOFFSET
-# By default enable the OFFSET usage. Otherwise some test fail.
-CFLAGS=-g -DOFFSET
-#BINDIR=--compiler-bindir ${HOME}/.theano.nvcc-bindir
-
-#NPY_PATH!=python -c "import numpy;print numpy.__path__"
-#NPY_INCLUDE=-I${NPY_PATH}/core/include
-CUDA_INCLUDE=-I${CUDA_ROOT}/include
-PYTHON_INCLUDE=-I$(shell python -c "import distutils.sysconfig;print distutils.sysconfig.get_python_inc()")
-INCLUDES=${CUDA_INCLUDE} ${PYTHON_INCLUDE}
-CUDA_FLAGS=-Xlinker -rpath,${CUDA_ROOT}/lib64 -Xlinker -rpath,${CUDA_ROOT}/lib
-
-pygpu_language_cuda.o: pygpu_language_cuda.cu pygpu_language.h
-	nvcc -c ${CFLAGS} -m64 -Xcompiler -fPIC,-m64 ${CUDA_FLAGS} ${INCLUDES} ${BINDIR} -o $@ $<
-
-pygpu_ndarray.so: pygpu_ndarray.cpp pygpu_ndarray.h pygpu_language_cuda.o pygpu_ndarray_object.h
-	nvcc -shared ${CFLAGS} -m64 -Xcompiler -fPIC,-m64 ${CUDA_FLAGS} ${INCLUDES} ${BINDIR} -o $@ pygpu_language_cuda.o $< -lpython${PYTHONVERSION} -lcublas -lcudart
-
-clean:
-	rm -f pygpu_ndarray.so core.* *.o *~
-	rm -rf build
-
-cleantmp:
-	rm -f core.* *.o *~