Add flag for Kokkos computes

This commit is contained in:
Stan Moore
2020-02-24 15:43:29 -07:00
parent 29075c8257
commit c505058c5c
4 changed files with 4 additions and 1 deletions

View File

@ -52,6 +52,7 @@ template<class DeviceType>
ComputeOrientOrderAtomKokkos<DeviceType>::ComputeOrientOrderAtomKokkos(LAMMPS *lmp, int narg, char **arg) :
ComputeOrientOrderAtom(lmp, narg, arg)
{
kokkosable = 1;
atomKK = (AtomKokkos *) atom;
execution_space = ExecutionSpaceFromDevice<DeviceType>::space;
datamask_read = EMPTY_MASK;

View File

@ -31,6 +31,7 @@ template<class DeviceType>
ComputeTempKokkos<DeviceType>::ComputeTempKokkos(LAMMPS *lmp, int narg, char **arg) :
ComputeTemp(lmp, narg, arg)
{
kokkosable = 1;
atomKK = (AtomKokkos *) atom;
execution_space = ExecutionSpaceFromDevice<DeviceType>::space;

View File

@ -102,6 +102,7 @@ Compute::Compute(LAMMPS *lmp, int narg, char **arg) :
datamask_modify = ALL_MASK;
copymode = 0;
kokkosable = 0;
}
/* ---------------------------------------------------------------------- */

View File

@ -91,7 +91,7 @@ class Compute : protected Pointers {
ExecutionSpace execution_space;
unsigned int datamask_read,datamask_modify;
int copymode;
int copymode,kokkosable;
Compute(class LAMMPS *, int, char **);
virtual ~Compute();