device movement behavior inconsistent

[chengchingwen]

julia> using Flux, CUDA

julia> x = randn(5); x2 = (x, x); cx2 = gpu(x2);

julia> cx2[1] === cx2[2] # Flux v0.14.24
false

julia> cx2[1] === cx2[2] # Flux v0.14.20
true

It would cause problems for weight sharing. Probably due to #2502

[CarloLucibello]

This issue is due to MLDataDevices specializing on tuples and not keeping an IdDict for those. It should be fixed.

For functor types instead it works as expected:

julia> struct A; x; y; end

julia> Functors.@functor A

julia> a = A(x, x);

julia> ca = gpu(a);

julia> ca.x === ca.y # Flux v0.14.24
true

[CarloLucibello]

this has been fixed in MLDataDevices