1import config.package 2 3class Configure(config.package.Package): 4 def __init__(self, framework): 5 config.package.Package.__init__(self, framework) 6 self.version = '5.6.2' 7 self.minversion = '5.2.1' 8 self.versionname = 'MUMPS_VERSION' 9 self.requiresversion = 1 10 self.gitcommit = 'v'+self.version 11 self.download = ['https://graal.ens-lyon.fr/MUMPS/MUMPS_'+self.version+'.tar.gz', 12 'https://web.cels.anl.gov/projects/petsc/download/externalpackages/MUMPS_'+self.version+'.tar.gz'] 13 self.downloaddirnames = ['petsc-pkg-mumps','MUMPS'] 14 self.buildLanguages = ['C','FC'] 15 self.precisions = ['single','double'] 16 self.downloadonWindows= 1 17 self.hastests = 1 18 self.hastestsdatafiles= 1 19 return 20 21 def setupHelp(self, help): 22 import nargs 23 config.package.Package.setupHelp(self, help) 24 help.addArgument('MUMPS', '-with-mumps-serial', nargs.ArgBool(None, 0, 'Use serial build of MUMPS')) 25 help.addArgument('MUMPS', '-download-mumps-avoid-mpi-in-place', nargs.ArgBool(None, 0, 'Let MUMPS not use MPI_IN_PLACE. Since MUMPS-5.6.2, it can be used to avoid a bug in MPICH older than 4.0b1')) 26 return 27 28 def setupDependencies(self, framework): 29 config.package.Package.setupDependencies(self, framework) 30 self.flibs = framework.require('config.packages.flibs',self) 31 self.blasLapack = framework.require('config.packages.BlasLapack',self) 32 self.mpi = framework.require('config.packages.MPI',self) 33 self.metis = framework.require('config.packages.metis',self) 34 self.parmetis = framework.require('config.packages.parmetis',self) 35 self.ptscotch = framework.require('config.packages.PTScotch',self) 36 self.scalapack = framework.require('config.packages.scalapack',self) 37 self.hwloc = framework.require('config.packages.hwloc',self) 38 self.openmp = framework.require('config.packages.openmp',self) 39 self.scalartypes = framework.require('PETSc.options.scalarTypes',self) 40 if self.argDB['with-mumps-serial']: 41 self.deps = [self.blasLapack,self.flibs] 42 self.odeps = [self.metis,self.openmp] 43 else: 44 self.deps = [self.scalapack,self.mpi,self.blasLapack,self.flibs] 45 self.odeps = [self.metis,self.parmetis,self.ptscotch,self.hwloc,self.openmp] 46 return 47 48 def configureLibrary(self): 49 for arg in ['with-64-bit-blas-indices','known-64-bit-blas-indices']: 50 if self.argDB.get(arg): 51 raise RuntimeError('MUMPS cannot be used with %s' % arg) 52 if self.scalartypes.precision == 'single': 53 if self.scalartypes.scalartype == 'real': l = 's' 54 else: l = 'c' 55 else: 56 if self.scalartypes.scalartype == 'real': l = 'd' 57 else: l = 'z' 58 self.functions = [l+'mumps_c'] 59 self.includes = [l+'mumps_c.h'] 60 liblist_common = [['libmumps_common.a','libpord.a','libpthread.a'], 61 ['libmumps_common.a','libpord.a'], 62 ['libmumps_common.a','libpord.a','libmpiseq.a'], 63 ['libmumps_common.a','libpord.a','libpthread.a','libmpiseq.a']] 64 self.liblist = [] 65 for libc in liblist_common: 66 self.liblist.append(['lib'+l+'mumps.a'] + libc) 67 config.package.Package.configureLibrary(self) 68 69 def consistencyChecks(self): 70 config.package.Package.consistencyChecks(self) 71 if self.argDB['with-'+self.package] or self.argDB['download-'+self.package]: 72 if self.mpi.usingMPIUni and not self.argDB['with-mumps-serial']: 73 raise RuntimeError('Since you are building without MPI you must use --with-mumps-serial to install the correct MUMPS.') 74 if self.argDB['with-mumps-serial']: 75 if not self.mpi.usingMPIUni: 76 raise RuntimeError('Serial MUMPS version is only compatible with MPIUni\nReconfigure using --with-mpi=0') 77 return 78 79 def Install(self): 80 import os 81 82 if self.openmp.found: 83 # MUMPS has no make flags for turning on/off OpenMP it just uses it if it can 84 self.usesopenmp = 'yes' 85 # use OMP_NUM_THREADS to control the number of threads used 86 87 if not self.fortran.FortranDefineCompilerOption: 88 raise RuntimeError('Fortran compiler cannot handle preprocessing directives from command line.') 89 g = open(os.path.join(self.packageDir,'Makefile.inc'),'w') 90 g.write('LPORDDIR = $(topdir)/PORD/lib/\n') 91 g.write('IPORD = -I$(topdir)/PORD/include/\n') 92 g.write('LPORD = -L$(LPORDDIR) -lpord\n') 93 g.write('PLAT = \n') 94 orderingsc = '-Dpord' 95 orderingsf = self.fortran.FortranDefineCompilerOption+'pord' 96 # Disable threads on BGL 97 if self.libraries.isBGL(): 98 orderingsc += ' -DWITHOUT_PTHREAD' 99 if self.metis.found: 100 g.write('IMETIS = '+self.headers.toString(self.metis.include)+'\n') 101 g.write('LMETIS = '+self.libraries.toString(self.metis.lib)+'\n') 102 orderingsc += ' -Dmetis' 103 orderingsf += ' '+self.fortran.FortranDefineCompilerOption+'metis' 104 if self.parmetis.found: 105 g.write('IPARMETIS = '+self.headers.toString(self.parmetis.include)+'\n') 106 g.write('LPARMETIS = '+self.libraries.toString(self.parmetis.lib)+'\n') 107 orderingsc += ' -Dparmetis' 108 orderingsf += ' '+self.fortran.FortranDefineCompilerOption+'parmetis' 109 if self.ptscotch.found: 110 g.write('ISCOTCH = '+self.headers.toString(self.ptscotch.include)+'\n') 111 g.write('LSCOTCH = '+self.libraries.toString(self.ptscotch.lib)+'\n') 112 orderingsc += ' -Dscotch -Dptscotch' 113 orderingsf += ' '+self.fortran.FortranDefineCompilerOption+'scotch '+self.fortran.FortranDefineCompilerOption+'ptscotch' 114 115 g.write('ORDERINGSC = '+orderingsc+'\n') 116 g.write('ORDERINGSF = '+orderingsf+'\n') 117 g.write('LORDERINGS = $(LPARMETIS) $(LMETIS) $(LPORD) $(LSCOTCH)\n') 118 g.write('IORDERINGSC = $(IPARMETIS) $(IMETIS) $(IPORD) $(ISCOTCH)\n') 119 g.write('IORDERINGSF = $(ISCOTCH)\n') 120 121 g.write('RM = /bin/rm -f\n') 122 self.pushLanguage('C') 123 g.write('CC = '+self.getCompiler()+'\n') 124 g.write('OPTC = '+self.updatePackageCFlags(self.getCompilerFlags())+'\n') 125 g.write('OUTC = -o \n') 126 self.popLanguage() 127 if not self.fortran.fortranIsF90: 128 raise RuntimeError('Installing MUMPS requires a F90 compiler') 129 self.pushLanguage('FC') 130 g.write('FC = '+self.getCompiler()+'\n') 131 g.write('FL = '+self.getCompiler()+'\n') 132 g.write('OPTF = '+self.updatePackageFFlags(self.getCompilerFlags())+'\n') 133 if self.blasLapack.checkForRoutine('dgemmt'): 134 g.write('OPTF += -DGEMMT_AVAILABLE\n') 135 g.write('OUTF = -o \n') 136 self.popLanguage() 137 138 # set fortran name mangling 139 # this mangling information is for both BLAS and the Fortran compiler so cannot use the BlasLapack mangling flag 140 if self.compilers.fortranManglingDoubleUnderscore: 141 g.write('CDEFS = -DAdd__\n') 142 elif self.compilers.fortranMangling == 'underscore': 143 g.write('CDEFS = -DAdd_\n') 144 elif self.compilers.fortranMangling == 'caps': 145 g.write('CDEFS = -DUPPPER\n') 146 147 g.write('AR = '+self.setCompilers.AR+' '+self.setCompilers.AR_FLAGS+' \n') 148 g.write('LIBEXT = .'+self.setCompilers.AR_LIB_SUFFIX+'\n') 149 g.write('RANLIB = '+self.setCompilers.RANLIB+'\n') 150 g.write('SCALAP = '+self.libraries.toString(self.scalapack.lib)+'\n') 151 if not self.argDB['with-mumps-serial']: 152 g.write('INCPAR = '+self.headers.toString(self.mpi.include)+'\n') 153 g.write('LIBPAR = $(SCALAP) '+self.libraries.toString(self.mpi.lib)+'\n') 154 else: 155 g.write('INCPAR = -I../libseq\n') 156 g.write('INCSEQ = -I$(topdir)/libseq\n') 157 g.write('LIBSEQ = $(LAPACK) -L$(topdir)/libseq -lmpiseq\n') 158 g.write('LIBBLAS = '+self.libraries.toString(self.blasLapack.dlib)+'\n') 159 g.write('OPTL = '+self.getLinkerFlags()+'\n') 160 g.write('INCS = $(INCPAR)\n') 161 g.write('LIBS = $(LIBPAR)\n') 162 if self.argDB['with-mumps-serial']: 163 g.write('LIBSEQNEEDED = libseqneeded\n') 164 g.write('LIBS = $(LIBSEQ)\n') 165 else: 166 g.write('LIBSEQNEEDED =\n') 167 if self.openmp.found and self.hwloc.found: 168 g.write('LIBS += '+self.libraries.toString(self.hwloc.lib)+'\n') 169 g.write('OPTF += -DUSE_LIBHWLOC\n') 170 g.write('OPTC += -DUSE_LIBHWLOC\n') 171 # To avoid a bug related to MPI_IN_PLACE and old MPICH releases, see MR 4410 172 self.avoid_mpi_in_place = 0 173 if 'download-mumps-avoid-mpi-in-place' in self.framework.clArgDB: # user-provided value takes precedence 174 self.avoid_mpi_in_place = self.clArgDB['download-mumps-avoid-mpi-in-place'] 175 elif hasattr(self.mpi, 'mpich_numversion') and int(self.mpi.mpich_numversion) < 40000101: 176 self.avoid_mpi_in_place = 1 177 if self.avoid_mpi_in_place: 178 g.write('CDEFS += -DAVOID_MPI_IN_PLACE') # only take effect since mumps-5.6.2 179 self.addDefine('HAVE_MUMPS_AVOID_MPI_IN_PLACE', 1) 180 g.close() 181 if self.installNeeded('Makefile.inc'): 182 try: 183 output1,err1,ret1 = config.package.Package.executeShellCommand('make clean', cwd=self.packageDir, timeout=60, log = self.log) 184 except RuntimeError as e: 185 pass 186 try: 187 self.logPrintBox('Compiling MUMPS; this may take several minutes') 188 output2,err2,ret2 = config.package.Package.executeShellCommand(self.make.make_jnp+' prerequisites', cwd=self.packageDir, timeout=2500, log = self.log) 189 output3,err3,ret3 = config.package.Package.executeShellCommand(self.make.make_jnp+' all', cwd=os.path.join(self.packageDir,'src'), timeout=2500, log = self.log) 190 libDir = self.libDir 191 includeDir = os.path.join(self.installDir, self.includedir) 192 self.logPrintBox('Installing MUMPS; this may take several minutes') 193 output,err,ret = config.package.Package.executeShellCommandSeq( 194 ['mkdir -p '+libDir+' '+includeDir, 195 'cp -f lib/*.* '+libDir+'/.', 196 'cp -f include/*.* '+includeDir+'/.' 197 ], cwd=self.packageDir, timeout=60, log = self.log) 198 if self.argDB['with-mumps-serial']: 199 output,err,ret = config.package.Package.executeShellCommand(['cp', '-f', 'libseq/libmpiseq.a', libDir+'/.'], cwd=self.packageDir, timeout=60, log = self.log) 200 except RuntimeError as e: 201 self.logPrint('Error running make on MUMPS: '+str(e)) 202 raise RuntimeError('Error running make on MUMPS') 203 self.postInstall(output1+err1+output2+err2+output3+err3,'Makefile.inc') 204 return self.installDir 205 206