From: Yimin Jin <jinym@cea-igp.ac.cn>
Date: Wed, 27 Nov 2024 07:53:57 +0000 (+0800)
Subject: add team barriers to CellAction in matrix_free/tools.h
X-Git-Url: https://gitweb.dealii.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=refs%2Fpull%2F17894%2Fhead;p=dealii.git

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 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)