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
2
2
using DynamicPPL. TestUtils. AD: run_ad
3
3
using ADTypes: AutoEnzyme
4
4
using Test: @test , @testset
5
- import Enzyme: set_runtime_activity, Forward, Reverse
5
+ import Enzyme: set_runtime_activity, Forward, Reverse, Const
6
6
using ForwardDiff: ForwardDiff # run_ad uses FD for correctness test
7
7
8
8
ADTYPES = 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),
11
13
)
12
14
13
15
@testset " $ad_key " for (ad_key, ad_type) in ADTYPES
You can’t perform that action at this time.
0 commit comments