Skip to content

Commit

Permalink
chore: unthunk dp
Browse files Browse the repository at this point in the history
  • Loading branch information
DhairyaLGandhi committed Nov 8, 2024
1 parent a1c5607 commit 3384966
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/concrete_solve.jl
Original file line number Diff line number Diff line change
Expand Up @@ -1162,7 +1162,7 @@ function DiffEqBase._concrete_solve_adjoint(
ntuple(_ -> NoTangent(), length(args))...)
else
(NoTangent(), NoTangent(), NoTangent(),
du0, repack_adjoint(dp)[1], NoTangent(),
du0, repack_adjoint(unthunk(dp))[1], NoTangent(),
ntuple(_ -> NoTangent(), length(args))...)
end
end
Expand Down

0 comments on commit 3384966

Please sign in to comment.