ok, ok these methods were needed

JuliaGPU · Aug 24, 2018 · ee75d66 · ee75d66
1 parent 682b40a
commit ee75d66
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/mapreduce.jl b/src/mapreduce.jl
@@ -5,7 +5,10 @@
 
 Base.any(A::GPUArray{Bool}) = mapreduce(identity, |, false, A)
 Base.all(A::GPUArray{Bool}) = mapreduce(identity, &, true, A)
-Base.count(pred, A::GPUArray) = Int(mapreduce(pred, +, 0, A))
+
+Base.any(f::Function, A::GPUArray) = mapreduce(f, |, false, A)
+Base.all(f::Function, A::GPUArray) = mapreduce(f, &, true, A)
+Base.count(pred::Function, A::GPUArray) = Int(mapreduce(pred, +, 0, A))
 
 Base.:(==)(A::GPUArray, B::GPUArray) = Bool(mapreduce(==, &, true, A, B))