Mark istrans as inactive (#658)

* Mark istrans as inactive * Update DynamicPPLEnzymeCoreExt.jl * Update DynamicPPLEnzymeCoreExt.jl * Expand Enzyme extension comment --------- Co-authored-by: Markus Hauru <[email protected]>
TuringLang · Sep 12, 2024 · ad88334 · ad88334
1 parent bf73fd0
commit ad88334
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ext/DynamicPPLEnzymeCoreExt.jl b/ext/DynamicPPLEnzymeCoreExt.jl
@@ -10,4 +10,9 @@ end
 
 @inline EnzymeCore.EnzymeRules.inactive_type(::Type{<:DynamicPPL.SamplingContext}) = true
 
+# 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
+
 end