Skip to content

Commit 70ba8e9

Browse files
fix mutation and test
1 parent ec7eb5f commit 70ba8e9

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/concrete_solve.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1288,7 +1288,7 @@ function DiffEqBase._concrete_solve_adjoint(
12881288
end
12891289

12901290
forward, reverse = Enzyme.autodiff_thunk(splitmode, Enzyme.Const{typeof(f)}, Enzyme.Duplicated, Enzyme.Duplicated{typeof(u0)}, Enzyme.Duplicated{typeof(p)})
1291-
tape, result, shadow_result = forward(Enzyme.Const(f), Enzyme.Duplicated(u0, du0), Enzyme.Duplicated(p, dp))
1291+
tape, result, shadow_result = forward(Enzyme.Const(f), Enzyme.Duplicated(copy(u0), du0), Enzyme.Duplicated(copy(p), dp))
12921292

12931293
function enzyme_sensitivity_backpass(Δ)
12941294
reverse(Const(f), Duplicated(u0, du0), Duplicated(p, dp), Δ, tape)

test/concrete_solve_derivatives.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -354,6 +354,7 @@ du09, dp9 = Zygote.gradient(
354354
@test ū0du06 rtol=1e-12
355355
@test_broken ū0du07 rtol=1e-12
356356
@test_broken ū0du08 rtol=1e-12
357+
@test ū0du09 rtol=1e-12
357358
@test adjdp1' rtol=1e-12
358359
@test adjdp2' rtol=1e-12
359360
@test adjdp3' rtol=1e-12

0 commit comments

Comments
 (0)