diff --git a/test/Project.toml b/test/Project.toml index c7583c672..2595633ab 100644 --- a/test/Project.toml +++ b/test/Project.toml @@ -11,6 +11,7 @@ Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b" Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f" DistributionsAD = "ced4e74d-a319-5a8a-b0ac-84af2272839c" Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4" +Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9" EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869" ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210" JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b" diff --git a/test/ad.jl b/test/ad.jl index 17981cf2a..976566deb 100644 --- a/test/ad.jl +++ b/test/ad.jl @@ -1,4 +1,6 @@ -@testset "AD: ForwardDiff, ReverseDiff, and Mooncake" begin +using Enzyme + +@testset "AD: ForwardDiff, ReverseDiff, Mooncake, and Enzyme" begin @testset "$(m.f)" for m in DynamicPPL.TestUtils.DEMO_MODELS f = DynamicPPL.LogDensityFunction(m) rand_param_values = DynamicPPL.TestUtils.rand_prior_true(m) @@ -21,6 +23,8 @@ ADTypes.AutoReverseDiff(; compile=false), ADTypes.AutoReverseDiff(; compile=true), ADTypes.AutoMooncake(; config=nothing), + ADTypes.AutoEnzyme(; mode=Enzyme.Forward), + ADTypes.AutoEnzyme(; mode=Enzyme.Reverse), ] # Mooncake can't currently handle something that is going on in # SimpleVarInfo{<:VarNamedVector}. Disable all SimpleVarInfo tests for now.