107 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
		
		
			
		
	
	
			107 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
|   | dnl @synopsis ACX_MPI([ACTION-IF-FOUND[, ACTION-IF-NOT-FOUND]]) | ||
|  | dnl @summary figure out how to compile/link code with MPI | ||
|  | dnl @category InstalledPackages | ||
|  | dnl | ||
|  | dnl This macro tries to find out how to compile programs that | ||
|  | dnl use MPI (Message Passing Interface), a standard API for | ||
|  | dnl parallel process communication (see http://www-unix.mcs.anl.gov/mpi/) | ||
|  | dnl | ||
|  | dnl On success, it sets the MPICC, MPICXX, or MPIF77 output variable to | ||
|  | dnl the name of the MPI compiler, depending upon the current language. | ||
|  | dnl (This may just be $CC/$CXX/$F77, but is more often something like | ||
|  | dnl mpicc/mpiCC/mpif77.)  It also sets MPILIBS to any libraries that are | ||
|  | dnl needed for linking MPI (e.g. -lmpi, if a special MPICC/MPICXX/MPIF77 | ||
|  | dnl was not found). | ||
|  | dnl | ||
|  | dnl If you want to compile everything with MPI, you should set: | ||
|  | dnl | ||
|  | dnl     CC="$MPICC" #OR# CXX="$MPICXX" #OR# F77="$MPIF77" | ||
|  | dnl     LIBS="$MPILIBS $LIBS" | ||
|  | dnl | ||
|  | dnl NOTE: The above assumes that you will use $CC (or whatever) | ||
|  | dnl       for linking as well as for compiling.  (This is the | ||
|  | dnl       default for automake and most Makefiles.) | ||
|  | dnl | ||
|  | dnl The user can force a particular library/compiler by setting the | ||
|  | dnl MPICC/MPICXX/MPIF77 and/or MPILIBS environment variables. | ||
|  | dnl | ||
|  | dnl ACTION-IF-FOUND is a list of shell commands to run if an MPI | ||
|  | dnl library is found, and ACTION-IF-NOT-FOUND is a list of commands | ||
|  | dnl to run it if it is not found.  If ACTION-IF-FOUND is not specified, | ||
|  | dnl the default action will define HAVE_MPI. | ||
|  | dnl | ||
|  | dnl @version 2005-09-02 | ||
|  | dnl @license GPLWithACException | ||
|  | dnl @author Steven G. Johnson <stevenj@alum.mit.edu> | ||
|  | 
 | ||
|  | AC_DEFUN([ACX_MPI], [ | ||
|  | AC_PREREQ(2.50) dnl for AC_LANG_CASE | ||
|  | 
 | ||
|  | AC_LANG_CASE([C], [ | ||
|  | 	AC_REQUIRE([AC_PROG_CC]) | ||
|  | 	AC_ARG_VAR(MPICC,[MPI C compiler command]) | ||
|  | 	AC_CHECK_PROGS(MPICC, mpicc hcc mpcc mpcc_r mpxlc cmpicc, $CC) | ||
|  | 	acx_mpi_save_CC="$CC" | ||
|  | 	CC="$MPICC" | ||
|  | 	AC_SUBST(MPICC) | ||
|  | ], | ||
|  | [C++], [ | ||
|  | 	AC_REQUIRE([AC_PROG_CXX]) | ||
|  | 	AC_ARG_VAR(MPICXX,[MPI C++ compiler command]) | ||
|  | 	AC_CHECK_PROGS(MPICXX, mpic++ mpiCC mpicxx mpCC hcp mpxlC mpxlC_r cmpic++, $CXX) | ||
|  | 	acx_mpi_save_CXX="$CXX" | ||
|  | 	CXX="$MPICXX" | ||
|  | 	AC_SUBST(MPICXX) | ||
|  | ], | ||
|  | [Fortran 77], [ | ||
|  | 	AC_REQUIRE([AC_PROG_F77]) | ||
|  | 	AC_ARG_VAR(MPIF77,[MPI Fortran compiler command]) | ||
|  | 	AC_CHECK_PROGS(MPIF77, mpif77 hf77 mpxlf mpf77 mpif90 mpf90 mpxlf90 mpxlf95 mpxlf_r cmpifc cmpif90c, $F77) | ||
|  | 	acx_mpi_save_F77="$F77" | ||
|  | 	F77="$MPIF77" | ||
|  | 	AC_SUBST(MPIF77) | ||
|  | ]) | ||
|  | 
 | ||
|  | if test x = x"$MPILIBS"; then | ||
|  | 	AC_LANG_CASE([C], [AC_CHECK_FUNC(MPI_Init, [MPILIBS=" "])], | ||
|  | 		[C++], [AC_CHECK_FUNC(MPI_Init, [MPILIBS=" "])], | ||
|  | 		[Fortran 77], [AC_MSG_CHECKING([for MPI_Init]) | ||
|  | 			AC_TRY_LINK([],[      call MPI_Init], [MPILIBS=" " | ||
|  | 				AC_MSG_RESULT(yes)], [AC_MSG_RESULT(no)])]) | ||
|  | fi | ||
|  | if test x = x"$MPILIBS"; then | ||
|  | 	AC_CHECK_LIB(mpi, MPI_Init, [MPILIBS="-lmpi"]) | ||
|  | fi | ||
|  | if test x = x"$MPILIBS"; then | ||
|  | 	AC_CHECK_LIB(mpich, MPI_Init, [MPILIBS="-lmpich"]) | ||
|  | fi | ||
|  | 
 | ||
|  | dnl We have to use AC_TRY_COMPILE and not AC_CHECK_HEADER because the | ||
|  | dnl latter uses $CPP, not $CC (which may be mpicc). | ||
|  | AC_LANG_CASE([C], [if test x != x"$MPILIBS"; then | ||
|  | 	AC_MSG_CHECKING([for mpi.h]) | ||
|  | 	AC_TRY_COMPILE([#include <mpi.h>],[],[AC_MSG_RESULT(yes)], [MPILIBS="" | ||
|  | 		AC_MSG_RESULT(no)]) | ||
|  | fi], | ||
|  | [C++], [if test x != x"$MPILIBS"; then | ||
|  | 	AC_MSG_CHECKING([for mpi.h]) | ||
|  | 	AC_TRY_COMPILE([#include <mpi.h>],[],[AC_MSG_RESULT(yes)], [MPILIBS="" | ||
|  | 		AC_MSG_RESULT(no)]) | ||
|  | fi]) | ||
|  | 
 | ||
|  | AC_LANG_CASE([C], [CC="$acx_mpi_save_CC"], | ||
|  | 	[C++], [CXX="$acx_mpi_save_CXX"], | ||
|  | 	[Fortran 77], [F77="$acx_mpi_save_F77"]) | ||
|  | 
 | ||
|  | AC_SUBST(MPILIBS) | ||
|  | 
 | ||
|  | # Finally, execute ACTION-IF-FOUND/ACTION-IF-NOT-FOUND: | ||
|  | if test x = x"$MPILIBS"; then | ||
|  |         $2 | ||
|  |         : | ||
|  | else | ||
|  |         ifelse([$1],,[AC_DEFINE(HAVE_MPI,1,[Define if you have the MPI library.])],[$1]) | ||
|  |         : | ||
|  | fi | ||
|  | ])dnl ACX_MPI |