Add flag for Kokkos computes
This commit is contained in:
@ -52,6 +52,7 @@ template<class DeviceType>
|
|||||||
ComputeOrientOrderAtomKokkos<DeviceType>::ComputeOrientOrderAtomKokkos(LAMMPS *lmp, int narg, char **arg) :
|
ComputeOrientOrderAtomKokkos<DeviceType>::ComputeOrientOrderAtomKokkos(LAMMPS *lmp, int narg, char **arg) :
|
||||||
ComputeOrientOrderAtom(lmp, narg, arg)
|
ComputeOrientOrderAtom(lmp, narg, arg)
|
||||||
{
|
{
|
||||||
|
kokkosable = 1;
|
||||||
atomKK = (AtomKokkos *) atom;
|
atomKK = (AtomKokkos *) atom;
|
||||||
execution_space = ExecutionSpaceFromDevice<DeviceType>::space;
|
execution_space = ExecutionSpaceFromDevice<DeviceType>::space;
|
||||||
datamask_read = EMPTY_MASK;
|
datamask_read = EMPTY_MASK;
|
||||||
|
|||||||
@ -31,6 +31,7 @@ template<class DeviceType>
|
|||||||
ComputeTempKokkos<DeviceType>::ComputeTempKokkos(LAMMPS *lmp, int narg, char **arg) :
|
ComputeTempKokkos<DeviceType>::ComputeTempKokkos(LAMMPS *lmp, int narg, char **arg) :
|
||||||
ComputeTemp(lmp, narg, arg)
|
ComputeTemp(lmp, narg, arg)
|
||||||
{
|
{
|
||||||
|
kokkosable = 1;
|
||||||
atomKK = (AtomKokkos *) atom;
|
atomKK = (AtomKokkos *) atom;
|
||||||
execution_space = ExecutionSpaceFromDevice<DeviceType>::space;
|
execution_space = ExecutionSpaceFromDevice<DeviceType>::space;
|
||||||
|
|
||||||
|
|||||||
@ -102,6 +102,7 @@ Compute::Compute(LAMMPS *lmp, int narg, char **arg) :
|
|||||||
datamask_modify = ALL_MASK;
|
datamask_modify = ALL_MASK;
|
||||||
|
|
||||||
copymode = 0;
|
copymode = 0;
|
||||||
|
kokkosable = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* ---------------------------------------------------------------------- */
|
/* ---------------------------------------------------------------------- */
|
||||||
|
|||||||
@ -91,7 +91,7 @@ class Compute : protected Pointers {
|
|||||||
ExecutionSpace execution_space;
|
ExecutionSpace execution_space;
|
||||||
unsigned int datamask_read,datamask_modify;
|
unsigned int datamask_read,datamask_modify;
|
||||||
|
|
||||||
int copymode;
|
int copymode,kokkosable;
|
||||||
|
|
||||||
Compute(class LAMMPS *, int, char **);
|
Compute(class LAMMPS *, int, char **);
|
||||||
virtual ~Compute();
|
virtual ~Compute();
|
||||||
|
|||||||
Reference in New Issue
Block a user