Wait for all communicators in miniapp_eigensolver

eth-cscs · May 27, 2024 · 16ad0a4 · 16ad0a4
1 parent 2752d88
commit 16ad0a4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/miniapp/miniapp_eigensolver.cpp b/miniapp/miniapp_eigensolver.cpp
@@ -161,6 +161,11 @@ struct EigensolverMiniapp {
 
       // 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();