Reduction fixes

inducer · May 22, 2016 · 173f3e0 · 173f3e0
1 parent eca4cfb
commit 173f3e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyopencl/reduction.py b/pyopencl/reduction.py
@@ -231,7 +231,8 @@ def get_reduction_kernel(stage,
     inf.kernel.set_scalar_arg_dtypes(
             [None, np.int64]
             + get_arg_list_scalar_arg_dtypes(inf.arg_types)
-            + [np.int64]*5)
+            + [np.int64]*3 + [np.uint32, np.int64]
+            )
 
     return inf
 

diff --git a/test/test_algorithm.py b/test/test_algorithm.py
@@ -271,7 +271,7 @@ def test_sum_without_data(ctx_factory):
             reduce_expr="a+b", map_expr="i",
             arguments=[])
 
-    result_dev = red(range=range(n), queue=queue).get()
+    result_dev = red(range=slice(n), queue=queue).get()
     result_ref = n*(n-1)//2
 
     assert result_dev == result_ref