1#!/usr/bin/env python 2from __future__ import generators 3import user 4import config.base 5 6class Configure(config.base.Configure): 7 def __init__(self, framework): 8 config.base.Configure.__init__(self, framework) 9 self.headerPrefix = '' 10 self.substPrefix = '' 11 return 12 13 def __str1__(self): 14 if not hasattr(self, 'memalign'): 15 return '' 16 return ' Memory alignment: ' + self.memalign + '\n' 17 18 def setupHelp(self, help): 19 import nargs 20 help.addArgument('PETSc', '-with-memalign=<4,8,16,32,64>', nargs.Arg(None, '16', 'Specify alignment of arrays allocated by PETSc')) 21 return 22 23 def setupDependencies(self, framework): 24 config.base.Configure.setupDependencies(self, framework) 25 self.types = framework.require('config.types', self) 26 self.languages = framework.require('PETSc.options.languages', self) 27 self.compilers = framework.require('config.compilers', self) 28 return 29 30 def configureMemAlign(self): 31 '''Choose alignment''' 32 # 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. 33 valid = ['4', '8', '16', '32', '64', '128', '256', '512', '1024', '2048', '4096', '8192'] 34 self.memalign = self.framework.argDB['with-memalign'] 35 if self.memalign in valid: 36 self.addDefine('MEMALIGN', self.memalign) 37 else: 38 raise RuntimeError('--with-memalign must be in' + str(valid)) 39 self.framework.logPrint('Memory alignment is ' + self.memalign) 40 return 41 42 def configure(self): 43 self.executeTest(self.configureMemAlign) 44 return 45