ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fuse some Kokkos kernels to reduce launch latency for small systems

Browse Source
This commit is contained in:
Stan Moore
2023-04-28 14:40:59 -06:00
parent 4a608dced6
commit 235372d6e8
14 changed files with 185 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/KOKKOS/modify_kokkos.cpp
View File

@ -392,6 +392,25 @@ void ModifyKokkos::final_integrate()
}
}
/* ----------------------------------------------------------------------
2nd half of integrate call, only for relevant fixes
------------------------------------------------------------------------- */
void ModifyKokkos::fused_integrate()
{
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();
lmp->kokkos->auto_sync = prev_auto_sync;
atomKK->modified(fix[list_final_integrate[i]]->execution_space,
fix[list_final_integrate[i]]->datamask_modify);
}
}
/* ----------------------------------------------------------------------
end-of-timestep call, only for relevant fixes
only call fix->end_of_step() on timesteps that are multiples of nevery