add comment in code about restrictions on supported dtypes
@@ -2211,6 +2211,8 @@ cpdef _scatter_op_single(ndarray a, ndarray indices, v, int axis=0, op=''):
_scatter_update_kernel(
v, indices, cdim, rdim, adim, a.reduced_view())
elif op == 'add':
+ # There is constraints on types because atomicAdd() in CUDA 7.5
+ # only supports int32, uint32, uint64, and float32.
if not issubclass(v.dtype.type,
(numpy.int32, numpy.float32,
numpy.uint32, numpy.uint64, numpy.ulonglong)):
