Skip to content

Commit 79caf8d

Browse files
fixup! fix: fix view adjoints
1 parent 5770229 commit 79caf8d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ext/RecursiveArrayToolsZygoteExt.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -126,8 +126,8 @@ end
126126
view_adjoint = let A = A, I = I
127127
function (y)
128128
A = recursivecopy(A)
129-
A .= y
130-
(A, map(_ -> nothing, I)...)
129+
copyto!(A, y)
130+
return (A, map(_ -> nothing, I)...)
131131
end
132132
end
133133
return view(A, I...), view_adjoint
@@ -139,7 +139,7 @@ end
139139
A = recursivecopy(A)
140140
recursivefill!(A, zero(eltype(A)))
141141
v = view(A, I...)
142-
v .= y
142+
copyto!(v, y)
143143
return (A, map(_ -> nothing, I)...)
144144
end
145145
end

0 commit comments

Comments
 (0)