Skip to content

Commit 6340a2f

Browse files
add convert handling
1 parent b309b49 commit 6340a2f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/init.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,5 +16,7 @@ function __init__()
1616
vecs = vec.(VA.u)
1717
return CuArrays.CuArray(reshape(reduce(hcat,vecs),size(VA.u[1])...,length(VA.u)))
1818
end
19+
20+
Base.convert(::Type{<:CuArrays.CuArray},VA::AbstractVectorOfArray) = CuArrays.CuArray(VA)
1921
end
2022
end

0 commit comments

Comments
 (0)