From f87f1661e4d914e163beb837144531ad540aedf5 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 19 Jul 2013 13:24:54 -0400
Subject: [PATCH] Fix any, all
---
pyopencl/reduction.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/pyopencl/reduction.py b/pyopencl/reduction.py
index 9a105d1d..f342909f 100644
--- a/pyopencl/reduction.py
+++ b/pyopencl/reduction.py
@@ -440,6 +440,7 @@ class ReductionTemplate(KernelTemplateBase):
@context_dependent_memoize
def get_any_kernel(ctx, dtype_in):
+ from pyopencl.tools import VectorArg
return ReductionKernel(ctx, np.int8, "false", "a || b",
map_expr="(bool) (in[i])",
arguments=[VectorArg(dtype_in, "in")])
@@ -447,6 +448,7 @@ def get_any_kernel(ctx, dtype_in):
@context_dependent_memoize
def get_all_kernel(ctx, dtype_in):
+ from pyopencl.tools import VectorArg
return ReductionKernel(ctx, np.int8, "true", "a && b",
map_expr="(bool) (in[i])",
arguments=[VectorArg(dtype_in, "in")])
--
GitLab