Update src/TracedRArray.jl

EnzymeAD · Dec 12, 2024 · 602dc6a · 602dc6a
1 parent a98911c
commit 602dc6a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/TracedRArray.jl b/src/TracedRArray.jl
@@ -728,6 +728,9 @@ for (minT, maxT) in Iterators.product((Number, TracedRNumber), (Number, TracedRN
     end
 end
 
+Base.all(f::Function, x::AnyTracedRArray) = mapreduce(f, &, x)
+Base.any(f::Function, x::AnyTracedRArray) = mapreduce(f, |, x)
+
 # outer repeat
 function Base.repeat(x::AnyTracedRArray{T,N}, counts::Vararg{Int,M}) where {T,N,M}
     P = max(N, M) # potentially padded