Skip to content

Commit ae2b4e4

Browse files
committed
correct getindex of p
1 parent af3e59c commit ae2b4e4

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/systems/clock_inference.jl

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -326,17 +326,20 @@ function generate_discrete_affect(
326326
# @show disc_range
327327
affect! = :(function (integrator, saved_values)
328328
@unpack u, p, t = integrator
329-
cache = p[$disc_range] # Cache needed for atomic state update
330329
c2d_obs = $cont_to_disc_obs
331330
d2c_obs = $disc_to_cont_obs
332331
$(
333332
if use_index_cache
334-
:(disc_unknowns = [$(parameter_values)(p, i) for i in $disc_range])
333+
quote
334+
disc_unknowns = [$(parameter_values)(p, i) for i in $disc_range]
335+
cache = copy(disc_unknowns) # Cache needed for atomic state update
336+
end
335337
else
336338
quote
337339
c2d_view = view(p, $cont_to_disc_idxs)
338340
d2c_view = view(p, $disc_to_cont_idxs)
339341
disc_unknowns = view(p, $disc_range)
342+
cache = copy(disc_unknowns)
340343
end
341344
end
342345
)

0 commit comments

Comments
 (0)