Commit c505058c authored by Stan Moore's avatar Stan Moore
Browse files

Add flag for Kokkos computes

parent 29075c82

src/KOKKOS/compute_orientorder_atom_kokkos.cpp

+1 −0
Original line number Diff line number Diff line
@@ -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;

src/KOKKOS/compute_temp_kokkos.cpp

+1 −0
Original line number Diff line number Diff line
@@ -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;

src/compute.cpp

+1 −0
Original line number Diff line number Diff line
@@ -102,6 +102,7 @@ Compute::Compute(LAMMPS *lmp, int narg, char **arg) : 
  datamask_modify = ALL_MASK;



  copymode = 0;

  kokkosable = 0;

}



/* ---------------------------------------------------------------------- */

src/compute.h

+1 −1
Original line number Diff line number Diff line
@@ -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();

CRA Git | Maintained and supported by SUSTech CRA and CCSE