Skip to content

Commit af77697

Browse files
committed
ok, ok these methods were needed
1 parent 3069863 commit af77697

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/mapreduce.jl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55

66
Base.any(A::GPUArray{Bool}) = mapreduce(identity, |, false, A)
77
Base.all(A::GPUArray{Bool}) = mapreduce(identity, &, true, A)
8-
Base.count(pred, A::GPUArray) = Int(mapreduce(pred, +, 0, A))
8+
9+
Base.any(f::Function, A::GPUArray) = mapreduce(f, |, false, A)
10+
Base.all(f::Function, A::GPUArray) = mapreduce(f, &, true, A)
11+
Base.count(pred::Function, A::GPUArray) = Int(mapreduce(pred, +, 0, A))
912

1013
Base.:(==)(A::GPUArray, B::GPUArray) = Bool(mapreduce(==, &, true, A, B))
1114

0 commit comments

Comments
 (0)