Skip to content

Commit 88736f3

Browse files
authored
Don't use mul!
1 parent 24bc0b8 commit 88736f3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/quadratic_newton.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,8 +120,7 @@ function descent_dir!(solver::QuadraticOTSolver{<:QuadraticOTNewton})
120120
G[(M + 1):end, (M + 1):end] .= Diagonal(vec(sum(σ; dims=1)))
121121
G[1:M, (M + 1):end] .= σ
122122
G[(M + 1):end, 1:M] .= σ'
123-
# G[diagind(G)] .+= δ # regularise cg
124-
mul!(G, δ, I, true, true)
123+
view(G, diagind(G)) .+= δ # regularise cg
125124

126125
# cg step
127126
b = -eps * vcat(vec(sum(γ; dims=2)) .- μ, vec(sum(γ; dims=1)) .- ν)

0 commit comments

Comments
 (0)