Skip to content

Commit 9f14a4f

Browse files
Revert "Make tprev mutable to make enzyme happy"
This reverts commit bcb3e84.
1 parent 99de0e1 commit 9f14a4f

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

src/callback_tracking.jl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -382,20 +382,20 @@ end
382382
function setup_w_wp(cb::Union{DiscreteCallback, ContinuousCallback},
383383
autojacvec::Union{ReverseDiffVJP, EnzymeVJP}, pos_neg, event_idx,
384384
tprev)
385-
w = let _tprev = Ref(tprev), pos_neg = pos_neg
385+
w = let tprev = tprev, pos_neg = pos_neg
386386
function (du, u, p, t)
387387
_affect! = get_affect!(cb, pos_neg)
388-
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, _tprev[])
388+
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, tprev)
389389
_affect!(fakeinteg)
390390
du .= fakeinteg.u
391391
nothing
392392
end
393393
end
394394

395-
wp = let _tprev = Ref(tprev), pos_neg = pos_neg
395+
wp = let tprev = tprev, pos_neg = pos_neg
396396
function (dp, p, u, t)
397397
_affect! = get_affect!(cb, pos_neg)
398-
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, _tprev[])
398+
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, tprev)
399399
_affect!(fakeinteg)
400400
dp .= fakeinteg.p
401401
nothing
@@ -407,20 +407,20 @@ end
407407
function setup_w_wp(cb::VectorContinuousCallback,
408408
autojacvec::Union{ReverseDiffVJP, EnzymeVJP}, pos_neg, event_idx,
409409
tprev)
410-
w = let _tprev = Ref(tprev), pos_neg = pos_neg, event_idx = event_idx
410+
w = let tprev = tprev, pos_neg = pos_neg, event_idx = event_idx
411411
function (du, u, p, t)
412412
_affect! = get_affect!(cb, pos_neg)
413-
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, _tprev[])
413+
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, tprev)
414414
_affect!(fakeinteg, event_idx)
415415
du .= fakeinteg.u
416416
nothing
417417
end
418418
end
419419

420-
wp = let _tprev = Ref(tprev), pos_neg = pos_neg, event_idx = event_idx
420+
wp = let tprev = tprev, pos_neg = pos_neg, event_idx = event_idx
421421
function (dp, p, u, t)
422422
_affect! = get_affect!(cb, pos_neg)
423-
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, _tprev[])
423+
fakeinteg = get_FakeIntegrator(autojacvec, u, p, t, tprev)
424424
_affect!(fakeinteg, event_idx)
425425
dp .= fakeinteg.p
426426
nothing

0 commit comments

Comments
 (0)