gpu fix

jgillis · Jul 15, 2024 · 176029b · 176029b
1 parent 8a5b886
commit 176029b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/MadNLP_C.jl b/src/MadNLP_C.jl
@@ -462,10 +462,10 @@ Base.@ccallable function madnlp_c_solve(s::Ptr{MadnlpCSolver})::Cint
     Guvar = CuArray{Float64}(undef, nvar)
     Glcon = CuArray{Float64}(undef, ncon)
     Gucon = CuArray{Float64}(undef, ncon)
-    Gnzj_i = CuArray{Int64}(undef, nnzj)
-    Gnzj_j = CuArray{Int64}(undef, nnzj)
-    Gnzh_i = CuArray{Int64}(undef, nnzh)
-    Gnzh_j = CuArray{Int64}(undef, nnzh)
+    Gnzj_i = CuArray{Int64}(undef, nlp_interface.nnzj)
+    Gnzj_j = CuArray{Int64}(undef, nlp_interface.nnzj)
+    Gnzh_i = CuArray{Int64}(undef, nlp_interface.nnzh)
+    Gnzh_j = CuArray{Int64}(undef, nlp_interface.nnzh)
 
     copyto!(Gx0, x0)
     copyto!(Gy0, y0)