Dot operators of hcat failed for CuArray

General Usage
,
1

Hi.

I’m trying to combining elements with the same locations from two arrays. It’s easy to do that with dot hcat like this.

a = randn(5,2)
b = randn(5,2)
hcat.(a,b)

When I store my arrays in GPU as CuArray, it won’t work, and seems that element-wise procedure could be done in CPU only.

a = gpu(randn(5,2))
b = gpu(randn(5,2))
hcat.(a,b)

I was wondering whether I missed anything here and many thanks for any advices.

2

This makes a 5×2 Matrix{Matrix{Float64}}. I don’t think arrays of arrays work well on the GPU, can you use tuple.(a,b) or cat(a,b; dims=3) instead?

3

Many thanks for your suggestions and I just realized that arrays of arrays won’t work on the GPU.