fixed rocALUTION missing copy from device to cpu for MPI

ElmerCSC · Dec 16, 2024 · 5d621cc · 5d621cc
1 parent b7938ac
commit 5d621cc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/fem/src/rocalution.cpp b/fem/src/rocalution.cpp
@@ -459,8 +459,9 @@ extern "C" void ROCParallelSolve( int *gn, int *n, int *rows, int *cols, double
 
     ls->Solve(rhs, &x);
 
-//  x.CopyToData(x_out);
-
+    // move solution back from device to host
+    x.MoveToHost();
+
     for(i=0; i<*n; i++ ) x_out[i]=x[i];
 
     ls->Clear();