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

Merge pull request #2460 from stanmoore1/kk_bugfix

Browse Source 
Kokkos bugfix for pair forward comm
This commit is contained in:
Axel Kohlmeyer
2020-10-28 12:46:46 -04:00
committed by GitHub
parent 5872b05e2f 99c99e86e0
commit 72f4650b52
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/KOKKOS/comm_kokkos.cpp
View File

@ -394,6 +394,7 @@ void CommKokkos::forward_comm_pair_device(Pair *pair)
{
int iswap,n;
MPI_Request request;
DAT::tdual_xfloat_1d k_buf_tmp;
int nsize = pair->comm_forward;
KokkosBase* pairKKBase = dynamic_cast<KokkosBase*>(pair);
@ -441,11 +442,12 @@ void CommKokkos::forward_comm_pair_device(Pair *pair)
k_buf_recv_pair.modify<LMPHostType>();
k_buf_recv_pair.sync<DeviceType>();
}
} else k_buf_recv_pair = k_buf_send_pair;
k_buf_tmp = k_buf_recv_pair;
} else k_buf_tmp = k_buf_send_pair;
// unpack buffer
pairKKBase->unpack_forward_comm_kokkos(recvnum[iswap],firstrecv[iswap],k_buf_recv_pair);
pairKKBase->unpack_forward_comm_kokkos(recvnum[iswap],firstrecv[iswap],k_buf_tmp);
DeviceType().fence();
}
}