fixup! fix: fix view adjoints

SciML · May 30, 2024 · 97edb58 · 97edb58
1 parent fc037f9
commit 97edb58
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ext/RecursiveArrayToolsZygoteExt.jl b/ext/RecursiveArrayToolsZygoteExt.jl
@@ -115,7 +115,7 @@ end
     adj = let VA = VA
         function Array_adjoint(y)
             VA = recursivecopy(VA)
-            copyto!(VA, y)
+            VA .= y
             return (VA,)
         end
     end
@@ -126,7 +126,7 @@ end
     view_adjoint = let A = A, I = I
         function (y)
             A = recursivecopy(A)
-            copyto!(A, y)
+            A .= y
             (A, map(_ -> nothing, I)...)
         end
     end
@@ -138,7 +138,8 @@ end
         function (y)
             A = recursivecopy(A)
             recursivefill!(A, zero(eltype(A)))
-            A[I...] .= y
+            v = view(A, I...)
+            v .= y
             return (A, map(_ -> nothing, I)...)
         end
     end