diff --git a/lib/MadNLPGPU/src/cudss.jl b/lib/MadNLPGPU/src/cudss.jl index ea03d4df..ebafed14 100644 --- a/lib/MadNLPGPU/src/cudss.jl +++ b/lib/MadNLPGPU/src/cudss.jl @@ -63,6 +63,8 @@ function MadNLP.factorize!(M::CUDSSSolver) CUDSS.cudss_set(M.inner.matrix, SparseArrays.nonzeros(M.tril)) CUDSS.cudss("factorization", M.inner, M.x_gpu, M.b_gpu) + synchronize(CUDABackend()) + return M end