Expose the nnzJ argument in KN_set_cb_grad. (#226)

jump-dev · Jul 20, 2022 · f61bed3 · f61bed3
1 parent e7be793
commit f61bed3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/callbacks.jl b/src/callbacks.jl
@@ -332,6 +332,7 @@ function KN_set_cb_grad(
     cb::CallbackContext,
     gradcallback;
     nV::Integer=KN_DENSE,
+    nnzJ::Integer=(iszero(KNITRO.KN_get_number_cons(m)) ? KNLONG(0) : KNITRO.KN_DENSE_COLMAJOR),
     objGradIndexVars=C_NULL,
     jacIndexCons=C_NULL,
     jacIndexVars=C_NULL,
@@ -344,7 +345,6 @@ function KN_set_cb_grad(
         @assert (objGradIndexVars != C_NULL) && (length(objGradIndexVars) == nV)
     end
 
-    nnzJ = KNLONG(0)
     if jacIndexCons != C_NULL && jacIndexVars != C_NULL
         @assert length(jacIndexCons) == length(jacIndexVars)
         nnzJ = KNLONG(length(jacIndexCons))
@@ -372,7 +372,7 @@ function KN_set_cb_grad(
         cb,
         nV,
         objGradIndexVars,
-        nnzJ,
+        KNLONG(nnzJ),
         jacIndexCons,
         jacIndexVars,
         c_grad_g