From 22e9d1139a541fb9e004cd30370eeca46112435d Mon Sep 17 00:00:00 2001 From: Yimin Jin Date: Wed, 27 Nov 2024 15:53:57 +0800 Subject: [PATCH] add team barriers to CellAction in matrix_free/tools.h --- include/deal.II/matrix_free/tools.h | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/include/deal.II/matrix_free/tools.h b/include/deal.II/matrix_free/tools.h index bf4a9cb361..01c9b19ecd 100644 --- 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) -- 2.39.5