add team barriers to CellAction in matrix_free/tools.h

diff --git a/include/deal.II/matrix_free/tools.h b/include/deal.II/matrix_free/tools.h
index bf4a9cb3614590deebe7c173b496b813750e0421..01c9b19ecd0476e625d904d13f1f6b5283bd0925 100644 (file)
--- a/include/deal.II/matrix_free/tools.h
+++ b/include/deal.II/matrix_free/tools.h
@@ -1429,6 +1429,8 @@ namespace MatrixFreeTools
 
           Kokkos::single(Kokkos::PerTeam(shared_data->team_member),
                          [&] { diagonal[i] = fe_eval.get_dof_value(i); });
+
+          shared_data->team_member.team_barrier();
         }
 
       Kokkos::single(Kokkos::PerTeam(shared_data->team_member), [&] {
@@ -1436,6 +1438,8 @@ namespace MatrixFreeTools
           fe_eval.submit_dof_value(diagonal[i], i);
       });
 
+      shared_data->team_member.team_barrier();
+
       // We need to do the same as distribute_local_to_global but without
       // constraints since we have already taken care of them earlier
       if (gpu_data->use_coloring)