Skip to content

Commit 799f04b

Browse files
oscardssmithoscardssmith
committed
update extrapolation
1 parent d0f41b1 commit 799f04b

File tree

1 file changed

+15
-17
lines changed

1 file changed

+15
-17
lines changed

‎lib/OrdinaryDiffEqExtrapolation/src/extrapolation_perform_step.jl

Lines changed: 15 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ function perform_step!(integrator, cache::ImplicitEulerExtrapolationCache,
290290
calc_J!(J, integrator, cache) # Store the calculated jac as it won't change in internal discretisation
291291
for index in 1:(n_curr + 1)
292292
dt_temp = dt / sequence[index]
293-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_temp, J, true)
293+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_temp, J)
294294
integrator.stats.nw += 1
295295
@.. broadcast=false k_tmps[1]=integrator.fsalfirst
296296
@.. broadcast=false u_tmps[1]=uprev
@@ -344,9 +344,7 @@ function perform_step!(integrator, cache::ImplicitEulerExtrapolationCache,
344344
endIndex = (i == 1) ? n_curr : n_curr + 1
345345
for index in startIndex:endIndex
346346
dt_temp = dt / sequence[index]
347-
jacobian2W!(
348-
W[Threads.threadid()], integrator.f.mass_matrix, dt_temp, J,
349-
true)
347+
jacobian2W!(W[Threads.threadid()], integrator.f.mass_matrix, dt_temp, J)
350348
@.. broadcast=false k_tmps[Threads.threadid()]=integrator.fsalfirst
351349
@.. broadcast=false u_tmps[Threads.threadid()]=uprev
352350
for j in 1:sequence[index]
@@ -445,7 +443,7 @@ function perform_step!(integrator, cache::ImplicitEulerExtrapolationCache,
445443
cache.n_curr = n_curr
446444

447445
dt_temp = dt / sequence[n_curr + 1]
448-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_temp, J, false)
446+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_temp, J)
449447
integrator.stats.nw += 1
450448
@.. broadcast=false k_tmps[1]=integrator.fsalfirst
451449
@.. broadcast=false u_tmps[1]=uprev
@@ -1170,7 +1168,7 @@ function perform_step!(integrator, cache::ImplicitDeuflhardExtrapolationCache,
11701168
for i in 0:n_curr
11711169
j_int = 4 * subdividing_sequence[i + 1]
11721170
dt_int = dt / j_int # Stepsize of the ith internal discretisation
1173-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J, true)
1171+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J)
11741172
integrator.stats.nw += 1
11751173
@.. broadcast=false u_temp2=uprev
11761174
@.. broadcast=false linsolve_tmps[1]=fsalfirst
@@ -1241,7 +1239,7 @@ function perform_step!(integrator, cache::ImplicitDeuflhardExtrapolationCache,
12411239
j_int_temp = 4 * subdividing_sequence[index + 1]
12421240
dt_int_temp = dt / j_int_temp # Stepsize of the ith internal discretisation
12431241
jacobian2W!(W[Threads.threadid()], integrator.f.mass_matrix,
1244-
dt_int_temp, J, true)
1242+
dt_int_temp, J)
12451243
@.. broadcast=false u_temp4[Threads.threadid()]=uprev
12461244
@.. broadcast=false linsolve_tmps[Threads.threadid()]=fsalfirst
12471245

@@ -1326,7 +1324,7 @@ function perform_step!(integrator, cache::ImplicitDeuflhardExtrapolationCache,
13261324
j_int_temp = 4 * subdividing_sequence[index + 1]
13271325
dt_int_temp = dt / j_int_temp # Stepsize of the ith internal discretisation
13281326
jacobian2W!(W[Threads.threadid()], integrator.f.mass_matrix,
1329-
dt_int_temp, J, true)
1327+
dt_int_temp, J)
13301328
@.. broadcast=false u_temp4[Threads.threadid()]=uprev
13311329
@.. broadcast=false linsolve_tmps[Threads.threadid()]=fsalfirst
13321330

