Skip to content

Commit cc55ee6

Browse files
fix: fix Array(::AbstractVectorOfArray) adjoint
1 parent f226441 commit cc55ee6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

ext/RecursiveArrayToolsZygoteExt.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ end
110110
@adjoint function Base.Array(VA::AbstractVectorOfArray)
111111
adj = let VA=VA
112112
function Array_adjoint(y)
113-
VA = copy(VA)
113+
VA = recursivecopy(VA)
114114
copyto!(VA, y)
115115
return (VA,)
116116
end

src/utils.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ end
2828

2929
function recursivecopy(a::AbstractVectorOfArray)
3030
b = copy(a)
31-
b.u = recursivecopy.(a.u)
31+
b.u .= recursivecopy.(a.u)
3232
return b
3333
end
3434

0 commit comments

Comments
 (0)