jianmu
/
lammps
forked from lijiext/lammps
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #659 from stanmoore1/kk_border_comm 

Enhance Kokkos communication logic
This commit is contained in:
Steve Plimpton 2017-09-22 13:47:44 -06:00 committed by GitHub
parent ee487ef4aa 3ed39ca844
commit bcde318ccb
2 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/KOKKOS/comm_kokkos.cpp
View File

@ -379,8 +379,8 @@ void CommKokkos::exchange()
if(!exchange_comm_classic) {
static int print = 1;
if(print && comm->me==0) {
error->warning(FLERR,"Fixes cannot send data in Kokkos communication, "
"switching to classic communication");
error->warning(FLERR,"Fixes cannot yet send data in Kokkos communication, "
"switching to classic communication");
}
print = 0;
exchange_comm_classic = true;
@ -620,6 +620,19 @@ void CommKokkos::exchange_device()
void CommKokkos::borders()
{
if (!exchange_comm_classic) {
static int print = 1;
if (style != SINGLE || bordergroup || ghost_velocity) {
if (print && comm->me==0) {
error->warning(FLERR,"Required border comm not yet implemented in Kokkos communication, "
"switching to classic communication");
}
print = 0;
exchange_comm_classic = true;
}
}
if (!exchange_comm_classic) {
if (exchange_comm_on_host) borders_device<LMPHostType>();
else borders_device<LMPDeviceType>();

9
src/KOKKOS/comm_kokkos.h
View File

@ -82,9 +82,14 @@ E: Ghost velocity forward comm not yet implemented with Kokkos
This is a current restriction.
W: Fixes cannot send data in Kokkos communication, switching to classic communication
W: Fixes cannot yet send data in Kokkos communication, switching to classic communication
This is current restriction with Kokkos.
This is a current restriction with Kokkos.
W: Required border comm not yet implemented in Kokkos communication, switching to classic communication
There are various limitations in the communication options supported
by Kokkos.
E: Required border comm not yet implemented with Kokkos