ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Kokkos library in LAMMPS to v2.7.24

Browse Source
This commit is contained in:
Stan Moore
2018-11-12 15:16:26 -07:00
parent 1651a21f92
commit b3f08b38a2
320 changed files with 42934 additions and 1993 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lib/kokkos/example/tutorial/Hierarchical_Parallelism/01_thread_teams/thread_teams.cpp
View File

@ -81,13 +81,22 @@ int main(int narg, char* args[]) {
Kokkos::initialize(narg,args);
// Launch 12 teams of the maximum number of threads per team
const team_policy policy( 12 , team_policy::team_size_max( hello_world() ) );
int sum = 0;
Kokkos::parallel_reduce( policy , hello_world() , sum );
const int team_size_max = team_policy(1,1).team_size_max(hello_world(), Kokkos::ParallelReduceTag());
const team_policy policy_a( 12 , team_size_max );
// The result will be 12*team_policy::team_size_max( hello_world())
printf("Result %i\n",sum);
int sum = 0;
Kokkos::parallel_reduce( policy_a , hello_world() , sum );
// The result will be 12*team_size_max
printf("Result A: %i == %i\n",sum, team_size_max*12);
// In practice it is often better to let Kokkos decide on the team_size
const team_policy policy_b( 12 , Kokkos::AUTO );
Kokkos::parallel_reduce( policy_b , hello_world() , sum );
// The result will be 12*policy_b.team_size_recommended( hello_world(), Kokkos::ParallelReduceTag())
const int team_size_recommended = policy_b.team_size_recommended( hello_world(), Kokkos::ParallelReduceTag());
printf("Result B: %i %i\n",sum, team_size_recommended*12);
Kokkos::finalize();
}

3
lib/kokkos/example/tutorial/Hierarchical_Parallelism/02_nested_parallel_for/nested_parallel_for.cpp
View File

@ -79,7 +79,8 @@ int main(int narg, char* args[]) {
Kokkos::initialize(narg,args);
// Launch 3 teams of the maximum number of threads per team
const team_policy policy( 3 , team_policy::team_size_max( hello_world() ) );
const int team_size_max = team_policy(3,1).team_size_max( hello_world(), Kokkos::ParallelReduceTag());
const team_policy policy( 3 , team_size_max );
int sum = 0;
Kokkos::parallel_reduce( policy , hello_world() , sum );