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

Optimize FixNeighHistoryKokkos

Browse Source
This commit is contained in:
Stan Moore
2022-10-14 14:22:08 -06:00
parent bf42f06a39
commit 85726b7ce2
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/KOKKOS/fix_neigh_history_kokkos.cpp
View File

@ -201,6 +201,9 @@ void FixNeighHistoryKokkos<DeviceType>::post_neighbor()
copymode = 1;
Kokkos::deep_copy(d_firstflag,0);
Kokkos::deep_copy(d_firstvalue,0);
FixNeighHistoryKokkosPostNeighborFunctor<DeviceType> f(this);
Kokkos::parallel_for(inum,f);
@ -235,16 +238,6 @@ void FixNeighHistoryKokkos<DeviceType>::post_neighbor_item(const int &ii) const
for (int k = 0; k < dnum; k++) {
d_firstvalue(i, dnum*jj+k) = d_valuepartner(i, dnum*m+k);
}
} else {
d_firstflag(i,jj) = 0;
for (int k = 0; k < dnum; k++) {
d_firstvalue(i, dnum*jj+k) = 0;
}
}
} else {
d_firstflag(i,jj) = 0;
for (int k = 0; k < dnum; k++) {
d_firstvalue(i, dnum*jj+k) = 0;
}
}
}