Skip to content
Snippets Groups Projects
Commit f87f1661 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Fix any, all

parent 05a4fce4
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 0 deletions
pyopencl/reduction.py
+ 2
− 0
View file @ f87f1661
...@@ -440,6 +440,7 @@ class ReductionTemplate(KernelTemplateBase): ...@@ -440,6 +440,7 @@ class ReductionTemplate(KernelTemplateBase):
@context_dependent_memoize @context_dependent_memoize
def get_any_kernel(ctx, dtype_in): def get_any_kernel(ctx, dtype_in):
from pyopencl.tools import VectorArg
return ReductionKernel(ctx, np.int8, "false", "a || b", return ReductionKernel(ctx, np.int8, "false", "a || b",
map_expr="(bool) (in[i])", map_expr="(bool) (in[i])",
arguments=[VectorArg(dtype_in, "in")]) arguments=[VectorArg(dtype_in, "in")])
...@@ -447,6 +448,7 @@ def get_any_kernel(ctx, dtype_in): ...@@ -447,6 +448,7 @@ def get_any_kernel(ctx, dtype_in):
@context_dependent_memoize @context_dependent_memoize
def get_all_kernel(ctx, dtype_in): def get_all_kernel(ctx, dtype_in):
from pyopencl.tools import VectorArg
return ReductionKernel(ctx, np.int8, "true", "a && b", return ReductionKernel(ctx, np.int8, "true", "a && b",
map_expr="(bool) (in[i])", map_expr="(bool) (in[i])",
arguments=[VectorArg(dtype_in, "in")]) arguments=[VectorArg(dtype_in, "in")])
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment