Merge branch 'main' into py/mld-ext

Author: penelopeysm

HISTORY.md

@@ -1,6 +1,6 @@
 # DynamicPPL Changelog
 
-## 0.37.3
+## 0.37.4
 
 An extension for MarginalLogDensities.jl has been added.
 
@@ -10,6 +10,10 @@ The `marginalize` function returns a `MarginalLogDensities.MarginalLogDensity`,
 By default, this uses the Laplace approximation and sparse AD, making the marginalization computationally very efficient.
 Please see [the documentation](https://turinglang.org/DynamicPPL.jl/v0.37/api/#Marginalization) for further information.
 
+## 0.37.3
+
+Prevents inlining of `DynamicPPL.istrans` with Enzyme, which allows Enzyme to differentiate models where `VarName`s have the same symbol but different types.
+
 ## 0.37.2
 
 Make the `resume_from` keyword work for multiple-chain (parallel) sampling as well.

Project.toml

@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.37.3"
+version = "0.37.4"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

ext/DynamicPPLEnzymeCoreExt.jl

@@ -12,7 +12,6 @@ end
 
 # Mark istrans as having 0 derivative. The `nothing` return value is not significant, Enzyme
 # only checks whether such a method exists, and never runs it.
-@inline EnzymeCore.EnzymeRules.inactive_noinl(::typeof(DynamicPPL.istrans), args...) =
-    nothing
+@inline EnzymeCore.EnzymeRules.inactive(::typeof(DynamicPPL.istrans), args...) = nothing
 
 end