fix: fix Array(::AbstractVectorOfArray) adjoint

SciML · Jan 19, 2024 · cc55ee6 · cc55ee6
1 parent f226441
commit cc55ee6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ext/RecursiveArrayToolsZygoteExt.jl b/ext/RecursiveArrayToolsZygoteExt.jl
@@ -110,7 +110,7 @@ end
 @adjoint function Base.Array(VA::AbstractVectorOfArray)
     adj = let VA=VA
         function Array_adjoint(y)
-            VA = copy(VA)
+            VA = recursivecopy(VA)
             copyto!(VA, y)
             return (VA,)
         end

diff --git a/src/utils.jl b/src/utils.jl
@@ -28,7 +28,7 @@ end
 
 function recursivecopy(a::AbstractVectorOfArray)
     b = copy(a)
-    b.u = recursivecopy.(a.u)
+    b.u .= recursivecopy.(a.u)
     return b
 end