fix: remove linear indexing in adjoint utility function

SciML · Jun 13, 2024 · 63cc3d6 · 63cc3d6
1 parent 9478431
commit 63cc3d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl
@@ -143,7 +143,7 @@ function not_obs_grads(VA::ODESolution{T}, sym, not_obss_idx, i, Δ) where {T}
         map(enumerate(us)) do (u_idx, u)
             if u_idx in i
                 idx = findfirst(isequal(u_idx), i)
-                Δ[t_idx][idx]
+                Δ.u[t_idx][idx]
             else
                 zero(T)
             end