fix

Author: Vaibhavdixit02

ext/OptimizationReverseDiffExt.jl

@@ -90,7 +90,7 @@ function Optimization.instantiate_function(f, cache::Optimization.ReInitCache,
 
     if f.hess === nothing
         hess = function (res, θ, args...)
-            res .= ReverseDiff.gradient(x -> _f(x, args...), θ)
+            ReverseDiff.hessian!(res, x -> _f(x, args...), θ)
         end
     else
         hess = (H, θ, args...) -> f.hess(H, θ, cache.p, args...)

test/ADtests.jl

@@ -84,6 +84,9 @@ H3 = [Array{Float64}(undef, 2, 2)]
 optprob.cons_h(H3, x0)
 @test H3 == [[2.0 0.0; 0.0 2.0]]
 
+G2 = Array{Float64}(undef, 2)
+H2 = Array{Float64}(undef, 2, 2)
+
 optf = OptimizationFunction(rosenbrock, Optimization.AutoEnzyme(), cons = con2_c)
 optprob = Optimization.instantiate_function(optf, x0, Optimization.AutoEnzyme(),
     nothing, 2)
@@ -101,6 +104,9 @@ H3 = [Array{Float64}(undef, 2, 2), Array{Float64}(undef, 2, 2)]
 optprob.cons_h(H3, x0)
 H3 == [[2.0 0.0; 0.0 2.0], [-0.0 1.0; 1.0 0.0]]
 
+G2 = Array{Float64}(undef, 2)
+H2 = Array{Float64}(undef, 2, 2)
+
 optf = OptimizationFunction(rosenbrock, Optimization.AutoReverseDiff(), cons = con2_c)
 optprob = Optimization.instantiate_function(optf, x0, Optimization.AutoReverseDiff(),
     nothing, 2)
@@ -118,6 +124,9 @@ H3 = [Array{Float64}(undef, 2, 2), Array{Float64}(undef, 2, 2)]
 optprob.cons_h(H3, x0)
 H3 == [[2.0 0.0; 0.0 2.0], [-0.0 1.0; 1.0 0.0]]
 
+G2 = Array{Float64}(undef, 2)
+H2 = Array{Float64}(undef, 2, 2)
+
 optf = OptimizationFunction(rosenbrock, Optimization.AutoZygote(), cons = con2_c)
 optprob = Optimization.instantiate_function(optf, x0, Optimization.AutoZygote(),
     nothing, 2)