@@ -1450,7 +1448,7 @@ function perform_step!(integrator, cache::ImplicitDeuflhardExtrapolationCache,
14501448
# Update cache.T
14511449
j_int = 4 * subdividing_sequence[n_curr + 1]
14521450
dt_int = dt / j_int # Stepsize of the new internal discretisation
1453-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J, true)
1451+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J)
14541452
integrator.stats.nw += 1
14551453
@.. broadcast=false u_temp2=uprev
14561454
@.. broadcast=false linsolve_tmps[1]=fsalfirst
@@ -2536,7 +2534,7 @@ function perform_step!(integrator, cache::ImplicitHairerWannerExtrapolationCache
25362534
for i in 0:n_curr
25372535
j_int = 4 * subdividing_sequence[i + 1]
25382536
dt_int = dt / j_int # Stepsize of the ith internal discretisation
2539-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J, true)
2537+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J)
25402538
integrator.stats.nw += 1
25412539
@.. broadcast=false u_temp2=uprev
25422540
@.. broadcast=false linsolve_tmps[1]=fsalfirst
@@ -2610,7 +2608,7 @@ function perform_step!(integrator, cache::ImplicitHairerWannerExtrapolationCache
26102608
j_int_temp = 4 * subdividing_sequence[index + 1]
26112609
dt_int_temp = dt / j_int_temp # Stepsize of the ith internal discretisation
26122610
jacobian2W!(W[Threads.threadid()], integrator.f.mass_matrix,
2613-
dt_int_temp, J, true)
2611+
dt_int_temp, J)
26142612
@.. broadcast=false u_temp4[Threads.threadid()]=uprev
26152613
@.. broadcast=false linsolve_tmps[Threads.threadid()]=fsalfirst
26162614

@@ -2701,7 +2699,7 @@ function perform_step!(integrator, cache::ImplicitHairerWannerExtrapolationCache
27012699
index == -1 && continue
27022700
j_int_temp = 4 * subdividing_sequence[index + 1]
27032701
dt_int_temp = dt / j_int_temp # Stepsize of the ith internal discretisation
2704-
jacobian2W!(W[tid], integrator.f.mass_matrix, dt_int_temp, J, true)
2702+
jacobian2W!(W[tid], integrator.f.mass_matrix, dt_int_temp, J)
27052703
@.. broadcast=false u_temp4[tid]=uprev
27062704
@.. broadcast=false linsolvetmp=fsalfirst
27072705

@@ -2815,7 +2813,7 @@ function perform_step!(integrator, cache::ImplicitHairerWannerExtrapolationCache
28152813
# Update cache.T
28162814
j_int = 4 * subdividing_sequence[n_curr + 1]
28172815
dt_int = dt / j_int # Stepsize of the new internal discretisation
2818-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J, true)
2816+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J)
28192817
integrator.stats.nw += 1
28202818
@.. broadcast=false u_temp2=uprev
28212819
@.. broadcast=false linsolve_tmps[1]=fsalfirst
@@ -3227,7 +3225,7 @@ function perform_step!(integrator, cache::ImplicitEulerBarycentricExtrapolationC
32273225
for i in 0:n_curr
32283226
j_int = sequence_factor * subdividing_sequence[i + 1]
32293227
dt_int = dt / j_int # Stepsize of the ith internal discretisation
3230-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J, true)
3228+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J)
32313229
integrator.stats.nw += 1
32323230
@.. broadcast=false u_temp2=uprev
32333231
@.. broadcast=false linsolve_tmps[1]=fsalfirst
@@ -3301,7 +3299,7 @@ function perform_step!(integrator, cache::ImplicitEulerBarycentricExtrapolationC
33013299
j_int_temp = sequence_factor * subdividing_sequence[index + 1]
33023300
dt_int_temp = dt / j_int_temp # Stepsize of the ith internal discretisation
33033301
jacobian2W!(W[Threads.threadid()], integrator.f.mass_matrix,
3304-
dt_int_temp, J, true)
3302+
dt_int_temp, J)
33053303
@.. broadcast=false u_temp4[Threads.threadid()]=uprev
33063304
@.. broadcast=false linsolve_tmps[Threads.threadid()]=fsalfirst
33073305

@@ -3389,7 +3387,7 @@ function perform_step!(integrator, cache::ImplicitEulerBarycentricExtrapolationC
33893387
j_int_temp = sequence_factor * subdividing_sequence[index + 1]
33903388
dt_int_temp = dt / j_int_temp # Stepsize of the ith internal discretisation
33913389
jacobian2W!(W[Threads.threadid()], integrator.f.mass_matrix,
3392-
dt_int_temp, J, true)
3390+
dt_int_temp, J)
33933391
@.. broadcast=false u_temp4[Threads.threadid()]=uprev
33943392
@.. broadcast=false linsolve_tmps[Threads.threadid()]=fsalfirst
33953393

@@ -3519,7 +3517,7 @@ function perform_step!(integrator, cache::ImplicitEulerBarycentricExtrapolationC
35193517
# Update cache.T
35203518
j_int = sequence_factor * subdividing_sequence[n_curr + 1]
35213519
dt_int = dt / j_int # Stepsize of the new internal discretisation
3522-
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J, true)
3520+
jacobian2W!(W[1], integrator.f.mass_matrix, dt_int, J)
35233521
integrator.stats.nw += 1
35243522
@.. broadcast=false u_temp2=uprev
35253523
@.. broadcast=false linsolve_tmps[1]=fsalfirst

0 commit comments

Comments
 (0)