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