feat: add an overload of Base.Tuple

EnzymeAD · Nov 9, 2024 · 264d530 · 264d530
1 parent 2858def
commit 264d530
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/TracedRArray.jl b/src/TracedRArray.jl
@@ -155,6 +155,8 @@ function Base.setindex!(
     return a
 end
 
+Base.Tuple(x::TracedRArray) = ntuple(Base.Fix1(Base.getindex, x), length(x))
+
 Base.size(x::TracedRArray) = x.shape
 
 Base.copy(A::TracedRArray{T,N}) where {T,N} = TracedRArray{T,N}((), A.mlir_data, size(A))