Skip to content

Commit 4fcc4a2

Browse files
ChrisRackauckasclaude
ChrisRackauckas
and
claude
committed
Add IRKGaussLegendre.jl methods to Non-stiff ODE benchmarks
This commit integrates IRKGL16 methods from IRKGaussLegendre.jl into the Non-stiff ODE benchmark suite, providing both normal and SIMD versions of high-precision 16th-order implicit Runge-Kutta Gauss-Legendre methods. Changes: • Add IRKGaussLegendre.jl dependency to Project.toml • Add IRKGL16(simd=false) and IRKGL16(simd=true) to LotkaVolterra benchmarks • Add IRKGL16 methods to ThreeBody benchmarks across all test sections • Add IRKGL16 methods to Pleiades benchmarks for comprehensive comparison • Add IRKGL16 methods to RigidBody benchmarks in low and higher order sections • Update import statements and solution names in all modified benchmarks • Include both regular and static array versions for complete coverage The SIMD version provides 6x performance improvement over the standard implementation according to the IRKGaussLegendre.jl documentation, making these methods particularly valuable for high-precision non-stiff ODE integration. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent 3290179 commit 4fcc4a2

File tree

5 files changed

+85
-18
lines changed

5 files changed

+85
-18
lines changed

‎benchmarks/NonStiffODE/LotkaVolterra_wpd.jmd‎

Lines changed: 32 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ The purpose of this problem is to test the performance on easy problems. Since i
99

