Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Showing content from https://kokkos.github.io/kokkos-core-wiki/API/core/policies/TeamThreadMDRange.html below:

TeamThreadMDRange - Kokkos documentation

Header File: <Kokkos_Core.hpp>

TeamThreadMDRange is a nested execution policy used inside of hierarchical parallelism.

template<class Rank, typename TeamHandle>
class TeamThreadMDRange¶

Constructor

TeamThreadMDRange(team, extent_1, extent_2, ...);¶

Splits the index range 0 to extent over the threads of the team, where extent is the backend-dependent rank that will be threaded

Parameters:

• team – TeamHandle to the calling team execution context

• extent_1, extent_2, ... – index range lengths of each rank

• Requirements

  • TeamHandle is a type that models TeamHandle

  • extent_1, extent_2, ... are ints

  • Every member thread of team must call the operation in the same branch, i.e. it is not legal to have some threads call this function in one branch, and the other threads of team call it in another branch

  • extent_i is such that i >= 2 && i <= 8 is true. For example:

    TeamThreadMDRange(team, 4);               // NOT OK, violates i>=2
    
    TeamThreadMDRange(team, 4,5);             // OK
    TeamThreadMDRange(team, 4,5,6);           // OK
    TeamThreadMDRange(team, 4,5,6,2,3,4,5,6); // OK, max num of extents allowed
    

Note that when used in parallel_reduce, the reduction is limited to a sum.

using TeamHandle = TeamPolicy<>::member_type;

parallel_for(TeamPolicy<>(N,AUTO),
  KOKKOS_LAMBDA (TeamHandle const& team) {

    int leagueRank = team.league_rank();

    auto range = TeamThreadMDRange<Rank<4>, TeamHandle>(team, n0, n1, n2, n3);

    parallel_for(range, [=](int i0, int i1, int i2, int i3) {
      A(leagueRank, i0, i1, i2, i3) = B(leagueRank, i1) + C(i1, i2, i3);
    });
    team.team_barrier();

    int teamSum = 0;
    parallel_reduce(range,
      [=](int i0, int i1, int i2, int i3, int& threadSum) {
        threadSum += D(leagueRank, i0, i1, i2, i3);
      }, teamSum
    );
    single(PerTeam(team), [&leagueSum, teamSum]() { leagueSum += teamSum; });
    A_rowSum[leagueRank] = leagueSum;
});

RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.4