Merge pull request #2298 from stanmoore1/mv2_var (cffc6672) · Commits · 郑智淋 / lammps

src/KOKKOS/kokkos.cpp

+2 −2

Original line number	Diff line number	Diff line
		@@ -253,13 +253,13 @@ KokkosLMP::KokkosLMP(LAMMPS lmp, int narg, char *arg) : Pointers(lmp)
		#if defined(MPICH) && defined(MVAPICH2_VERSION)
		char* str;
		cuda_aware_flag = 0;
		if ((str = getenv("MV2_ENABLE_CUDA")))
		if ((str = getenv("MV2_USE_CUDA")))
		if ((strcmp(str,"1") == 0))
		cuda_aware_flag = 1;

		if (!cuda_aware_flag)
		if (me == 0)
		error->warning(FLERR,"MVAPICH2 'MV2_ENABLE_CUDA' environment variable is not set. Disabling CUDA-aware MPI");
		error->warning(FLERR,"MVAPICH2 'MV2_USE_CUDA' environment variable is not set. Disabling CUDA-aware MPI");
		// pure MPICH or some unsupported MPICH derivative
		#elif defined(MPICH) && !defined(MVAPICH2_VERSION)
		if (me == 0)