ext

Author: wsmoses

Project.toml

@@ -21,8 +21,10 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 
 [weakdeps]
+ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 BFloat16s = "ab4f0b2a-ad5b-11e8-123f-65d77653426b"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+DynamicPPL = "366bfd00-2699-11ea-058f-f148b4cae6d8"
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
@@ -31,6 +33,7 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 [extensions]
 EnzymeBFloat16sExt = "BFloat16s"
 EnzymeChainRulesCoreExt = "ChainRulesCore"
+EnzymeDynamicPPLExt = ["ADTypes", "DynamicPPL"]
 EnzymeGPUArraysCoreExt = "GPUArraysCore"
 EnzymeLogExpFunctionsExt = "LogExpFunctions"
 EnzymeSpecialFunctionsExt = "SpecialFunctions"
@@ -40,6 +43,7 @@ EnzymeStaticArraysExt = "StaticArrays"
 BFloat16s = "0.2, 0.3, 0.4, 0.5"
 CEnum = "0.4, 0.5"
 ChainRulesCore = "1"
+DynamicPPL = "0.35, 0.36, 0.37"
 EnzymeCore = "0.8.13"
 Enzyme_jll = "0.0.199"
 GPUArraysCore = "0.1.6, 0.2"
@@ -55,8 +59,10 @@ StaticArrays = "1"
 julia = "1.10"
 
 [extras]
+ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 BFloat16s = "ab4f0b2a-ad5b-11e8-123f-65d77653426b"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+DynamicPPL = "366bfd00-2699-11ea-058f-f148b4cae6d8"
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"

ext/EnzymeDynamicPPLExt.jl

@@ -0,0 +1,10 @@
+module EnzymeDynamicPPLExt
+
+using ADTypes
+using DynamicPPL
+
+@static if isdefined(DynamicPPL, :is_supported)
+  DynamicPPL.is_supported(::ADTypes.AutoEnzyme) = true
+end
+
+end # module