Skip to content

Commit 1b77877

Browse files
committed
fix: minor cleanups
1 parent f3b317d commit 1b77877

File tree

3 files changed

+7
-9
lines changed

3 files changed

+7
-9
lines changed

lib/NonlinearSolveBase/src/autodiff.jl

+5-8
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,8 @@ const ForwardADs = (
3434
function select_forward_mode_autodiff(
3535
prob::AbstractNonlinearProblem, ad::AbstractADType; warn_check_mode::Bool = true)
3636
if warn_check_mode && !(ADTypes.mode(ad) isa ADTypes.ForwardMode) &&
37-
!(ADTypes.mode(ad) isa ADTypes.ForwardOrReverseMode)
37+
!(ADTypes.mode(ad) isa ADTypes.ForwardOrReverseMode) &&
38+
!is_finite_differences_backend(ad)
3839
@warn "The chosen AD backend $(ad) is not a forward mode AD. Use with caution."
3940
end
4041
if incompatible_backend_and_problem(prob, ad)
@@ -60,13 +61,9 @@ end
6061
function select_reverse_mode_autodiff(
6162
prob::AbstractNonlinearProblem, ad::AbstractADType; warn_check_mode::Bool = true)
6263
if warn_check_mode && !(ADTypes.mode(ad) isa ADTypes.ReverseMode) &&
63-
!(ADTypes.mode(ad) isa ADTypes.ForwardOrReverseMode)
64-
if !is_finite_differences_backend(ad)
65-
@warn "The chosen AD backend $(ad) is not a reverse mode AD. Use with caution."
66-
else
67-
@warn "The chosen AD backend $(ad) is a finite differences backend. This might \
68-
be slow and inaccurate. Use with caution."
69-
end
64+
!(ADTypes.mode(ad) isa ADTypes.ForwardOrReverseMode) &&
65+
!is_finite_differences_backend(ad)
66+
@warn "The chosen AD backend $(ad) is not a reverse mode AD. Use with caution."
7067
end
7168
if incompatible_backend_and_problem(prob, ad)
7269
adₙ = select_reverse_mode_autodiff(prob, nothing; warn_check_mode)

lib/SimpleNonlinearSolve/src/SimpleNonlinearSolve.jl

+1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
module SimpleNonlinearSolve
22

33
using Accessors: @reset
4+
using BracketingNonlinearSolve: BracketingNonlinearSolve
45
using CommonSolve: CommonSolve, solve, init, solve!
56
using ConcreteStructs: @concrete
67
using FastClosures: @closure

src/core/generalized_first_order.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ function GeneralizedFirstOrderAlgorithm{concrete_jac, name}(;
6060
max_shrink_times::Int = typemax(Int)) where {concrete_jac, name}
6161
return GeneralizedFirstOrderAlgorithm{concrete_jac, name}(
6262
linesearch, trustregion, descent, max_shrink_times,
63-
autodiff, jvp_autodiff, vjp_autodiff)
63+
autodiff, vjp_autodiff, jvp_autodiff)
6464
end
6565

6666
concrete_jac(::GeneralizedFirstOrderAlgorithm{CJ}) where {CJ} = CJ

0 commit comments

Comments
 (0)