Skip to content

Commit 872a252

Browse files
committed
Fix runtime activity
1 parent abafa86 commit 872a252

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

  • DifferentiationInterface/test/Back/Enzyme

DifferentiationInterface/test/Back/Enzyme/test.jl

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -78,11 +78,15 @@ end
7878
)
7979

8080
test_differentiation(
81-
# TODO: replace with AutoEnzyme() when https://github.com/EnzymeAD/Enzyme.jl/issues/2854 is fixed
82-
SecondOrder(
83-
AutoEnzyme(; mode = Enzyme.set_runtime_activity(Enzyme.Forward)),
84-
AutoEnzyme(; mode = Enzyme.set_runtime_activity(Enzyme.Reverse))
85-
),
81+
# TODO: simplify when https://github.com/EnzymeAD/Enzyme.jl/issues/2854 and https://github.com/EnzymeAD/Enzyme.jl/issues/2925 are fixed
82+
if VERSION >= v"1.11"
83+
SecondOrder(
84+
AutoEnzyme(; mode = Enzyme.set_runtime_activity(Enzyme.Forward)),
85+
AutoEnzyme(; mode = Enzyme.set_runtime_activity(Enzyme.Reverse))
86+
)
87+
else
88+
AutoEnzyme()
89+
end,
8690
default_scenarios(; include_normal = false, include_constantified = false, include_cachified = true);
8791
excluded = vcat(FIRST_ORDER, :second_derivative),
8892
logging = LOGGING,

0 commit comments

Comments
 (0)