Try another mask for GPU overload - #182

@ChrisRackauckas It seems that GPUArrays still complains about vec.(A.u[I]). The only workaround that I can think of is:

vecs = vec.(A.u)
return Base.getindex(Adapt.adapt(__parameterless_type(T),reduce(hcat,vecs),size(A.u[1])...,length(A.u))),
                     I, J...)

which is not elegant...

I guess so. I don't know why it does that but that's not an awful solution so let's do it.

seems to still not work?

@ChrisRackauckas Yeah, we were missing the reshape bit. But, actually if you replace the return line with:

return A[repeat([:], length(size(A)))...][I,J...]

It's shorter and easier to read. I tested it locally and it works. Wanted to commit, but didn't have permission.

Open a new PR.