Merge pull request #17894 from YiminJin/add-barriers-to-CellAction

add team barriers to CellAction in matrix_free/tools.h
dealii · Dec 1, 2024 · ef665a2 · ef665a2
2 parents 8cf211a + 22e9d11
commit ef665a2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/include/deal.II/matrix_free/tools.h b/include/deal.II/matrix_free/tools.h
@@ -1429,13 +1429,17 @@ 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), [&] {
         for (unsigned int i = 0; i < dofs_per_cell; ++i)
           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)