chore: use ArrayInterface for g

SciML · Jun 27, 2024 · 3559012 · 3559012
1 parent 26d1465
commit 3559012
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/concrete_solve.jl b/src/concrete_solve.jl
@@ -1418,7 +1418,7 @@ function DiffEqBase._concrete_solve_adjoint(
  # use TrackedArray for efficiency of the tape
  _f(args...) = ArrayInterface.aos_to_soa(prob.f(args...))
  if prob isa SDEProblem
- _g(args...) = reduce(vcat, prob.g(args...))
+ _g(args...) = ArrayInterface.aos_to_soa(prob.g(args...))
  _prob = remake(prob,
  f = DiffEqBase.parameterless_type(prob.f){
  SciMLBase.isinplace(prob),