eltype instability

Author: matbesancon

src/multivariate/dirichlet.jl

@@ -400,15 +400,15 @@ end
 function ChainRulesCore.frule((_, Δd, Δx), ::typeof(_logpdf), d::Dirichlet, x::AbstractVector{<:Real})
     lp = _logpdf(d, x)
     if !insupport(d, x)
-        return (lp, zero(lp))
+        return (lp, zero(lp) + zero(eltype(Δx)) + zero(eltype(Δd.alpha)) + zero(eltype(Δd.lmnB)))
     end
     ∂α = sum(Δd.alpha[i] * log(x[i]) for i in eachindex(x))
     ∂l = - Δd.lmnB
     ∂x = sum((d.alpha[i] - 1) * Δx[i] / x[i] for i in eachindex(x))
     return (lp, ∂α + ∂l + ∂x)
 end
 
-function ChainRulesCore.rrule(::typeof(_logpdf), d::Dirichlet, x::AbstractVector{T}) where {T}
+function ChainRulesCore.rrule(::typeof(_logpdf), d::Dirichlet, x::AbstractVector{T}) where {T <: Real}
     y = _logpdf(d, x)
     function Dirichlet_logpdf_pullback(dy)
         if !isfinite(y)