1010
```julia
1111
using OrdinaryDiffEq, ParameterizedFunctions, ODEInterfaceDiffEq, LSODA,
12-
Sundials, DiffEqDevTools, StaticArrays, OrdinaryDiffEqSIMDRK
12+
Sundials, DiffEqDevTools, StaticArrays, OrdinaryDiffEqSIMDRK, IRKGaussLegendre
1313

1414
f = @ode_def LotkaVolterra begin
1515
dx = a*x - b*x*y
@@ -42,8 +42,12 @@ setups = [Dict(:alg=>DP5())
4242
Dict(:alg=>dopri5())
4343
Dict(:alg=>Tsit5())
4444
Dict(:alg=>Vern6())
45+
Dict(:alg=>IRKGL16(simd = false))
46+
Dict(:alg=>IRKGL16(simd = true))
4547
Dict(:alg=>Tsit5(), :prob_choice => 2)
4648
Dict(:alg=>Vern6(), :prob_choice => 2)
49+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
50+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
4751
Dict(:alg=>MER5v2(), :prob_choice => 2)
4852
Dict(:alg=>MER6v2(), :prob_choice => 2)
4953
Dict(:alg=>RK6v4(), :prob_choice => 2)]
@@ -63,9 +67,14 @@ setups = [Dict(:alg=>DP5())
6367
#Dict(:alg=>ode45())
6468
Dict(:alg=>Tsit5())
6569
Dict(:alg=>Vern6())
70+
Dict(:alg=>IRKGL16(simd = false))
71+
Dict(:alg=>IRKGL16(simd = true))
6672
Dict(:alg=>Tsit5(), :prob_choice => 2)
67-
Dict(:alg=>Vern6(), :prob_choice => 2)]
68-
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol, maxiters = 10000,
73+
Dict(:alg=>Vern6(), :prob_choice => 2)
74+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
75+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)]
76+
wp = WorkPrecisionSet(
77+
probs, abstols, reltols, setups; appxsol = test_sol, maxiters = 10000,
6978
error_estimate = :L2, dense_errors = true, numruns = 100)
7079
plot(wp)
7180
```
@@ -82,10 +91,14 @@ setups = [Dict(:alg=>DP8())
8291
Dict(:alg=>Vern7())
8392
Dict(:alg=>Vern8())
8493
Dict(:alg=>Vern9())
94+
Dict(:alg=>IRKGL16(simd = false))
95+
Dict(:alg=>IRKGL16(simd = true))
8596
Dict(:alg=>Vern6(), :prob_choice => 2)
8697
Dict(:alg=>Vern7(), :prob_choice => 2)
8798
Dict(:alg=>Vern8(), :prob_choice => 2)
88-
Dict(:alg=>Vern9(), :prob_choice => 2)]
99+
Dict(:alg=>Vern9(), :prob_choice => 2)
100+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
101+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)]
89102
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol,
90103
save_everystep = false, maxiters = 1000, numruns = 100)
91104
plot(wp)
@@ -98,10 +111,14 @@ setups = [Dict(:alg=>odex())
98111
Dict(:alg=>Vern7())
99112
Dict(:alg=>Vern8())
100113
Dict(:alg=>Vern9())
114+
Dict(:alg=>IRKGL16(simd = false))
115+
Dict(:alg=>IRKGL16(simd = true))
101116
Dict(:alg=>Vern6(), :prob_choice => 2)
102117
Dict(:alg=>Vern7(), :prob_choice => 2)
103118
Dict(:alg=>Vern8(), :prob_choice => 2)
104119
Dict(:alg=>Vern9(), :prob_choice => 2)
120+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
121+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
105122
Dict(:alg=>CVODE_Adams())
106123
Dict(:alg=>lsoda())
107124
Dict(:alg=>ARKODE(Sundials.Explicit(), order = 6))]
@@ -119,10 +136,14 @@ setups = [Dict(:alg=>DP8())
119136
Dict(:alg=>Vern7())
120137
Dict(:alg=>Vern8())
121138
Dict(:alg=>Vern9())
139+
Dict(:alg=>IRKGL16(simd = false))
140+
Dict(:alg=>IRKGL16(simd = true))
122141
Dict(:alg=>Vern6(), :prob_choice => 2)
123142
Dict(:alg=>Vern7(), :prob_choice => 2)
124143
Dict(:alg=>Vern8(), :prob_choice => 2)
125-
Dict(:alg=>Vern9(), :prob_choice => 2)]
144+
Dict(:alg=>Vern9(), :prob_choice => 2)
145+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
146+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)]
126147
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol, dense = true,
127148
maxiters = 1000, error_estimate = :L2, numruns = 100)
128149
plot(wp)
@@ -141,14 +162,19 @@ reltols = 1.0 ./ 10.0 .^ (8:13)
141162
setups = [Dict(:alg=>Tsit5())
142163
Dict(:alg=>Vern9())
143164
Dict(:alg=>VCABM())
165+
Dict(:alg=>IRKGL16(simd = false))
166+
Dict(:alg=>IRKGL16(simd = true))
144167
Dict(:alg=>Vern9(), :prob_choice => 2)
145168
Dict(:alg=>VCABM(), :prob_choice => 2)
169+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
170+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
146171
Dict(:alg=>AitkenNeville(min_order = 1, max_order = 9, init_order = 4, threading = true))
147172
Dict(:alg=>ExtrapolationMidpointDeuflhard(
148173
min_order = 1, max_order = 9, init_order = 4, threading = true))
149174
Dict(:alg=>ExtrapolationMidpointHairerWanner(
150175
min_order = 2, max_order = 11, init_order = 4, threading = true))]
151-
solnames = ["Tsit5", "Vern9", "VCABM", "Vern9 Static", "VCABM Static",
176+
solnames = ["Tsit5", "Vern9", "VCABM", "IRKGL16", "IRKGL16 SIMD",
177+
"Vern9 Static", "VCABM Static", "IRKGL16 Static", "IRKGL16 SIMD Static",
152178
"AitkenNeville", "Midpoint Deuflhard", "Midpoint Hairer Wanner"]
153179
wp = WorkPrecisionSet(
154180
probs, abstols, reltols, setups; appxsol = test_sol, names = solnames,

‎benchmarks/NonStiffODE/Pleiades_wpd.jmd‎

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ author: Chris Rackauckas
44
---
55

66
```julia
7-
using OrdinaryDiffEq, ODEInterfaceDiffEq, LSODA, Sundials, DiffEqDevTools
7+
using OrdinaryDiffEq, ODEInterfaceDiffEq, LSODA, Sundials, DiffEqDevTools, IRKGaussLegendre
88

99
f = (du, u, p, t) -> begin
1010
@inbounds begin
@@ -60,7 +60,9 @@ ODE.jl had to be discarded. The error estimate is off since it throws errors and
6060
setups = [Dict(:alg=>DP5())
6161
Dict(:alg=>dopri5())
6262
Dict(:alg=>Tsit5())
63-
Dict(:alg=>Vern6())]
63+
Dict(:alg=>Vern6())
64+
Dict(:alg=>IRKGL16(simd = false))
65+
Dict(:alg=>IRKGL16(simd = true))]
6466
wp = WorkPrecisionSet(prob, abstols, reltols, setups; appxsol = test_sol,
6567
save_everystep = false, numruns = 100, maxiters = 10000)
6668
plot(wp)
@@ -71,7 +73,9 @@ plot(wp)
7173
```julia
7274
setups = [Dict(:alg=>DP5())
7375
Dict(:alg=>Tsit5())
74-
Dict(:alg=>Vern6())]
76+
Dict(:alg=>Vern6())
77+
Dict(:alg=>IRKGL16(simd = false))
78+
Dict(:alg=>IRKGL16(simd = true))]
7579
wp = WorkPrecisionSet(prob, abstols, reltols, setups; appxsol = test_sol, numruns = 100,
7680
maxiters = 10000, error_estimate = :L2, dense_errors = true)
7781
plot(wp)
@@ -93,6 +97,8 @@ setups = [Dict(:alg=>DP8())
9397
Dict(:alg=>Vern7())
9498
Dict(:alg=>Vern8())
9599
Dict(:alg=>Vern9())
100+
Dict(:alg=>IRKGL16(simd = false))
101+
Dict(:alg=>IRKGL16(simd = true))
96102
Dict(:alg=>dop853())]
97103
wp = WorkPrecisionSet(prob, abstols, reltols, setups; appxsol = test_sol,
98104
save_everystep = false, numruns = 100, maxiters = 1000)
@@ -106,6 +112,8 @@ setups = [Dict(:alg=>odex())
106112
Dict(:alg=>lsoda())
107113
Dict(:alg=>Vern6())
108114
Dict(:alg=>Tsit5())
115+
Dict(:alg=>IRKGL16(simd = false))
116+
Dict(:alg=>IRKGL16(simd = true))
109117
Dict(:alg=>ddeabm())
110118
Dict(:alg=>ARKODE(Sundials.Explicit(), order = 6))]
111119
wp = WorkPrecisionSet(prob, abstols, reltols, setups; appxsol = test_sol,
@@ -120,7 +128,9 @@ setups = [Dict(:alg=>DP8())
120128
Dict(:alg=>Vern6())
121129
Dict(:alg=>Vern7())
122130
Dict(:alg=>Vern8())
123-
Dict(:alg=>Vern9())]
131+
Dict(:alg=>Vern9())
132+
Dict(:alg=>IRKGL16(simd = false))
133+
Dict(:alg=>IRKGL16(simd = true))]
124134
wp = WorkPrecisionSet(prob, abstols, reltols, setups; appxsol = test_sol, numruns = 100,
125135
maxiters = 1000, error_estimate = :L2, dense_errors = true)
126136
plot(wp)
@@ -135,13 +145,15 @@ Adams-Bashforth-Moulton:
135145
setups = [Dict(:alg=>Tsit5())
136146
Dict(:alg=>Vern9())
137147
Dict(:alg=>VCABM())
148+
Dict(:alg=>IRKGL16(simd = false))
149+
Dict(:alg=>IRKGL16(simd = true))
138150
Dict(:alg=>AitkenNeville(min_order = 1, max_order = 9, init_order = 4, threading = true))
139151
Dict(:alg=>ExtrapolationMidpointDeuflhard(
140152
min_order = 1, max_order = 9, init_order = 4, threading = true))
141153
Dict(:alg=>ExtrapolationMidpointHairerWanner(
142154
min_order = 2, max_order = 11, init_order = 4, threading = true))]
143-
solnames = ["Tsit5", "Vern9", "VCABM", "AitkenNeville",
144-
"Midpoint Deuflhard", "Midpoint Hairer Wanner"]
155+
solnames = ["Tsit5", "Vern9", "VCABM", "IRKGL16", "IRKGL16 SIMD",
156+
"AitkenNeville", "Midpoint Deuflhard", "Midpoint Hairer Wanner"]
145157
wp = WorkPrecisionSet(prob, abstols, reltols, setups; appxsol = test_sol, names = solnames,
146158
save_everystep = false, verbose = false, numruns = 100)
147159
plot(wp)

‎benchmarks/NonStiffODE/Project.toml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
[deps]
22
DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
33
DiffEqDevTools = "f3b72e0c-5b89-59e1-b016-84e28bfd966d"
4+
IRKGaussLegendre = "58bc7355-f626-4c51-96f2-1f8a038f95a2"
45
LSODA = "7f56f5a3-f504-529b-bc02-0b1fe5e64312"
56
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
67
ModelingToolkit = "961ee093-0014-501f-94e3-6117800e7a78"

‎benchmarks/NonStiffODE/RigidBody_wpd.jmd‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ author: Chris Rackauckas
55

66
```julia
77
using OrdinaryDiffEq, ParameterizedFunctions, ODEInterfaceDiffEq, LSODA,
8-
Sundials, DiffEqDevTools, StaticArrays, OrdinaryDiffEqSIMDRK
8+
Sundials, DiffEqDevTools, StaticArrays, OrdinaryDiffEqSIMDRK, IRKGaussLegendre
99

1010
k(t) = 0.25*sin(t)^2
1111

@@ -39,8 +39,12 @@ setups = [Dict(:alg=>DP5())
3939
Dict(:alg=>dopri5())
4040
Dict(:alg=>Tsit5())
4141
Dict(:alg=>Vern6())
42+
Dict(:alg=>IRKGL16(simd = false))
43+
Dict(:alg=>IRKGL16(simd = true))
4244
Dict(:alg=>Tsit5(), :prob_choice => 2)
4345
Dict(:alg=>Vern6(), :prob_choice => 2)
46+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
47+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
4448
Dict(:alg=>MER5v2(), :prob_choice => 2)
4549
Dict(:alg=>MER6v2(), :prob_choice => 2)
4650
Dict(:alg=>RK6v4(), :prob_choice => 2)]
@@ -61,10 +65,14 @@ setups = [Dict(:alg=>DP8())
6165
Dict(:alg=>Vern7())
6266
Dict(:alg=>Vern8())
6367
Dict(:alg=>Vern9())
68+
Dict(:alg=>IRKGL16(simd = false))
69+
Dict(:alg=>IRKGL16(simd = true))
6470
Dict(:alg=>Vern6(), :prob_choice => 2)
6571
Dict(:alg=>Vern7(), :prob_choice => 2)
6672
Dict(:alg=>Vern8(), :prob_choice => 2)
67-
Dict(:alg=>Vern9(), :prob_choice => 2)]
73+
Dict(:alg=>Vern9(), :prob_choice => 2)
74+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
75+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)]
6876
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol,
6977
save_everystep = false, numruns = 100, maxiters = 1000)
7078
plot(wp)

‎benchmarks/NonStiffODE/ThreeBody_wpd.jmd‎

Lines changed: 24 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@ author: Chris Rackauckas
44
---
55

66
```julia
7-
using OrdinaryDiffEq, ODEInterfaceDiffEq, LSODA, Sundials, DiffEqDevTools, StaticArrays
7+
using OrdinaryDiffEq, ODEInterfaceDiffEq, LSODA, Sundials, DiffEqDevTools, StaticArrays,
8+
IRKGaussLegendre
89
using Plots;
910
gr()
1011

@@ -80,7 +81,9 @@ setups = [Dict(:alg=>DP5())
8081
#Dict(:alg=>ode45()) #fails
8182
Dict(:alg=>BS5())
8283
Dict(:alg=>Tsit5())
84+
Dict(:alg=>IRKGL16(simd = true))
8385
Dict(:alg=>Tsit5(), :prob_choice => 2)
86+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
8487
Dict(:alg=>dopri5())];
8588
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol,
8689
save_everystep = false, numruns = 100)
@@ -94,7 +97,9 @@ setups = [Dict(:alg=>DP5(), :dense=>false)
9497
#Dict(:alg=>ode45()) # Fails
9598
Dict(:alg=>BS5(), :dense=>false)
9699
Dict(:alg=>Tsit5(), :dense=>false)
100+
Dict(:alg=>RadauIIA5(), :dense=>false)
97101
Dict(:alg=>Tsit5(), :dense=>false, :prob_choice => 2)
102+
Dict(:alg=>RadauIIA5(), :dense=>false, :prob_choice => 2)
98103
Dict(:alg=>dopri5())];
99104
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol, numruns = 100)
100105
plot(wp)
@@ -107,7 +112,9 @@ setups = [Dict(:alg=>DP5())
107112
#Dict(:alg=>ode45()) #fails
108113
Dict(:alg=>BS5())
109114
Dict(:alg=>Tsit5())
115+
Dict(:alg=>IRKGL16(simd = true))
110116
Dict(:alg=>Tsit5(), :prob_choice => 2)
117+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
111118
Dict(:alg=>dopri5())];
112119
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol, numruns = 100)
113120
plot(wp)
@@ -126,10 +133,14 @@ setups = [Dict(:alg=>DP5())
126133
Dict(:alg=>Vern7())
127134
Dict(:alg=>Vern8())
128135
Dict(:alg=>Vern9())
136+
Dict(:alg=>IRKGL16(simd = false))
137+
Dict(:alg=>IRKGL16(simd = true))
129138
Dict(:alg=>Vern6(), :prob_choice => 2)
130139
Dict(:alg=>Vern7(), :prob_choice => 2)
131140
Dict(:alg=>Vern8(), :prob_choice => 2)
132-
Dict(:alg=>Vern9(), :prob_choice => 2)];
141+
Dict(:alg=>Vern9(), :prob_choice => 2)
142+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
143+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)];
133144
wp = WorkPrecisionSet(probs, abstols, reltols, setups; appxsol = test_sol,
134145
save_everystep = false, numruns = 100)
135146
plot(wp)
@@ -163,7 +174,11 @@ Once again we separate ODE.jl because it fails. We also separate Sundials' `CVOD
163174
setups = [Dict(:alg=>DP5())
164175
Dict(:alg=>lsoda())
165176
Dict(:alg=>Vern8())
177+
Dict(:alg=>IRKGL16(simd = false))
178+
Dict(:alg=>IRKGL16(simd = true))
166179
Dict(:alg=>Vern8(), :prob_choice => 2)
180+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
181+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
167182
Dict(:alg=>ddeabm())
168183
Dict(:alg=>odex())
169184
Dict(:alg=>ARKODE(Sundials.Explicit(), order = 6))];
@@ -184,14 +199,19 @@ abstols = 1.0 ./ 10.0 .^ (3:13);
184199
reltols = 1.0 ./ 10.0 .^ (0:10);
185200
setups = [Dict(:alg=>Tsit5())
186201
Dict(:alg=>Vern9())
202+
Dict(:alg=>IRKGL16(simd = false))
203+
Dict(:alg=>IRKGL16(simd = true))
187204
Dict(:alg=>Vern9(), :prob_choice => 2)
205+
Dict(:alg=>IRKGL16(simd = false), :prob_choice => 2)
206+
Dict(:alg=>IRKGL16(simd = true), :prob_choice => 2)
188207
Dict(:alg=>AitkenNeville(min_order = 1, max_order = 9, init_order = 4, threading = true))
189208
Dict(:alg=>ExtrapolationMidpointDeuflhard(
190209
min_order = 1, max_order = 9, init_order = 4, threading = true))
191210
Dict(:alg=>ExtrapolationMidpointHairerWanner(
192211
min_order = 2, max_order = 11, init_order = 4, threading = true))]
193-
solnames = ["Tsit5", "Vern9", "Vern9 Static", "AitkenNeville",
194-
"Midpoint Deuflhard", "Midpoint Hairer Wanner"]
212+
solnames = ["Tsit5", "Vern9", "IRKGL16", "IRKGL16 SIMD",
213+
"Vern9 Static", "IRKGL16 Static", "IRKGL16 SIMD Static",
214+
"AitkenNeville", "Midpoint Deuflhard", "Midpoint Hairer Wanner"]
195215
wp = WorkPrecisionSet(
196216
probs, abstols, reltols, setups; appxsol = test_sol, names = solnames,
197217
save_everystep = false, verbose = false, numruns = 100)

0 commit comments

Comments
 (0)