Merge pull request #318 from AayushSabharwal/as/gpu

fix: GPU tests, CuArray conversion, autodiff

Author: ChrisRackauckas

ext/RecursiveArrayToolsZygoteExt.jl

@@ -95,17 +95,17 @@ end
     VectorOfArray(u),
     y -> begin
         y isa Ref && (y = VectorOfArray(y[].u))
-        (VectorOfArray([y[ntuple(x -> Colon(), ndims(y.u) - 1)..., i]
-                        for i in 1:size(y.u)[end]]),)
+        (VectorOfArray([y[ntuple(x -> Colon(), ndims(y) - 1)..., i]
+                        for i in 1:size(y)[end]]),)
     end
 end
 
 @adjoint function DiffEqArray(u, t)
     DiffEqArray(u, t),
     y -> begin
         y isa Ref && (y = VectorOfArray(y[].u))
-        (DiffEqArray([y[ntuple(x -> Colon(), ndims(y.u) - 1)..., i]
-                       for i in 1:size(y.u)[end]],
+        (DiffEqArray([y[ntuple(x -> Colon(), ndims(y) - 1)..., i]
+                       for i in 1:size(y)[end]],
             t), nothing)
     end
 end

src/RecursiveArrayTools.jl

@@ -28,6 +28,7 @@ end
 
 import GPUArraysCore
 Base.convert(T::Type{<:GPUArraysCore.AnyGPUArray}, VA::AbstractVectorOfArray) = stack(VA.u)
+(T::Type{<:GPUArraysCore.AnyGPUArray})(VA::AbstractVectorOfArray) = T(Array(VA))
 
 import Requires
 @static if !isdefined(Base, :get_extension)