45 #ifndef KOKKOS_ACQUIRE_UNIQUE_TOKEN_IMPL_HPP 46 #define KOKKOS_ACQUIRE_UNIQUE_TOKEN_IMPL_HPP 48 #include <Kokkos_Core.hpp> 49 #include <Kokkos_UniqueToken.hpp> 53 template <
typename TeamPolicy>
54 KOKKOS_FUNCTION AcquireTeamUniqueToken<TeamPolicy>::AcquireTeamUniqueToken(
55 AcquireTeamUniqueToken<TeamPolicy>::token_type t, team_member_type team)
56 : my_token(t), my_team_acquired_val(team.team_scratch(0)), my_team(team) {
57 Kokkos::single(Kokkos::PerTeam(my_team),
58 [&]() { my_team_acquired_val() = my_token.acquire(); });
59 my_team.team_barrier();
61 my_acquired_val = my_team_acquired_val();
64 template <
typename TeamPolicy>
65 KOKKOS_FUNCTION AcquireTeamUniqueToken<TeamPolicy>::~AcquireTeamUniqueToken() {
66 my_team.team_barrier();
67 Kokkos::single(Kokkos::PerTeam(my_team),
68 [&]() { my_token.release(my_acquired_val); });
69 my_team.team_barrier();
75 #endif // KOKKOS_UNIQUE_TOKEN_HPP