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