Merge pull request #183 from ma-sadeghi/fix-mask-indexing

Fix boolean mask indexing

Author: ChrisRackauckas

src/vector_of_array.jl

@@ -78,7 +78,7 @@ Base.@propagate_inbounds function Base.getindex(A::AbstractVectorOfArray{T, N},
   @assert length(J) == ndims(A.u[1])+1-ndims(I)
   @assert size(I) == size(A)[1:ndims(A)-length(J)]
   vecs = vec.(A.u)
-  return Adapt.adapt(__parameterless_type(T),reshape(reduce(hcat,vecs[I]),size(A.u[1])...,length(A.u)))
+  return A[repeat([:], length(size(A)))...][I,J...]
 end
 
 Base.@propagate_inbounds Base.getindex(A::AbstractDiffEqArray{T, N}, i::Int,::Colon) where {T, N} = [A.u[j][i] for j in 1:length(A)]

test/gpu/vectorofarray_gpu.jl

@@ -18,9 +18,7 @@ x[m, :]
 
 va = VectorOfArray([slice for slice in eachslice(x, dims=3)])
 @test va[m, :] ≈ x[m, :]
-va[m, :]
 
 xc = Array(x)
 mc = Array(m)
-xc[mc, :]
-
+@test xc[mc, :] ≈ Array(va[m, :])