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