jianmu
/
lammps
forked from lijiext/lammps
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add threshold for using Kokkos teams

This commit is contained in:
Stan Moore 2019-04-08 15:21:42 -06:00
parent 7f2c81657b
commit 149a57f3ae
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/KOKKOS/pair_kokkos.h
View File

@ -866,6 +866,11 @@ int GetTeamSize(FunctorStyle& functor, int team_size, int vector_length) {
template<class PairStyle, unsigned NEIGHFLAG, class Specialisation>
EV_FLOAT pair_compute_neighlist (PairStyle* fpair, typename Kokkos::Impl::enable_if<(NEIGHFLAG&PairStyle::EnabledNeighFlags) != 0, NeighListKokkos<typename PairStyle::device_type>*>::type list) {
EV_FLOAT ev;
if (!fpair->lmp->kokkos->team_flag_set)
if (list->inum <= 16384)
fpair->lmp->kokkos->team_flag = 1;
if (fpair->lmp->kokkos->team_flag) {
int vector_length = 8;
int atoms_per_team = 32;