Do not use internal functions from Distributions (#208)

Author: devmotion

Project.toml

@@ -1,6 +1,6 @@
 name = "DistributionsAD"
 uuid = "ced4e74d-a319-5a8a-b0ac-84af2272839c"
-version = "0.6.33"
+version = "0.6.34"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/multivariate.jl

@@ -45,15 +45,15 @@ function Distributions._rand!(
     @inbounds for (i, αi) in zip(eachindex(x), d.alpha)
         x[i] = rand(rng, Gamma(αi))
     end
-    Distributions.multiply!(x, inv(sum(x))) # this returns x
+    lmul!(inv(sum(x)), x) # this returns x
 end
 function Distributions._rand!(
     rng::AbstractRNG,
     d::TuringDirichlet{<:Real,<:FillArrays.AbstractFill},
     x::AbstractVector{<:Real}
 )
     rand!(rng, Gamma(FillArrays.getindex_value(d.alpha)), x)
-    Distributions.multiply!(x, inv(sum(x))) # this returns x
+    lmul!(inv(sum(x)), x) # this returns x
 end
 
 function Distributions._logpdf(d::TuringDirichlet, x::AbstractVector{<:Real})
@@ -246,10 +246,14 @@ MvLogNormal(d::TuringDiagMvNormal) = TuringMvLogNormal(d)
 MvLogNormal(d::TuringScalMvNormal) = TuringMvLogNormal(d)
 Distributions.length(d::TuringMvLogNormal) = length(d.normal)
 function Distributions.rand(rng::Random.AbstractRNG, d::TuringMvLogNormal)
-    return Distributions.exp!(rand(rng, d.normal))
+    x = rand(rng, d.normal)
+    map!(exp, x, x)
+    return x
 end
 function Distributions.rand(rng::Random.AbstractRNG, d::TuringMvLogNormal, n::Int)
-    return Distributions.exp!(rand(rng, d.normal, n))
+    x = rand(rng, d.normal, n)
+    map!(exp, x, x)
+    return x
 end
 
 function Distributions._logpdf(d::TuringMvLogNormal, x::AbstractVector{T}) where {T<:Real}