Merge pull request #1078 from SciML/ChrisRackauckas-patch-2

Revert breakage in GPU
SciML · Jul 8, 2024 · 1cad860 · 1cad860
2 parents 21223ca + 6466c93
commit 1cad860
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/concrete_solve.jl b/src/concrete_solve.jl
@@ -461,10 +461,8 @@ function DiffEqBase._concrete_solve_adjoint(
  no_start && (sol_idxs = sol_idxs[2:end])
  no_end && (sol_idxs = sol_idxs[1:(end - 1)])
  only_end = length(sol_idxs) <= 1
- uf = save_idxs === nothing ?
- getu(sol, eachindex(IndexCartesian(), first(state_values(sol)))) :
- getu(sol, save_idxs)
- u = uf(sol, sol_idxs)
+ _u = sol.u[sol_idxs]
+ u = save_idxs === nothing ? _u : [x[save_idxs] for x in _u]
  ts = current_time(sol, sol_idxs)
  out = DiffEqBase.sensitivity_solution(sol, u, ts)
  else