Skip to content

Commit d354ab1

Browse files
Avik Palavik-pal
authored andcommitted
Use inplace accumulation
1 parent 19312b1 commit d354ab1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/simplegmres.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -502,7 +502,7 @@ function SciMLBase.solve!(cache::SimpleGMRESCache{true}, lincache::LinearCache)
502502
mul!(w, A, p) # w ← ANvₖ
503503
PlisI || ldiv!(q, Pl, w) # q ← MANvₖ
504504
for i in 1:inner_iter
505-
R[nr + i] .= vec(sum(__batch(V[i]) .* __batch(q); dims = 1)) # hᵢₖ = (vᵢ)ᴴq
505+
sum!(R[nr + i]', __batch(V[i]) .* __batch(q))
506506
q .-= vec(R[nr + i]' .* __batch(V[i])) # q ← q - hᵢₖvᵢ
507507
end
508508

0 commit comments

Comments
 (0)