Skip to content

Commit 16ba23e

Browse files
committed
fix merge
1 parent fd35cc6 commit 16ba23e

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

lib/OrdinaryDiffEqRosenbrock/src/rosenbrock_perform_step.jl

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ end
5252
linres = dolinsolve(integrator, cache.linsolve; A = repeat_step ? nothing : W, b = _vec(linsolve_tmp))
5353
veck₁ = _vec(k₁)
5454

55-
@.. veck₁ = vecu * neginvdtγ
55+
@.. veck₁ = linres.u * neginvdtγ
5656
integrator.stats.nsolve += 1
5757

5858
@.. u=uprev + dto2 * k₁
@@ -69,10 +69,9 @@ end
6969
@.. linsolve_tmp = f₁ - tmp
7070

7171
linres = dolinsolve(integrator, linres.cache; b = _vec(linsolve_tmp))
72-
vecu = _vec(linres.u)
7372
veck₂ = _vec(k₂)
7473

75-
@.. veck₂ = vecu * neginvdtγ + veck₁
74+
@.. veck₂ = linres.u * neginvdtγ + veck₁
7675
integrator.stats.nsolve += 1
7776

7877
@.. u = uprev + dt * k₂
@@ -95,7 +94,7 @@ end
9594

9695
linres = dolinsolve(integrator, linres.cache; b = _vec(linsolve_tmp))
9796
veck3 = _vec(k₃)
98-
@.. veck3 = vecu * neginvdtγ
97+
@.. veck3 = linres.u * neginvdtγ
9998

10099
integrator.stats.nsolve += 1
101100

@@ -150,9 +149,8 @@ end
150149
integrator.opts.internalnorm, t)
151150

152151
linres = dolinsolve(integrator, cache.linsolve; A = repeat_step ? nothing : W, b = _vec(linsolve_tmp))
153-
154152
veck₁ = _vec(k₁)
155-
@.. veck₁ = vecu * neginvdtγ
153+
@.. veck₁ = linres.u * neginvdtγ
156154
integrator.stats.nsolve += 1
157155

158156
@.. broadcast=false u=uprev + dto2 * k₁
@@ -169,9 +167,8 @@ end
169167
@.. broadcast=false linsolve_tmp=f₁ - tmp
170168

171169
linres = dolinsolve(integrator, linres.cache; b = _vec(linsolve_tmp))
172-
vecu = _vec(linres.u)
173170
veck₂ = _vec(k₂)
174-
@.. veck₂ = vecu * neginvdtγ + veck₁
171+
@.. veck₂ = linres.u * neginvdtγ + veck₁
175172
integrator.stats.nsolve += 1
176173

177174
@.. tmp = uprev + dt * k₂
@@ -191,7 +188,7 @@ end
191188
linres = dolinsolve(integrator, linres.cache; b = _vec(linsolve_tmp))
192189
veck3 = _vec(k₃)
193190

194-
@.. veck3 = vecu * neginvdtγ
191+
@.. veck3 = linres.u * neginvdtγ
195192
integrator.stats.nsolve += 1
196193

197194
@.. broadcast=false u=uprev + dto6 * (k₁ + 4k₂ + k₃)

0 commit comments

Comments
 (0)