Skip to content

Commit f228263

Browse files
SebastianM-CSebastianM-C
committed
reorder fields in OptimizationCache for cleaner dispatches
Since we always dispatch on the optimizer, it makes much more sense to have this as the first argument (and first type parameter).
1 parent 7c28259 commit f228263

File tree

19 files changed

+159
-621
lines changed

19 files changed

+159
-621
lines changed

‎lib/OptimizationAuglag/src/OptimizationAuglag.jl‎

Lines changed: 3 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -59,32 +59,7 @@ function __map_optimizer_args(cache::OptimizationBase.OptimizationCache, opt::Au
5959
return mapped_args
6060
end
6161

62-
function SciMLBase.__solve(cache::OptimizationCache{
63-
F,
64-
RC,
65-
LB,
66-
UB,
67-
LC,
68-
UC,
69-
S,
70-
O,
71-
D,
72-
P,
73-
C
74-
}) where {
75-
F,
76-
RC,
77-
LB,
78-
UB,
79-
LC,
80-
UC,
81-
S,
82-
O <:
83-
AugLag,
84-
D,
85-
P,
86-
C
87-
}
62+
function SciMLBase.__solve(cache::OptimizationCache{O}) where {O <: AugLag}
8863
maxiters = OptimizationBase._check_and_convert_maxiters(cache.solver_args.maxiters)
8964

9065
local x
@@ -121,7 +96,8 @@ function SciMLBase.__solve(cache::OptimizationCache{
12196
if cache.callback(opt_state, x...)
12297
error("Optimization halted by callback.")
12398
end
124-
return x[1] + sum(@. λ * cons_tmp[eq_inds] + ρ / 2 * (cons_tmp[eq_inds] .^ 2)) + 1 / (2 * ρ) * sum((max.(Ref(0.0), μ .+ (ρ .* cons_tmp[ineq_inds]))) .^ 2)
99+
return x[1] + sum(@. λ * cons_tmp[eq_inds] + ρ / 2 * (cons_tmp[eq_inds] .^ 2)) +
100+
1 / (2 * ρ) * sum((max.(Ref(0.0), μ .+ (ρ .* cons_tmp[ineq_inds]))) .^ 2)
125101
end
126102

127103
prev_eqcons = zero(λ)

‎lib/OptimizationBBO/src/OptimizationBBO.jl‎

Lines changed: 7 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
11
module OptimizationBBO
22

33
using Reexport
4-
import OptimizationBase
5-
import OptimizationBase: SciMLBase
6-
import BlackBoxOptim
7-
import SciMLBase: MultiObjectiveOptimizationFunction
4+
using OptimizationBase
5+
using SciMLBase
6+
using BlackBoxOptim: BlackBoxOptim
87

98
abstract type BBO end
109

@@ -36,19 +35,19 @@ function decompose_trace(opt::BlackBoxOptim.OptRunController, progress)
3635
if iszero(max_time)
3736
# we stop at either convergence or max_steps
3837
n_steps = BlackBoxOptim.num_steps(opt)
39-
Base.@logmsg(Base.LogLevel(-1), msg, progress=n_steps/maxiters,
38+
Base.@logmsg(Base.LogLevel(-1), msg, progress=n_steps / maxiters,
4039
_id=:OptimizationBBO)
4140
else
4241
# we stop at either convergence or max_time
4342
elapsed = BlackBoxOptim.elapsed_time(opt)
44-
Base.@logmsg(Base.LogLevel(-1), msg, progress=elapsed/max_time,
43+
Base.@logmsg(Base.LogLevel(-1), msg, progress=elapsed / max_time,
4544
_id=:OptimizationBBO)
4645
end
4746
end
4847
return BlackBoxOptim.best_candidate(opt)
4948
end
5049

51-
function __map_optimizer_args(prob::OptimizationBase.OptimizationCache, opt::BBO;
50+
function __map_optimizer_args(prob::OptimizationCache, opt::BBO;
5251
callback = nothing,
5352
maxiters::Union{Number, Nothing} = nothing,
5453
maxtime::Union{Number, Nothing} = nothing,
@@ -96,32 +95,7 @@ function map_objective(obj::BlackBoxOptim.IndexedTupleFitness)
9695
obj.orig
9796
end
9897

99-
function SciMLBase.__solve(cache::OptimizationBase.OptimizationCache{
100-
F,
101-
RC,
102-
LB,
103-
UB,
104-
LC,
105-
UC,
106-
S,
107-
O,
108-
D,
109-
P,
110-
C
111-
}) where {
112-
F,
113-
RC,
114-
LB,
115-
UB,
116-
LC,
117-
UC,
118-
S,
119-
O <:
120-
BBO,
121-
D,
122-
P,
123-
C
124-
}
98+
function SciMLBase.__solve(cache::OptimizationCache{O}) where {O <: BBO}
12599
function _cb(trace)
126100
if cache.callback === OptimizationBase.DEFAULT_CALLBACK
127101
cb_call = false

‎lib/OptimizationCMAEvolutionStrategy/src/OptimizationCMAEvolutionStrategy.jl‎

Lines changed: 3 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@ SciMLBase.requireshessian(::CMAEvolutionStrategyOpt) = false
2121
SciMLBase.requiresconsjac(::CMAEvolutionStrategyOpt) = false
2222
SciMLBase.requiresconshess(::CMAEvolutionStrategyOpt) = false
2323

24-
function __map_optimizer_args(prob::OptimizationBase.OptimizationCache, opt::CMAEvolutionStrategyOpt;
24+
function __map_optimizer_args(
25+
prob::OptimizationBase.OptimizationCache, opt::CMAEvolutionStrategyOpt;
2526
callback = nothing,
2627
maxiters::Union{Number, Nothing} = nothing,
2728
maxtime::Union{Number, Nothing} = nothing,
@@ -53,32 +54,7 @@ function __map_optimizer_args(prob::OptimizationBase.OptimizationCache, opt::CMA
5354
return mapped_args
5455
end
5556

56-
function SciMLBase.__solve(cache::OptimizationBase.OptimizationCache{
57-
F,
58-
RC,
59-
LB,
60-
UB,
61-
LC,
62-
UC,
63-
S,
64-
O,
65-
D,
66-
P,
67-
C
68-
}) where {
69-
F,
70-
RC,
71-
LB,
72-
UB,
73-
LC,
74-
UC,
75-
S,
76-
O <:
77-
CMAEvolutionStrategyOpt,
78-
D,
79-
P,
80-
C
81-
}
57+
function SciMLBase.__solve(cache::OptimizationCache{O}) where {O <: CMAEvolutionStrategyOpt}
8258
local x, cur, state
8359

8460
function _cb(opt, y, fvals, perm)

‎lib/OptimizationEvolutionary/src/OptimizationEvolutionary.jl‎

Lines changed: 5 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,9 @@ SciMLBase.allowsconstraints(opt::Evolutionary.AbstractOptimizer) = true
1010
SciMLBase.supports_opt_cache_interface(opt::Evolutionary.AbstractOptimizer) = true
1111
end
1212
@static if isdefined(OptimizationBase, :supports_opt_cache_interface)
13-
OptimizationBase.supports_opt_cache_interface(opt::Evolutionary.AbstractOptimizer) = true
13+
function OptimizationBase.supports_opt_cache_interface(opt::Evolutionary.AbstractOptimizer)
14+
true
15+
end
1416
end
1517
SciMLBase.requiresgradient(opt::Evolutionary.AbstractOptimizer) = false
1618
SciMLBase.requiresgradient(opt::Evolutionary.NSGA2) = false
@@ -76,32 +78,8 @@ function __map_optimizer_args(cache::OptimizationBase.OptimizationCache,
7678
return Evolutionary.Options(; mapped_args...)
7779
end
7880

79-
function SciMLBase.__solve(cache::OptimizationBase.OptimizationCache{
80-
F,
81-
RC,
82-
LB,
83-
UB,
84-
LC,
85-
UC,
86-
S,
87-
O,
88-
D,
89-
P,
90-
C
91-
}) where {
92-
F,
93-
RC,
94-
LB,
95-
UB,
96-
LC,
97-
UC,
98-
S,
99-
O <:
100-
Evolutionary.AbstractOptimizer,
101-
D,
102-
P,
103-
C
104-
}
81+
function SciMLBase.__solve(cache::OptimizationCache{O}) where {O <:
82+
Evolutionary.AbstractOptimizer}
10583
local x, cur, state
10684

10785
function _cb(trace)

‎lib/OptimizationGCMAES/src/OptimizationGCMAES.jl‎

Lines changed: 1 addition & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -61,32 +61,7 @@ function SciMLBase.__init(prob::SciMLBase.OptimizationProblem,
6161
kwargs...)
6262
end
6363

64-
function SciMLBase.__solve(cache::OptimizationBase.OptimizationCache{
65-
F,
66-
RC,
67-
LB,
68-
UB,
69-
LC,
70-
UC,
71-
S,
72-
O,
73-
D,
74-
P,
75-
C
76-
}) where {
77-
F,
78-
RC,
79-
LB,
80-
UB,
81-
LC,
82-
UC,
83-
S,
84-
O <:
85-
GCMAESOpt,
86-
D,
87-
P,
88-
C
89-
}
64+
function SciMLBase.__solve(cache::OptimizationCache{O}) where {O <: GCMAESOpt}
9065
local x
9166
local G = similar(cache.u0)
9267

‎lib/OptimizationLBFGSB/src/OptimizationLBFGSB.jl‎

Lines changed: 3 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -78,32 +78,7 @@ function __map_optimizer_args(cache::OptimizationBase.OptimizationCache, opt::LB
7878
return mapped_args
7979
end
8080

81-
function SciMLBase.__solve(cache::OptimizationBase.OptimizationCache{
82-
F,
83-
RC,
84-
LB,
85-
UB,
86-
LC,
87-
UC,
88-
S,
89-
O,
90-
D,
91-
P,
92-
C
93-
}) where {
94-
F,
95-
RC,
96-
LB,
97-
UB,
98-
LC,
99-
UC,
100-
S,
101-
O <:
102-
LBFGSB,
103-
D,
104-
P,
105-
C
106-
}
81+
function SciMLBase.__solve(cache::OptimizationCache{O}) where {O <: LBFGSB}
10782
maxiters = OptimizationBase._check_and_convert_maxiters(cache.solver_args.maxiters)
10883

10984
local x
@@ -142,7 +117,8 @@ function SciMLBase.__solve(cache::OptimizationBase.OptimizationCache{
142117
if cache.callback(opt_state, x...)
143118
error("Optimization halted by callback.")
144119
end
145-
return x[1] + sum(@. λ * cons_tmp[eq_inds] + ρ / 2 * (cons_tmp[eq_inds] .^ 2)) + 1 / (2 * ρ) * sum((max.(Ref(0.0), μ .+ (ρ .* cons_tmp[ineq_inds]))) .^ 2)
120+
return x[1] + sum(@. λ * cons_tmp[eq_inds] + ρ / 2 * (cons_tmp[eq_inds] .^ 2)) +
121+
1 / (2 * ρ) * sum((max.(Ref(0.0), μ .+ (ρ .* cons_tmp[ineq_inds]))) .^ 2)
146122
end
147123

148124
prev_eqcons = zero(λ)

‎lib/OptimizationMadNLP/src/OptimizationMadNLP.jl‎

Lines changed: 1 addition & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -269,11 +269,6 @@ function SciMLBase.allowsconsjvp(opt::MadNLPOptimizer)
269269
true
270270
end
271271

272-
function SciMLBase.__init(prob::SciMLBase.OptimizationProblem, opt::MadNLPOptimizer;
273-
kwargs...)
274-
return OptimizationCache(prob, opt; kwargs...)
275-
end
276-
277272
function map_madnlp_status(status::MadNLP.Status)
278273
if status in [
279274
MadNLP.SOLVE_SUCCEEDED,
@@ -407,31 +402,7 @@ function __map_optimizer_args(cache,
407402
)
408403
end
409404

410-
function SciMLBase.__solve(cache::OptimizationBase.OptimizationCache{
411-
F,
412-
RC,
413-
LB,
414-
UB,
415-
LC,
416-
UC,
417-
S,
418-
O,
419-
D,
420-
P,
421-
C
422-
}) where {
423-
F,
424-
RC,
425-
LB,
426-
UB,
427-
LC,
428-
UC,
429-
S,
430-
O <: MadNLPOptimizer,
431-
D,
432-
P,
433-
C
434-
}
405+
function SciMLBase.__solve(cache::OptimizationCache{O}) where {O <: MadNLPOptimizer}
435406
maxiters = OptimizationBase._check_and_convert_maxiters(cache.solver_args.maxiters)
436407
maxtime = OptimizationBase._check_and_convert_maxtime(cache.solver_args.maxtime)
437408

0 commit comments

Comments
 (0)