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

Need local capture for lamdas on GPUs

Browse Source
This commit is contained in:
Stan Moore
2025-01-21 16:13:52 -07:00
parent fbfdcbbac1
commit 4fdd85df4f
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/KOKKOS/fix_langevin_kokkos.cpp
View File

@ -153,6 +153,9 @@ void FixLangevinKokkos<DeviceType>::setup(int vflag)
auto rmass = atomKK->k_rmass.view<DeviceType>();
auto mass = atomKK->k_mass.view<DeviceType>();
auto type = atomKK->k_type.view<DeviceType>();
auto groupbit = this->groupbit;
auto gjfa = this->gjfa;
auto gjfsib = this->gjfsib;
if (atom->rmass) {
atomKK->sync(execution_space,V_MASK|F_MASK|MASK_MASK|RMASS_MASK);
@ -256,6 +259,7 @@ void FixLangevinKokkos<DeviceType>::setup(int vflag)
auto rmass = atomKK->k_rmass.view<DeviceType>();
auto mass = atomKK->k_mass.view<DeviceType>();
auto type = atomKK->k_type.view<DeviceType>();
auto groupbit = this->groupbit;
k_lv.template sync<DeviceType>();
auto l_lv = d_lv;