Update the function for the lagrangian

JuliaSmoothOptimizers · Dec 3, 2024 · b6a1e3e · b6a1e3e
1 parent deda5fa
commit b6a1e3e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/enzyme.jl b/src/enzyme.jl
@@ -216,8 +216,12 @@ function SparseEnzymeADHessian(
   cx = similar(x0, ncon)
   grad = similar(x0)
   function ℓ(x, y, obj_weight, cx)
-    c!(cx, x)
-    obj_weight * f(x) + sum(cx[i] * y[i] for i = 1:ncon)
+    res = obj_weight * f(x)
+    if ncon == 0
+      c!(cx, x)
+      res += sum(cx[i] * y[i] for i = 1:ncon)
+    end
+    return res
   end
 
   return SparseEnzymeADHessian(