fixup! fix: fix view adjoints

SciML · May 30, 2024 · 5770229 · 5770229
1 parent 97edb58
commit 5770229
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/ext/RecursiveArrayToolsReverseDiffExt.jl b/ext/RecursiveArrayToolsReverseDiffExt.jl
@@ -23,4 +23,16 @@ end
     end
     return Array(VA), Array_adjoint
 end
+
+@adjoint function Base.view(A::AbstractVectorOfArray{<:ReverseDiff.TrackedReal, N}, I::Colon...) where {N}
+    view_adjoint = let A = A, I = I
+        function (y)
+            A = recursivecopy(A)
+            trackedarraycopyto!(A, y)
+            (A, map(_ -> nothing, I)...)
+        end
+    end
+    return view(A, I...), view_adjoint
+end
+
 end # module