On this topic, there’s currently an issue here that ArrayPartition is not compatible with esp. stiff solvers since linear algebra is not implemented for it, (it’s mentioned that PartitionedODEProblem use them internally so maybe it’s also affected? btw, PartitionedODEProblem not defined appears to be undefined, looks like code is absent)
Is there a roadmap about what needs to be done if one wishes to help with it (overloading the functions with ArrayPartition or copy it in a linearized way to a cache)? It can be very useful for a set of coupled odes that included matrices, vectors and numbers, e.g.
v’ = fv(m_i,v,s)
m_i’ = fm(m_j,v,s)
s’ = fs(m_i,v,s)
where v is a vector, m_{1…n} are matrices and s is a number.
right now, I’m handling it through reshaping v,m,s to 1d and vcat them together, speed is okay for small systems but for bigger systems it may require preallocating and passing the preallocated arrays via the p parameter (may still generate some allocations).