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

Temporarily disable team policy for ghost neigh list build due to known bug

Browse Source
This commit is contained in:
Stan Gerald Moore
2023-06-16 10:52:21 -06:00
parent 9673c22cc7
commit 2d3bbd2e72
1 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/KOKKOS/npair_kokkos.cpp
View File

@ -240,22 +240,25 @@ void NPairKokkos<DeviceType,HALF,NEWTON,GHOST,TRI,SIZE>::build(NeighList *list_)
// assumes newton off
NPairKokkosBuildFunctorGhost<DeviceType,HALF> f(data,atoms_per_bin * 5 * sizeof(X_FLOAT) * factor);
#ifdef LMP_KOKKOS_GPU
if (ExecutionSpaceFromDevice<DeviceType>::space == Device) {
int team_size = atoms_per_bin*factor;
int team_size_max = Kokkos::TeamPolicy<DeviceType>(team_size,Kokkos::AUTO).team_size_max(f,Kokkos::ParallelForTag());
if (team_size <= team_size_max) {
Kokkos::TeamPolicy<DeviceType> config((mbins+factor-1)/factor,team_size);
Kokkos::parallel_for(config, f);
} else { // fall back to flat method
f.sharedsize = 0;
Kokkos::parallel_for(nall, f);
}
} else
Kokkos::parallel_for(nall, f);
#else
// temporarily disable team policy for ghost due to known bug
//#ifdef LMP_KOKKOS_GPU
// if (ExecutionSpaceFromDevice<DeviceType>::space == Device) {
// int team_size = atoms_per_bin*factor;
// int team_size_max = Kokkos::TeamPolicy<DeviceType>(team_size,Kokkos::AUTO).team_size_max(f,Kokkos::ParallelForTag());
// if (team_size <= team_size_max) {
// Kokkos::TeamPolicy<DeviceType> config((mbins+factor-1)/factor,team_size);
// Kokkos::parallel_for(config, f);
// } else { // fall back to flat method
// f.sharedsize = 0;
// Kokkos::parallel_for(nall, f);
// }
// } else
// Kokkos::parallel_for(nall, f);
//#else
Kokkos::parallel_for(nall, f);
#endif
//#endif
} else {
if (SIZE) {
NPairKokkosBuildFunctorSize<DeviceType,HALF,NEWTON,TRI> f(data,atoms_per_bin * 6 * sizeof(X_FLOAT) * factor);