Change default binsize for Kokkos if running on GPUs
This commit is contained in:
@ -30,6 +30,7 @@
|
|||||||
#include "style_nstencil.h"
|
#include "style_nstencil.h"
|
||||||
#include "style_npair.h"
|
#include "style_npair.h"
|
||||||
#include "style_ntopo.h"
|
#include "style_ntopo.h"
|
||||||
|
#include "comm.h"
|
||||||
|
|
||||||
using namespace LAMMPS_NS;
|
using namespace LAMMPS_NS;
|
||||||
|
|
||||||
@ -359,6 +360,14 @@ void NeighborKokkos::modify_mol_intra_grow_kokkos(){
|
|||||||
k_ex_mol_intra.modify<LMPHostType>();
|
k_ex_mol_intra.modify<LMPHostType>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* ---------------------------------------------------------------------- */
|
||||||
|
void NeighborKokkos::set_binsize_kokkos() {
|
||||||
|
if (!binsizeflag && lmp->kokkos->ngpu > 0) {
|
||||||
|
binsize_user = cutneighmax;
|
||||||
|
binsizeflag = 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/* ---------------------------------------------------------------------- */
|
/* ---------------------------------------------------------------------- */
|
||||||
|
|
||||||
void NeighborKokkos::init_topology() {
|
void NeighborKokkos::init_topology() {
|
||||||
|
|||||||
@ -87,6 +87,7 @@ class NeighborKokkos : public Neighbor {
|
|||||||
void modify_ex_group_grow_kokkos();
|
void modify_ex_group_grow_kokkos();
|
||||||
void modify_mol_group_grow_kokkos();
|
void modify_mol_group_grow_kokkos();
|
||||||
void modify_mol_intra_grow_kokkos();
|
void modify_mol_intra_grow_kokkos();
|
||||||
|
void set_binsize_kokkos();
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -471,6 +471,9 @@ void Neighbor::init()
|
|||||||
error->warning(FLERR,"Neighbor exclusions used with KSpace solver "
|
error->warning(FLERR,"Neighbor exclusions used with KSpace solver "
|
||||||
"may give inconsistent Coulombic energies");
|
"may give inconsistent Coulombic energies");
|
||||||
|
|
||||||
|
if (lmp->kokkos)
|
||||||
|
set_binsize_kokkos();
|
||||||
|
|
||||||
// ------------------------------------------------------------------
|
// ------------------------------------------------------------------
|
||||||
// create pairwise lists
|
// create pairwise lists
|
||||||
// one-time call to init_styles() to scan style files and setup
|
// one-time call to init_styles() to scan style files and setup
|
||||||
|
|||||||
@ -233,6 +233,7 @@ class Neighbor : protected Pointers {
|
|||||||
virtual void init_ex_bit_kokkos() {}
|
virtual void init_ex_bit_kokkos() {}
|
||||||
virtual void init_ex_mol_bit_kokkos() {}
|
virtual void init_ex_mol_bit_kokkos() {}
|
||||||
virtual void grow_ex_mol_intra_kokkos() {}
|
virtual void grow_ex_mol_intra_kokkos() {}
|
||||||
|
virtual void set_binsize_kokkos() {}
|
||||||
};
|
};
|
||||||
|
|
||||||
namespace NeighConst {
|
namespace NeighConst {
|
||||||
|
|||||||
Reference in New Issue
Block a user