Merge pull request #108 from SciML/ChrisRackauckas-patch-1

CUDA.jl compat

Author: ChrisRackauckas

src/init.jl

@@ -19,6 +19,15 @@ function __init__()
     Base.convert(::Type{<:CuArrays.CuArray},VA::AbstractVectorOfArray) = CuArrays.CuArray(VA)
     @adjoint CuArrays.CuArray(xs::AbstractVectorOfArray) = CuArrays.CuArray(xs), ȳ -> (ȳ,)
   end
+  
+  @require CUDA="052768ef-5323-5732-b1bb-66c8b64840ba" begin
+    function CUDA.CuArray(VA::AbstractVectorOfArray)
+      vecs = vec.(VA.u)
+      return CUDA.CuArray(reshape(reduce(hcat,vecs),size(VA.u[1])...,length(VA.u)))
+    end
+    Base.convert(::Type{<:CUDA.CuArray},VA::AbstractVectorOfArray) = CUDA.CuArray(VA)
+    @adjoint CUDA.CuArray(xs::AbstractVectorOfArray) = CUDA.CuArray(xs), ȳ -> (ȳ,)
+  end
 
   @require Tracker="9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c" begin
     function recursivecopy!(b::AbstractArray{T,N},a::AbstractArray{T2,N}) where {T<:Tracker.TrackedArray,T2<:Tracker.TrackedArray,N}