fix: missing copyto!

EnzymeAD · Jan 7, 2025 · 5881c3e · 5881c3e
1 parent 3ec714d
commit 5881c3e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/TracedRArray.jl b/src/TracedRArray.jl
@@ -483,6 +483,10 @@ function Base.copyto!(dest::TracedRArray{T,N}, src::TracedRArray{T,N}) where {T,
     return dest
 end
 
+function Base.copyto!(dest::TracedRArray{T,N}, src::TracedRArray{T2,N}) where {T,N}
+    return copyto!(dest, Ops.convert(TracedRArray{T,N}, src))
+end
+
 function _copyto!(dest::AnyTracedRArray, bc::Broadcasted)
     axes(dest) == axes(bc) || Broadcast.throwdm(axes(dest), axes(bc))
     isempty(dest) && return dest