Skip to content

Commit 2ed9af3

Browse files
add nu_sub variable for R2N
1 parent 526c0fd commit 2ed9af3

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/R2N_alg.jl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,6 +296,8 @@ function SolverCore.solve!(
296296
end
297297

298298
ν₁ = 1 / ((λmax + σk) * (1 + θ))
299+
ν_sub = ν₁
300+
299301
sqrt_ξ1_νInv = one(T)
300302

301303
@. mν∇fk = -ν₁ * ∇fk
@@ -355,13 +357,14 @@ function SolverCore.solve!(
355357

356358
solver.subpb.model.σ = σk
357359
isa(solver.subsolver, R2DHSolver) && (solver.subsolver.D.d[1] = 1/ν₁)
360+
ν_sub = isa(solver.subsolver, R2DHSolver) ? 1/σk : ν₁
358361
solve!(
359362
solver.subsolver,
360363
solver.subpb,
361364
solver.substats;
362365
x = s1,
363366
atol = sub_atol,
364-
ν = 1/σk,
367+
ν = ν_sub,
365368
kwargs...
366369
)
367370

0 commit comments

Comments
 (0)