File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -2,12 +2,14 @@ using DynamicPPL.TestUtils: DEMO_MODELS
22using DynamicPPL. TestUtils. AD: run_ad
33using ADTypes: AutoEnzyme
44using Test: @test , @testset
5- import Enzyme: set_runtime_activity, Forward, Reverse
5+ import Enzyme: set_runtime_activity, Forward, Reverse, Const
66using ForwardDiff: ForwardDiff # run_ad uses FD for correctness test
77
88ADTYPES = Dict (
9- " EnzymeForward" => AutoEnzyme (; mode= set_runtime_activity (Forward)),
10- " EnzymeReverse" => AutoEnzyme (; mode= set_runtime_activity (Reverse)),
9+ " EnzymeForward" =>
10+ AutoEnzyme (; mode= set_runtime_activity (Forward), function_annotation= Const),
11+ " EnzymeReverse" =>
12+ AutoEnzyme (; mode= set_runtime_activity (Reverse), function_annotation= Const),
1113)
1214
1315@testset " $ad_key " for (ad_key, ad_type) in ADTYPES
You can’t perform that action at this time.
0 commit comments