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

Add more missing Kokkos data movement

Browse Source
This commit is contained in:
Stan Moore
2022-06-28 15:04:10 -06:00
committed by Axel Kohlmeyer
parent 67597722d5
commit 53806d4601
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/KOKKOS/min_kokkos.cpp
View File

@ -495,6 +495,7 @@ double MinKokkos::energy_force(int resetflag)
if (force->newton) {
comm->reverse_comm();
timer->stamp(Timer::COMM);
atomKK->sync(Device,F_MASK);
}
// update per-atom minimization variables stored by pair styles
@ -567,7 +568,7 @@ void MinKokkos::force_clear()
}
});
}
atomKK->modified(Device,F_MASK);
atomKK->modified(Device,F_MASK|TORQUE_MASK);
}
/* ----------------------------------------------------------------------