CuArray
is the host-representation of a CUDA array, you need CuDeviceArray
. That conversion happens automatically for known types, but you need to register your own Adapt.jl conversions for custom types.
2 Likes