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