Skip to content

Commit 97edb58

Browse files
fixup! fix: fix view adjoints
1 parent fc037f9 commit 97edb58

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

ext/RecursiveArrayToolsZygoteExt.jl

+4-3
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ end
115115
adj = let VA = VA
116116
function Array_adjoint(y)
117117
VA = recursivecopy(VA)
118-
copyto!(VA, y)
118+
VA .= y
119119
return (VA,)
120120
end
121121
end
@@ -126,7 +126,7 @@ end
126126
view_adjoint = let A = A, I = I
127127
function (y)
128128
A = recursivecopy(A)
129-
copyto!(A, y)
129+
A .= y
130130
(A, map(_ -> nothing, I)...)
131131
end
132132
end
@@ -138,7 +138,8 @@ end
138138
function (y)
139139
A = recursivecopy(A)
140140
recursivefill!(A, zero(eltype(A)))
141-
A[I...] .= y
141+
v = view(A, I...)
142+
v .= y
142143
return (A, map(_ -> nothing, I)...)
143144
end
144145
end

0 commit comments

Comments
 (0)