Skip to content

Commit bba01be

Browse files
committed
set runtime activity for enzyme
1 parent 0671fe3 commit bba01be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/adjoint_tests.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
∂p_forwarddiff = ForwardDiff.gradient(solve_nlprob, p)
1717
∂p_tracker = Tracker.data(only(Tracker.gradient(solve_nlprob, p)))
1818
∂p_reversediff = ReverseDiff.gradient(solve_nlprob, p)
19-
∂p_enzyme = Enzyme.gradient(Enzyme.Reverse, solve_nlprob, p)[1]
19+
∂p_enzyme = Enzyme.gradient(Enzyme.set_runtime_activity(Enzyme.Reverse), solve_nlprob, p)[1]
2020

2121
cache = Mooncake.prepare_gradient_cache(solve_nlprob, p)
2222
∂p_mooncake = Mooncake.value_and_gradient!!(cache, solve_nlprob, p)[2][2]

0 commit comments

Comments
 (0)