Merge branch 'master' into ar/NonLinearProgram

Author: joaquimg

src/moi_wrapper.jl

@@ -587,6 +587,9 @@ end
 
 function empty_input_sensitivities!(model::Optimizer)
     empty!(model.input_cache)
+    if model.diff !== nothing
+        empty!(model.diff.model.input_cache)
+    end
     return
 end
 

test/parameters.jl

@@ -671,7 +671,6 @@ function test_diff_errors_POI()
     return
 end
 
-
 function test_diff_errors()
     model = Model(
         () -> DiffOpt.diff_optimizer(
@@ -719,7 +718,55 @@ function test_diff_errors()
         DiffOpt.ReverseConstraintFunction(),
         cons,
     )
+    return
+end
+
+function is_empty(cache::DiffOpt.InputCache)
+    return isempty(cache.dx) &&
+           isempty(cache.scalar_constraints) &&
+           isempty(cache.vector_constraints) &&
+           cache.objective === nothing
+end
+
+# Credit to @klamike
+function test_empty_cache()
+    m = Model(
+        () -> DiffOpt.diff_optimizer(
+            HiGHS.Optimizer;
+            with_parametric_opt_interface = true,
+        ),
+    )
+    @variable(m, x)
+    @variable(m, p ∈ Parameter(1.0))
+    @variable(m, q ∈ Parameter(2.0))
+    @constraint(m, x ≥ p)
+    @constraint(m, x ≥ q)
+    @objective(m, Min, x)
+    optimize!(m)
+    @assert is_solved_and_feasible(m)
+
+    function get_sensitivity(m, xᵢ, pᵢ)
+        DiffOpt.empty_input_sensitivities!(m)
+        @test is_empty(unsafe_backend(m).optimizer.input_cache)
+        if !isnothing(unsafe_backend(m).optimizer.diff) &&
+           !isnothing(unsafe_backend(m).optimizer.diff.model.input_cache)
+            @test is_empty(unsafe_backend(m).optimizer.diff.model.input_cache)
+        end
+        MOI.set(
+            m,
+            DiffOpt.ForwardConstraintSet(),
+            ParameterRef(pᵢ),
+            Parameter(1.0),
+        )
+        DiffOpt.forward_differentiate!(m)
+        return MOI.get(m, DiffOpt.ForwardVariablePrimal(), xᵢ)
+    end
 
+    sp1 = get_sensitivity(m, x, p)
+    sp2 = get_sensitivity(m, x, q)
+    sp3 = get_sensitivity(m, x, p)
+    @test sp1 ≈ sp3
+    @test sp2 ≠ sp3
     return
 end