add copyto of subarraay (#1405)

* add copyto of subarraay

* fix

* fix

* fix

* Update src/TracedRArray.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

Author: wsmoses

src/TracedRArray.jl

@@ -726,15 +726,11 @@ end
 
 Base.copyto!(dest::AnyTracedRArray, bc::Broadcasted{Nothing}) = _copyto!(dest, bc) # Keep it for ArrayConflict
 
-function Base.copyto!(dest::TracedRArray{T,N}, src::TracedRArray{T,N}) where {T,N}
-    dest.mlir_data = src.mlir_data
+function Base.copyto!(dest::AnyTracedRArray{T,N}, src::TracedRArray{T,N}) where {T,N}
+    TracedUtils.set_mlir_data!(dest, src.mlir_data)
     return dest
 end
 
-function Base.copyto!(dest::TracedRArray, src::AnyTracedRArray)
-    return copyto!(dest, materialize_traced_array(src))
-end
-
 function Base.copyto!(
     dest::Reactant.TracedRArray{T},
     dstart::Integer,
@@ -747,14 +743,22 @@ function Base.copyto!(
 end
 
 function Base.copyto!(dest::TracedRArray{T,N}, src::TracedRArray{T2,N}) where {T,T2,N}
-    return copyto!(dest, Ops.convert(TracedRArray{T,N}, src))
+    src2 = if T != T2
+        Ops.convert(TracedRArray{T,N}, src)
+    else
+        src
+    end
+    TracedUtils.set_mlir_data!(dest, src2.mlir_data)
+    return dest
 end
 
-function Base.copyto!(dest::AnyTracedRArray, src::AnyTracedRArray)
+function Base.copyto!(
+    dest::AnyTracedRArray{T1,N} where {T1}, src::AnyTracedRArray{T2,N} where {T2}
+) where {N}
     return copyto!(dest, materialize_traced_array(src))
 end
 
-function Base.copyto!(dest::TracedRArray{T,N}, src::Array{T2,N}) where {T,T2,N}
+function Base.copyto!(dest::AnyTracedRArray{T,N}, src::Array{T2,N}) where {T,T2,N}
     return copyto!(dest, TracedUtils.promote_to(TracedRArray{T2,N}, src))
 end