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

Fix issues with CUDA-aware MPI

Browse Source
This commit is contained in:
Stan Moore
2019-12-09 15:21:47 -07:00
parent e68824a392
commit 1b9214c3e9
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/KOKKOS/kokkos.cpp
View File

@ -220,7 +220,7 @@ KokkosLMP::KokkosLMP(LAMMPS *lmp, int narg, char **arg) : Pointers(lmp)
int nmpi = 0;
MPI_Comm_size(world,&nmpi);
if (nmpi > 0) {
if (nmpi > 1) {
#if defined(MPI_VERSION) && (MPI_VERSION > 2)
// Check for IBM Spectrum MPI
@ -397,6 +397,8 @@ void KokkosLMP::accelerator(int narg, char **arg)
} else error->all(FLERR,"Illegal package kokkos command");
}
#ifdef KOKKOS_ENABLE_CUDA
// if "cuda/aware off" and "comm device", change to "comm host"
if (!cuda_aware_flag) {
@ -431,6 +433,8 @@ void KokkosLMP::accelerator(int narg, char **arg)
}
}
#endif
// set newton flags
// set neighbor binsize, same as neigh_modify command