Merge pull request #328 from abergeron/reduce_f16

Raise an error for float16 in ReductionKernel.

Author: nouiz

pygpu/reduction.py

@@ -150,6 +150,11 @@ def __init__(self, context, dtype_out, neutral, reduce_expr, redux,
         else:
             self.arguments = arguments
 
+        if (self.dtype_out == numpy.dtype('float16') or
+                any(ar.dtype == numpy.dtype('float16')
+                    for ar in self.arguments)):
+            raise NotImplementedError('float16 not supported for the reduction interface')
+
         self.reduce_expr = reduce_expr
         if map_expr is None:
             if len(self.arguments) != 1:

pygpu/tests/test_reduction.py

@@ -1,5 +1,7 @@
 import numpy
 
+from nose.tools import assert_raises
+
 from pygpu import gpuarray, ndgpuarray as elemary
 from pygpu.reduction import ReductionKernel
 
@@ -130,3 +132,9 @@ def test_reduction_0d():
     rg = g.all()
 
     assert numpy.all(rc == numpy.asarray(rg))
+
+
+def test_reduction_f16():
+    c, g = gen_gpuarray((3,), dtype='float16', ctx=context, cls=elemary)
+
+    assert_raises(NotImplementedError, g.sum)