@@ -20,7 +20,7 @@ daesolvers = [Ascher2, Ascher4, Ascher6]
2020 parammap = [α => 7.5 , β => 4 , γ => 8.0 , δ => 5.0 ]
2121 tspan = (0.0 , 10.0 )
2222
23- @mtkbuild lotkavolterra = System (eqs, t)
23+ @mtkcompile lotkavolterra = System (eqs, t)
2424 op = ODEProblem (lotkavolterra, u0map, tspan, parammap)
2525 osol = solve (op, Vern9 ())
2626
5252 eqs = [D (θ) ~ θ_t
5353 D (θ_t) ~ - (g / L) * sin (θ)]
5454
55- @mtkbuild pend = System (eqs, t)
55+ @mtkcompile pend = System (eqs, t)
5656
5757 u0map = [θ => π / 2 , θ_t => π / 2 ]
5858 parammap = [:L => 1.0 , :g => 9.81 ]
9191 D (y (t)) ~ - γ * y (t) + δ * x (t) * y (t)]
9292
9393 tspan = (0.0 , 1.0 )
94- @mtkbuild lksys = System (eqs, t)
94+ @mtkcompile lksys = System (eqs, t)
9595
9696 function lotkavolterra! (du, u, p, t)
9797 du[1 ] = p[1 ] * u[1 ] - p[2 ] * u[1 ] * u[2 ]
104104
105105 # Test with a constraint.
106106 constr = [y (0.5 ) ~ 2.0 ]
107- @mtkbuild lksys = System (eqs, t; constraints = constr)
107+ @mtkcompile lksys = System (eqs, t; constraints = constr)
108108
109109 function bc! (resid, u, p, t)
110110 resid[1 ] = u (0.0 )[1 ] - 1.0
@@ -177,21 +177,21 @@ end
177177 tspan = (0.0 , 1.0 )
178178 guess = [x (t) => 4.0 , y (t) => 2.0 ]
179179 constr = [x (0.6 ) ~ 3.5 , x (0.3 ) ~ 7.0 ]
180- @mtkbuild lksys = System (eqs, t; constraints = constr)
180+ @mtkcompile lksys = System (eqs, t; constraints = constr)
181181
182182 bvp = SciMLBase. BVProblem {true, SciMLBase.AutoSpecialize} (
183183 lksys, u0map, tspan; guesses = guess)
184184 test_solvers (solvers, bvp, u0map, constr; dt = 0.05 )
185185
186186 # Testing that more complicated constraints give correct solutions.
187187 constr = [y (0.2 ) + x (0.8 ) ~ 3.0 , y (0.3 ) ~ 2.0 ]
188- @mtkbuild lksys = System (eqs, t; constraints = constr)
188+ @mtkcompile lksys = System (eqs, t; constraints = constr)
189189 bvp = SciMLBase. BVProblem {false, SciMLBase.FullSpecialize} (
190190 lksys, u0map, tspan; guesses = guess)
191191 test_solvers (solvers, bvp, u0map, constr; dt = 0.05 )
192192
193193 constr = [α * β - x (0.6 ) ~ 0.0 , y (0.2 ) ~ 3.0 ]
194- @mtkbuild lksys = System (eqs, t; constraints = constr)
194+ @mtkcompile lksys = System (eqs, t; constraints = constr)
195195 bvp = SciMLBase. BVProblem {true, SciMLBase.AutoSpecialize} (
196196 lksys, u0map, tspan; guesses = guess)
197197 test_solvers (solvers, bvp, u0map, constr)
205205# eqs = [D(D(x)) ~ λ * x
206206# D(D(y)) ~ λ * y - g
207207# x^2 + y^2 ~ 1]
208- # @mtkbuild pend = System(eqs, t)
208+ # @mtkcompile pend = System(eqs, t)
209209#
210210# tspan = (0.0, 1.5)
211211# u0map = [x => 1, y => 0]
243243# D(D(y)) ~ λ * y - g
244244# x(t)^2 + y^2 ~ 1]
245245# constr = [x(0.5) ~ 1]
246- # @mtkbuild pend = System(eqs, t; constr)
246+ # @mtkcompile pend = System(eqs, t; constr)
247247#
248248# tspan = (0.0, 1.5)
249249# u0map = [x(t) => 0.6, y => 0.8]
@@ -262,13 +262,13 @@ end
262262#
263263# constr = [x(0.5) ~ 1,
264264# x(0.3)^3 + y(0.6)^2 ~ 0.5]
265- # @mtkbuild pend = System(eqs, t; constr)
265+ # @mtkcompile pend = System(eqs, t; constr)
266266# bvp = SciMLBase.BVProblem{true, SciMLBase.AutoSpecialize}(pend, u0map, tspan, parammap; guesses, check_length = false)
267267# test_solvers(daesolvers, bvp, u0map, constr, get_alg_eqs(pend))
268268#
269269# constr = [x(0.4) * g ~ y(0.2),
270270# y(0.7) ~ 0.3]
271- # @mtkbuild pend = System(eqs, t; constr)
271+ # @mtkcompile pend = System(eqs, t; constr)
272272# bvp = SciMLBase.BVProblem{true, SciMLBase.AutoSpecialize}(pend, u0map, tspan, parammap; guesses, check_length = false)
273273# test_solvers(daesolvers, bvp, u0map, constr, get_alg_eqs(pend))
274274# end
286286 parammap = [α => 7.5 , β => 4 , γ => 8.0 , δ => 5.0 ]
287287 costs = [x (0.6 ), x (0.3 )^ 2 ]
288288 consolidate (u, sub) = (u[1 ] + 3 )^ 2 + u[2 ] + sum (sub; init = 0 )
289- @mtkbuild lksys = System (eqs, t; costs, consolidate)
289+ @mtkcompile lksys = System (eqs, t; costs, consolidate)
290290
291291 @test_throws ModelingToolkit. SystemCompatibilityError ODEProblem (
292292 lksys, u0map, tspan, parammap)
301301 @parameters t_c
302302 costs = [y (t_c) + x (0.0 ), x (0.4 )^ 2 ]
303303 consolidate (u, sub) = log (u[1 ]) - u[2 ] + sum (sub; init = 0 )
304- @mtkbuild lksys = System (eqs, t; costs, consolidate)
304+ @mtkcompile lksys = System (eqs, t; costs, consolidate)
305305 @test t_c ∈ Set (parameters (lksys))
306306 push! (parammap, t_c => 0.56 )
307307 prob = ODEProblem (lksys, u0map, tspan, parammap; check_compatibility = false )
0 commit comments