tx258/lammps-gran-kokkos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor Kokkos AtomVec

Browse Source
This commit is contained in:
Stan Moore
2022-12-12 17:34:18 -07:00
parent 2421f9098c
commit 6d8e7e1ece
38 changed files with 121 additions and 8761 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/KOKKOS/atom_kokkos.cpp
View File

@ -104,23 +104,23 @@ AtomKokkos::~AtomKokkos()
void AtomKokkos::sync(const ExecutionSpace space, unsigned int mask)
{
if (space == Device && lmp->kokkos->auto_sync) ((AtomVecKokkos *) avec)->modified(Host, mask);
if (space == Device && lmp->kokkos->auto_sync) avecKK->modified(Host, mask);
((AtomVecKokkos *) avec)->sync(space, mask);
avecKK->sync(space, mask);
}
/* ---------------------------------------------------------------------- */
void AtomKokkos::modified(const ExecutionSpace space, unsigned int mask)
{
((AtomVecKokkos *) avec)->modified(space, mask);
avecKK->modified(space, mask);
if (space == Device && lmp->kokkos->auto_sync) ((AtomVecKokkos *) avec)->sync(Host, mask);
if (space == Device && lmp->kokkos->auto_sync) avecKK->sync(Host, mask);
}
void AtomKokkos::sync_overlapping_device(const ExecutionSpace space, unsigned int mask)
{
((AtomVecKokkos *) avec)->sync_overlapping_device(space, mask);
avecKK->sync_overlapping_device(space, mask);
}
/* ---------------------------------------------------------------------- */
@ -380,5 +380,8 @@ AtomVec *AtomKokkos::new_avec(const std::string &style, int trysuffix, int &sfla
{
AtomVec *avec = Atom::new_avec(style, trysuffix, sflag);
if (!avec->kokkosable) error->all(FLERR, "KOKKOS package requires a kokkos enabled atom_style");
avecKK = dynamic_cast<AtomVecKokkos*>(avec);
return avec;
}