Skip to content

Commit 71856b2

Browse files
committed
initial sketch
1 parent 93f058d commit 71856b2

1 file changed

Lines changed: 63 additions & 0 deletions

File tree

src/jump_wrapper.jl

Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
"""
2+
"""
3+
function diff_model(
4+
optimizer_constructor;
5+
method = nothing,
6+
with_parametric_opt_interface::Bool = false,
7+
with_bridge_type = Float64,
8+
with_cache::Bool = true,
9+
)
10+
inner = diff_optimizer(
11+
optimizer_constructor;
12+
method = method,
13+
with_parametric_opt_interface = with_parametric_opt_interface,
14+
with_bridge_type = with_bridge_type,
15+
with_cache = with_cache,
16+
)
17+
return JuMP.direct_model(inner)
18+
end
19+
20+
# nonlinear_diff_model
21+
# conic_diff_model
22+
# quadratic_diff_model
23+
24+
"""
25+
"""
26+
function set_forward_parameter(
27+
model::JuMP.Model,
28+
variable::JuMP.VariableRef,
29+
value::Number,
30+
)
31+
return MOI.set(
32+
model,
33+
DiffOpt.ForwardConstraintSet(),
34+
ParameterRef(variable),
35+
value,
36+
)
37+
end
38+
39+
"""
40+
"""
41+
function get_reverse_parameter(model::JuMP.Model, variable::JuMP.VariableRef)
42+
return MOI.get(
43+
model,
44+
DiffOpt.ReverseConstraintSet(),
45+
ParameterRef(variable),
46+
)
47+
end
48+
49+
"""
50+
"""
51+
function set_reverse_variable(
52+
model::JuMP.Model,
53+
variable::JuMP.VariableRef,
54+
value::Number,
55+
)
56+
return MOI.set(model, DiffOpt.ReverseVariablePrimal(), variable, value)
57+
end
58+
59+
"""
60+
"""
61+
function get_forward_variable(model::JuMP.Model, variable::JuMP.VariableRef)
62+
return MOI.get(model, DiffOpt.ForwardVariablePrimal(), variable)
63+
end

0 commit comments

Comments
 (0)