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

Small refactor

Browse Source
This commit is contained in:
Stan Gerald Moore
2023-05-01 13:08:01 -06:00
parent 059601734f
commit fd65000afc
9 changed files with 67 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/KOKKOS/modify_kokkos.cpp
View File

@ -396,14 +396,14 @@ void ModifyKokkos::final_integrate()
fused initial and final integrate call, only for relevant fixes
------------------------------------------------------------------------- */
void ModifyKokkos::fused_integrate()
void ModifyKokkos::fused_integrate(int vflag)
{
for (int i = 0; i < n_final_integrate; i++) {
atomKK->sync(fix[list_final_integrate[i]]->execution_space,
fix[list_final_integrate[i]]->datamask_read);
int prev_auto_sync = lmp->kokkos->auto_sync;
if (!fix[list_final_integrate[i]]->kokkosable) lmp->kokkos->auto_sync = 1;
fix[list_final_integrate[i]]->fused_integrate();
fix[list_final_integrate[i]]->fused_integrate(vflag);
lmp->kokkos->auto_sync = prev_auto_sync;
atomKK->modified(fix[list_final_integrate[i]]->execution_space,
fix[list_final_integrate[i]]->datamask_modify);