From fde6cae56df482ac80aae0846810096416784702 Mon Sep 17 00:00:00 2001 From: Stan Gerald Moore Date: Thu, 22 Jul 2021 08:44:32 -0600 Subject: [PATCH] Add logic for GPU-enabled MPICH --- src/KOKKOS/kokkos.cpp | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/KOKKOS/kokkos.cpp b/src/KOKKOS/kokkos.cpp index 2588bf5d49..aa435d5b0b 100644 --- a/src/KOKKOS/kokkos.cpp +++ b/src/KOKKOS/kokkos.cpp @@ -293,9 +293,14 @@ KokkosLMP::KokkosLMP(LAMMPS *lmp, int narg, char **arg) : Pointers(lmp) error->warning(FLERR,"MVAPICH2 'MV2_USE_CUDA' environment variable is not set. Disabling GPU-aware MPI"); // pure MPICH or some unsupported MPICH derivative #elif defined(MPICH) && !defined(MVAPICH2_VERSION) - if (me == 0) - error->warning(FLERR,"Detected MPICH. Disabling GPU-aware MPI"); + char* str; gpu_aware_flag = 0; + if ((str = getenv("MPICH_GPU_SUPPORT_ENABLED"))) + if ((strcmp(str,"1") == 0)) + gpu_aware_flag = 1; + + if (!gpu_aware_flag && me == 0) + error->warning(FLERR,"Detected MPICH. Disabling GPU-aware MPI"); #else if (me == 0) error->warning(FLERR,"Kokkos with CUDA, HIP, or SYCL assumes CUDA-aware MPI is available,"