build: add SciMLSensitivity tests to downstream CI

.github/workflows/Downstream.yml

@@ -25,6 +25,11 @@ jobs:
           - {user: SciML, repo: OrdinaryDiffEq.jl, group: Core}
           - {user: SciML, repo: OrdinaryDiffEq.jl, group: Interface}
           - {user: SciML, repo: DelayDiffEq.jl, group: Interface}
+          - {user: SciML, repo: SciMLSensitivity.jl, group: Core1}
+          - {user: SciML, repo: SciMLSensitivity.jl, group: Core2}
+          - {user: SciML, repo: SciMLSensitivity.jl, group: Core3}
+          - {user: SciML, repo: SciMLSensitivity.jl, group: Core4}
+          - {user: SciML, repo: SciMLSensitivity.jl, group: Core5}
     steps:
       - uses: actions/checkout@v4
       - uses: julia-actions/setup-julia@v1

ext/RecursiveArrayToolsZygoteExt.jl

@@ -110,23 +110,29 @@
 @adjoint function Base.Array(VA::AbstractVectorOfArray)
     adj = let VA=VA
         function Array_adjoint(y)
-            VA = copy(VA)
+            VA = recursivecopy(VA)
             copyto!(VA, y)
             return (VA,)
         end
     end
     Array(VA), adj
 end
 
+@adjoint function Base.view(A::AbstractVectorOfArray, I::Colon...)
+    function adjoint(y)
+        (recursivecopy(parent(y)), map(_ -> nothing, I)...)
+    end
+    return view(A, I...), adjoint
+end
+
 @adjoint function Base.view(A::AbstractVectorOfArray, I...)
-    adj = let A = A, I = I
-        function view_adjoint(y)
-            A = zero(A)
-            view(A, I...) .= y
-            return (A, map(_ -> nothing, I)...)
-        end
+    function view_adjoint(y)
+        A = recursivecopy(parent(y))
+        recursivefill!(A, zero(eltype(A)))
+        A[I...] .= y
+        return (A, map(_ -> nothing, I)...)
     end
-    view(A, I...), adj
+    view(A, I...), view_adjoint
 end
 
 ChainRulesCore.ProjectTo(a::AbstractVectorOfArray) = ChainRulesCore.ProjectTo{VectorOfArray}((sz = size(a)))

src/utils.jl

@@ -28,7 +28,7 @@
 
 function recursivecopy(a::AbstractVectorOfArray)
     b = copy(a)
-    b.u = recursivecopy.(a.u)
+    b.u .= recursivecopy.(a.u)
     return b
 end