Broken gpu macro #72
Description
With the latest changes in ConditionalDists, the gpu macro is broken:
model = VAE(ldim, CMeanVarGaussian{DiagVar}(Dense(3,4)), CMeanVarGaussian{ScalarVar}(Dense(2,4))) |> gpu
ERROR: MethodError: Cannot `convert` an object of type Dense{typeof(identity),CuArray{Float32,2,Nothing},CuArray{Float32,1,Nothing}} to an object of type Dense{typeof(identity),Array{Float32,2},Array{Float32,1}}
Closest candidates are:
convert(::Type{T}, ::T) where T at essentials.jl:168
Dense{typeof(identity),Array{Float32,2},Array{Float32,1}}(::Any, ::Any, ::Any) where {F, S, T} at /home/vit/.julia/packages/Flux/oObnA/src/layers/basic.jl:86
Stacktrace:
[1] CMeanVarGaussian{DiagVar,Dense{typeof(identity),Array{Float32,2},Array{Float32,1}}}(::Dense{typeof(identity),CuArray{Float32,2,Nothing},CuArray{Float32,1,Nothing}}) at /home/vit/.julia/dev/ConditionalDists/src/cmeanvar_gaussian.jl:34
...