Don't use closure for enzyme (#1048)

wsmoses · penelopeysm · web-flow · commit fcdfe0354c8b · 2025-09-24T16:46:40.000+01:00
* Don't use closure for enzyme

* no annotation

* bump patch

---------

Co-authored-by: Penelope Yong &lt;penelopeysm@gmail.com&gt;
diff --git a/HISTORY.md b/HISTORY.md
@@ -1,5 +1,9 @@
 # DynamicPPL Changelog
 
+## 0.37.5
+
+A minor optimisation for Enzyme AD on DynamicPPL models.
+
 ## 0.37.4
 
 An extension for MarginalLogDensities.jl has been added.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.37.4"
+version = "0.37.5"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
diff --git a/src/logdensityfunction.jl b/src/logdensityfunction.jl
@@ -350,6 +350,7 @@ closure approach will be used. By default, this function returns `false`, i.e.
 the constant approach will be used.
 """
 use_closure(::ADTypes.AbstractADType) = true
+use_closure(::ADTypes.AutoEnzyme) = false
 
 """
     getmodel(f)
diff --git a/test/integration/enzyme/main.jl b/test/integration/enzyme/main.jl
@@ -6,10 +6,8 @@ import Enzyme: set_runtime_activity, Forward, Reverse, Const
 using ForwardDiff: ForwardDiff  # run_ad uses FD for correctness test
 
 ADTYPES = Dict(
-    "EnzymeForward" =>
-        AutoEnzyme(; mode=set_runtime_activity(Forward), function_annotation=Const),
-    "EnzymeReverse" =>
-        AutoEnzyme(; mode=set_runtime_activity(Reverse), function_annotation=Const),
+    "EnzymeForward" => AutoEnzyme(; mode=set_runtime_activity(Forward)),
+    "EnzymeReverse" => AutoEnzyme(; mode=set_runtime_activity(Reverse)),
 )
 
 @testset "$ad_key" for (ad_key, ad_type) in ADTYPES
