Add Enzyme as an automatic differentiation backend

wsmoses · web-flow · commit 54f94b839331 · 2025-09-13T22:37:48.000-05:00
diff --git a/benchmarks/src/DynamicPPLBenchmarks.jl b/benchmarks/src/DynamicPPLBenchmarks.jl
@@ -37,6 +37,7 @@ const SYMBOL_TO_BACKEND = Dict(
     :reversediff => ADTypes.AutoReverseDiff(; compile=false),
     :reversediff_compiled => ADTypes.AutoReverseDiff(; compile=true),
     :mooncake => ADTypes.AutoMooncake(; config=nothing),
+    :enzyme => ADTypes.AutoEnzyme(; config=nothing),
 )
 
 to_backend(x) = error("Unknown backend: $x")

Original file line number	Diff line number	Diff line change
`@@ -37,6 +37,7 @@ const SYMBOL_TO_BACKEND = Dict(`
`37`	`37`	`:reversediff => ADTypes.AutoReverseDiff(; compile=false),`
`38`	`38`	`:reversediff_compiled => ADTypes.AutoReverseDiff(; compile=true),`
`39`	`39`	`:mooncake => ADTypes.AutoMooncake(; config=nothing),`
	`40`	`+ :enzyme => ADTypes.AutoEnzyme(; config=nothing),`
`40`	`41`	`)`
`41`	`42`
`42`	`43`	`to_backend(x) = error("Unknown backend: $x")`