Skip to content

Commit 3ad7fa6

Browse files
authored
test: add runtime activity to enzyme (#159)
* test: add runtime activity to enzyme * Runtime activity in recommended backend
1 parent fd12ae2 commit 3ad7fa6

2 files changed

Lines changed: 9 additions & 3 deletions

File tree

ext/ImplicitDifferentiationEnzymeExt.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,9 @@ using Enzyme
55
using Enzyme.EnzymeRules
66
using ImplicitDifferentiation: ImplicitFunction, build_A, build_B, byproduct, output
77

8-
const FORWARD_BACKEND = AutoEnzyme(; mode=Enzyme.Forward, function_annotation=Enzyme.Const)
8+
const FORWARD_BACKEND = AutoEnzyme(;
9+
mode=Enzyme.set_runtime_activity(Enzyme.Forward), function_annotation=Enzyme.Const
10+
)
911

1012
function EnzymeRules.forward(
1113
config::EnzymeRules.FwdConfig,

test/systematic.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@ include("utils.jl")
1212

1313
backends = [
1414
AutoForwardDiff(; chunksize=1), #
15-
AutoEnzyme(; mode=Enzyme.Forward, function_annotation=Enzyme.Const),
15+
AutoEnzyme(;
16+
mode=Enzyme.set_runtime_activity(Enzyme.Forward), function_annotation=Enzyme.Const
17+
),
1618
AutoZygote(),
1719
]
1820

@@ -24,7 +26,9 @@ linear_solver_candidates = (
2426
conditions_backend_candidates = (
2527
nothing, #
2628
AutoForwardDiff(; chunksize=1),
27-
AutoEnzyme(; mode=Enzyme.Forward, function_annotation=Enzyme.Const),
29+
AutoEnzyme(;
30+
mode=Enzyme.set_runtime_activity(Enzyme.Forward), function_annotation=Enzyme.Const
31+
),
2832
);
2933

3034
x_candidates = (

0 commit comments

Comments
 (0)