1from __future__ import generators 2import config.base 3 4class Configure(config.base.Configure): 5 def __init__(self, framework): 6 config.base.Configure.__init__(self, framework) 7 self.headerPrefix = '' 8 self.substPrefix = '' 9 return 10 11 def __str1__(self): 12 if not hasattr(self, 'memalign'): 13 return '' 14 return ' Memory alignment from malloc(): ' + self.memalign + ' bytes\n' 15 16 def setupHelp(self, help): 17 import nargs 18 help.addArgument('PETSc', '-with-memalign=<4,8,16,32,64>', nargs.Arg(None, '16', 'Specify alignment of arrays allocated by PETSc')) 19 return 20 21 def setupDependencies(self, framework): 22 config.base.Configure.setupDependencies(self, framework) 23 self.types = framework.require('config.types', self) 24 self.languages = framework.require('PETSc.options.languages', self) 25 self.compilers = framework.require('config.compilers', self) 26 return 27 28 def configureMemAlign(self): 29 '''Choose memory alignment''' 30 # Intel/AMD cache lines are 64 bytes, default page sizes are usually 4kB. It would be pretty silly to want that much alignment by default. 31 valid = ['4', '8', '16', '32', '64', '128', '256', '512', '1024', '2048', '4096', '8192'] 32 self.memalign = self.framework.argDB['with-memalign'] 33 if self.memalign in valid: 34 self.addDefine('MEMALIGN', self.memalign) 35 else: 36 raise RuntimeError('--with-memalign must be in' + str(valid)) 37 self.logPrint('Memory alignment is ' + self.memalign) 38 return 39 40 def configure(self): 41 self.executeTest(self.configureMemAlign) 42 return 43