@@ -382,20 +382,20 @@ end
382
382
function setup_w_wp (cb:: Union{DiscreteCallback, ContinuousCallback} ,
383
383
autojacvec:: Union{ReverseDiffVJP, EnzymeVJP} , pos_neg, event_idx,
384
384
tprev)
385
- w = let _tprev = Ref ( tprev) , pos_neg = pos_neg
385
+ w = let tprev = tprev, pos_neg = pos_neg
386
386
function (du, u, p, t)
387
387
_affect! = get_affect! (cb, pos_neg)
388
- fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, _tprev[] )
388
+ fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, tprev )
389
389
_affect! (fakeinteg)
390
390
du .= fakeinteg. u
391
391
nothing
392
392
end
393
393
end
394
394
395
- wp = let _tprev = Ref ( tprev) , pos_neg = pos_neg
395
+ wp = let tprev = tprev, pos_neg = pos_neg
396
396
function (dp, p, u, t)
397
397
_affect! = get_affect! (cb, pos_neg)
398
- fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, _tprev[] )
398
+ fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, tprev )
399
399
_affect! (fakeinteg)
400
400
dp .= fakeinteg. p
401
401
nothing
@@ -407,20 +407,20 @@ end
407
407
function setup_w_wp (cb:: VectorContinuousCallback ,
408
408
autojacvec:: Union{ReverseDiffVJP, EnzymeVJP} , pos_neg, event_idx,
409
409
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
411
411
function (du, u, p, t)
412
412
_affect! = get_affect! (cb, pos_neg)
413
- fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, _tprev[] )
413
+ fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, tprev )
414
414
_affect! (fakeinteg, event_idx)
415
415
du .= fakeinteg. u
416
416
nothing
417
417
end
418
418
end
419
419
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
421
421
function (dp, p, u, t)
422
422
_affect! = get_affect! (cb, pos_neg)
423
- fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, _tprev[] )
423
+ fakeinteg = get_FakeIntegrator (autojacvec, u, p, t, tprev )
424
424
_affect! (fakeinteg, event_idx)
425
425
dp .= fakeinteg. p
426
426
nothing
0 commit comments