CuDSS synchronize added (MadNLP#314)

* cudss synchronize * solve synchronize removed
zavalab · Mar 27, 2024 · de04e2a · de04e2a
1 parent b4d1935
commit de04e2a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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