We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0548ddf commit ad5b573Copy full SHA for ad5b573
test/ad.jl
@@ -1,4 +1,6 @@
1
-@testset "AD: ForwardDiff, ReverseDiff, and Mooncake" begin
+using Enzyme
2
+
3
+@testset "AD: ForwardDiff, ReverseDiff, Mooncake, and Enzyme" begin
4
@testset "$(m.f)" for m in DynamicPPL.TestUtils.DEMO_MODELS
5
f = DynamicPPL.LogDensityFunction(m)
6
rand_param_values = DynamicPPL.TestUtils.rand_prior_true(m)
@@ -21,6 +23,8 @@
21
23
ADTypes.AutoReverseDiff(; compile=false),
22
24
ADTypes.AutoReverseDiff(; compile=true),
25
ADTypes.AutoMooncake(; config=nothing),
26
+ ADTypes.AutoEnzyme(; mode=Enzyme.Forward),
27
+ ADTypes.AutoEnzyme(; mode=Enzyme.Reverse),
28
]
29
# Mooncake can't currently handle something that is going on in
30
# SimpleVarInfo{<:VarNamedVector}. Disable all SimpleVarInfo tests for now.
0 commit comments