Skip to content

Commit 751a340

Browse files
authored
Merge branch 'breaking' into mhauru/accumulators-stage2
2 parents e5af163 + 8b67e96 commit 751a340

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

test/integration/enzyme/main.jl

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,14 @@ using DynamicPPL.TestUtils: DEMO_MODELS
22
using DynamicPPL.TestUtils.AD: run_ad
33
using ADTypes: AutoEnzyme
44
using Test: @test, @testset
5-
import Enzyme: set_runtime_activity, Forward, Reverse
5+
import Enzyme: set_runtime_activity, Forward, Reverse, Const
66
using ForwardDiff: ForwardDiff # run_ad uses FD for correctness test
77

88
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),
1113
)
1214

1315
@testset "$ad_key" for (ad_key, ad_type) in ADTYPES

0 commit comments

Comments
 (0)