Skip to content

Commit ad5b573

Browse files
authored
Add Enzyme testing
1 parent 0548ddf commit ad5b573

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

test/ad.jl

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
1-
@testset "AD: ForwardDiff, ReverseDiff, and Mooncake" begin
1+
using Enzyme
2+
3+
@testset "AD: ForwardDiff, ReverseDiff, Mooncake, and Enzyme" begin
24
@testset "$(m.f)" for m in DynamicPPL.TestUtils.DEMO_MODELS
35
f = DynamicPPL.LogDensityFunction(m)
46
rand_param_values = DynamicPPL.TestUtils.rand_prior_true(m)
@@ -21,6 +23,8 @@
2123
ADTypes.AutoReverseDiff(; compile=false),
2224
ADTypes.AutoReverseDiff(; compile=true),
2325
ADTypes.AutoMooncake(; config=nothing),
26+
ADTypes.AutoEnzyme(; mode=Enzyme.Forward),
27+
ADTypes.AutoEnzyme(; mode=Enzyme.Reverse),
2428
]
2529
# Mooncake can't currently handle something that is going on in
2630
# SimpleVarInfo{<:VarNamedVector}. Disable all SimpleVarInfo tests for now.

0 commit comments

Comments
 (0)