diff --git a/src/accelerator_kokkos.h b/src/accelerator_kokkos.h index 5d7124e15a..2b2ba1a957 100644 --- a/src/accelerator_kokkos.h +++ b/src/accelerator_kokkos.h @@ -20,16 +20,15 @@ #ifdef LMP_KOKKOS #include "kokkos.h" +#include "atom_kokkos.h" +#include "comm_kokkos.h" +#include "modify_kokkos.h" #else // dummy interface to KOKKOS // needed for compiling when KOKKOS is not installed -//#include "comm.h" -//#include "modify.h" -//#include "verlet.h" - namespace LAMMPS_NS { class KokkosLMP { @@ -41,18 +40,18 @@ class KokkosLMP { void accelerator(int, char **) {} }; +class AtomKokkos : public Atom { + public: + AtomKokkos(class LAMMPS *lmp) : Atom(lmp) {} + ~AtomKokkos() {} +}; + class CommKokkos : public Comm { public: CommKokkos(class LAMMPS *lmp) : Comm(lmp) {} ~CommKokkos() {} }; -class DomainKokkos : public Domain { - public: - DomainKokkos(class LAMMPS *lmp) : Domain(lmp) {} - ~DomainKokkos() {} -}; - class NeighborKokkos : public Neighbor { public: NeighborKokkos(class LAMMPS *lmp) : Neighbor(lmp) {} @@ -65,12 +64,6 @@ class ModifyKokkos : public Modify { ~ModifyKokkos() {} }; -class VerletKokkos : public Verlet { - public: - VerletKokkos(class LAMMPS *lmp, int narg, char **arg) : Verlet(lmp,narg,arg) {} - ~VerletKokkos() {} -}; - } #endif