Passing jacobian generated from ModelingToolkit to ODEFunction for EnsembleGPUArray

Specific Domains Modelling & Simulations
, ,
2

ODEFunction on sys just needs Bools. I’m not sure why this was made so complicated.

function lorenz(du,u,p,t)
    du[1] = p[1]*(u[2]-u[1])
    du[2] = u[1]*(p[2]-u[3]) - u[2]
    du[3] = u[1]*u[2] - p[3]*u[3]
end
u0 = Float32[1.0;0.0;0.0]
tspan = (0.0f0,100.0f0)
p = [10.0f0,28.0f0,8/3f0]
prob = ODEProblem(lorenz,u0,tspan,p)
sys = modelingtoolkitize(prob)
func = ODEFunction(sys,jac=true,tgrad=true)
2 Likes