Skip to content

Commit 844d78c

Browse files
committed
Fix check_forward_mode and check_reverse_mode
1 parent b728451 commit 844d78c

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

src/globalization/line_search.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ function __internal_init(
122122
end
123123
else
124124
autodiff = get_concrete_reverse_ad(
125-
alg.autodiff, prob; check_forward_mode = true)
125+
alg.autodiff, prob; check_reverse_mode = true)
126126
vjp_op = VecJacOperator(prob, fu, u; autodiff)
127127
if isinplace(prob)
128128
g_cache = similar(u)

src/internal/approximate_initialization.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ function __internal_init(
149149
prob::AbstractNonlinearProblem, alg::TrueJacobianInitialization, solver, f::F, fu,
150150
u, p; linsolve = missing, internalnorm::IN = DEFAULT_NORM, kwargs...) where {F, IN}
151151
autodiff = get_concrete_forward_ad(
152-
alg.autodiff, prob; check_reverse_mode = false, kwargs...)
152+
alg.autodiff, prob; check_forward_mode = false, kwargs...)
153153
jac_cache = JacobianCache(prob, solver, prob.f, fu, u, p; autodiff, linsolve)
154154
J = alg.structure(jac_cache(nothing))
155155
return InitializedApproximateJacobianCache(

src/internal/jacobian.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,11 @@ function JacobianCache(
5656
iip = isinplace(prob)
5757
uf = JacobianWrapper{iip}(f, p)
5858

59-
autodiff = get_concrete_forward_ad(autodiff, prob; check_reverse_mode = false)
59+
autodiff = get_concrete_forward_ad(autodiff, prob; check_forward_mode = false)
6060
jvp_autodiff = get_concrete_forward_ad(
61-
jvp_autodiff, prob, Val(false); check_reverse_mode = true)
61+
jvp_autodiff, prob, Val(false); check_forward_mode = true)
6262
vjp_autodiff = get_concrete_reverse_ad(
63-
vjp_autodiff, prob, Val(false); check_forward_mode = false)
63+
vjp_autodiff, prob, Val(false); check_reverse_mode = false)
6464

6565
has_analytic_jac = SciMLBase.has_jac(f)
6666
linsolve_needs_jac = concrete_jac(alg) === nothing && (linsolve === missing ||
@@ -100,7 +100,7 @@ end
100100
function JacobianCache(
101101
prob, alg, f::F, ::Number, u::Number, p; autodiff = nothing, kwargs...) where {F}
102102
uf = JacobianWrapper{false}(f, p)
103-
autodiff = get_concrete_forward_ad(autodiff, prob; check_reverse_mode = false)
103+
autodiff = get_concrete_forward_ad(autodiff, prob; check_forward_mode = false)
104104
if !(autodiff isa AutoForwardDiff ||
105105
autodiff isa AutoPolyesterForwardDiff ||
106106
autodiff isa AutoFiniteDiff)

0 commit comments

Comments
 (0)