Add threshold for using Kokkos teams (149a57f3) · Commits · 郑智淋 / lammps

src/KOKKOS/pair_kokkos.h

+5 −0

Original line number	Diff line number	Diff line
		@@ -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;