Wait for all communicators in miniapp_gen_eigensolver

eth-cscs · May 28, 2024 · f5c6fcb · f5c6fcb
1 parent 16ad0a4
commit f5c6fcb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/miniapp/miniapp_gen_eigensolver.cpp b/miniapp/miniapp_gen_eigensolver.cpp
@@ -141,6 +141,11 @@ struct GenEigensolverMiniapp {
 
       // wait and barrier for all ranks
       eigenvectors.waitLocalTiles();
+      for (std::size_t i = 0; i < comm_grid.num_pipelines(); ++i) {
+        sync_wait(comm_grid.full_communicator_pipeline().exclusive());
+        sync_wait(comm_grid.row_communicator_pipeline().exclusive());
+        sync_wait(comm_grid.col_communicator_pipeline().exclusive());
+      }
       DLAF_MPI_CHECK_ERROR(MPI_Barrier(world));
       double elapsed_time = timeit.elapsed();