feat(typing): return int from kernel_get_work_group_info

Author: alexfikl

pyopencl/_monkeypatch.py

@@ -259,7 +259,7 @@ 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):
+def kernel_get_work_group_info(self: _cl.Kernel, param: int, device: _cl.Device) -> int:
     try:
         wg_info_cache = self._wg_info_cache
     except AttributeError:

pyopencl/array.py

@@ -244,9 +244,9 @@ def kernel_runner(out: Array, *args: P.args, **kwargs: P.kwargs) -> cl.Event:
         assert queue is not None
 
         knl = kernel_getter(out, *args, **kwargs)
-        work_group_info = cast("int", knl.get_work_group_info(
+        work_group_info = knl.get_work_group_info(
             cl.kernel_work_group_info.WORK_GROUP_SIZE,
-            queue.device))
+            queue.device)
         gs, ls = out._get_sizes(queue, work_group_info)
 
         knl_args = (out, *args, out.size)

pyopencl/characterize/__init__.py

@@ -24,8 +24,6 @@
 """
 
 
-from typing import cast
-
 from pytools import memoize
 
 import pyopencl as cl
@@ -70,9 +68,9 @@ def reasonable_work_group_size_multiple(
         }
         """)
     prg.build()
-    return cast("int", prg.knl.get_work_group_info(
+    return prg.knl.get_work_group_info(
             cl.kernel_work_group_info.PREFERRED_WORK_GROUP_SIZE_MULTIPLE,
-            dev))
+            dev)
 
 
 def nv_compute_capability(dev: cl.Device):