1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
#!/usr/bin/env python
from __future__ import generators
import user
import config.base
class Configure(config.base.Configure):
def __init__(self, framework):
config.base.Configure.__init__(self, framework)
self.headerPrefix = ''
self.substPrefix = ''
return
def __str1__(self):
desc = []
if hasattr(self, 'scalartype'):
desc.append(' Scalar type: ' + self.scalartype)
if hasattr(self, 'precision'):
desc.append(' Precision: ' + self.precision)
return '\n'.join(desc)+'\n'
def setupHelp(self, help):
import nargs
help.addArgument('PETSc', '-with-precision=<single,double,__float128>', nargs.Arg(None, 'double', 'Specify numerical precision'))
help.addArgument('PETSc', '-with-scalar-type=<real or complex>', nargs.Arg(None, 'real', 'Specify real or complex numbers'))
help.addArgument('PETSc', '-with-mixed-precision=<bool>', nargs.ArgBool(None, 0, 'Allow single precision linear solve'))
return
def setupDependencies(self, framework):
config.base.Configure.setupDependencies(self, framework)
self.types = framework.require('config.types', self)
self.languages = framework.require('PETSc.utilities.languages', self)
self.compilers = framework.require('config.compilers', self)
self.libraries = framework.require('config.libraries',self)
return
def configureScalarType(self):
'''Choose between real and complex numbers'''
self.scalartype = self.framework.argDB['with-scalar-type'].lower()
if self.scalartype == 'complex':
self.addDefine('USE_COMPLEX', '1')
if self.languages.clanguage == 'C' and not self.types.c99_complex:
raise RuntimeError('C Compiler provided doest not support C99 complex')
if self.languages.clanguage == 'Cxx' and not self.types.cxx_complex:
raise RuntimeError('Cxx compiler provided does not support std::complex')
elif not self.scalartype == 'real':
raise RuntimeError('--with-scalar-type must be real or complex')
self.addDefine('USE_SCALAR_'+self.scalartype.upper(), '1')
self.framework.logPrint('Scalar type is '+str(self.scalartype))
# On apple isinf() and isnan() do not work when <complex> is included
self.pushLanguage(self.languages.clanguage)
if self.scalartype == 'complex' and self.languages.clanguage == 'Cxx':
if self.checkLink('#include <math.h>\n#include <complex>\n','double b = 2.0;int a = isnan(b);\n'):
self.addDefine('HAVE_ISNAN',1)
if self.checkLink('#include <math.h>\n#include <complex>\n','double b = 2.0;int a = isinf(b);\n'):
self.addDefine('HAVE_ISINF',1)
if self.checkLink('#include <math.h>\n#include <complex>\n','double b = 2.0;int a = _isnan(b);\n'):
self.addDefine('HAVE__ISNAN',1)
if self.checkLink('#include <math.h>\n#include <complex>\n','double b = 2.0;int a = _finite(b);\n'):
self.addDefine('HAVE__FINITE',1)
else:
if self.checkLink('#include <math.h>\n','double b = 2.0; int a = isnan(b);\n'):
self.addDefine('HAVE_ISNAN',1)
if self.checkLink('#include <math.h>\n','double b = 2.0; int a = isinf(b);\n'):
self.addDefine('HAVE_ISINF',1)
if self.checkLink('#include <math.h>\n','double b = 2.0;int a = _isnan(b);\n'):
self.addDefine('HAVE__ISNAN',1)
if self.checkLink('#include <math.h>\n','double b = 2.0;int a = _finite(b);\n'):
self.addDefine('HAVE__FINITE',1)
self.popLanguage()
return
def configurePrecision(self):
'''Set the default real number precision for PETSc objects'''
self.precision = self.framework.argDB['with-precision'].lower()
if self.precision == 'single':
self.addDefine('USE_REAL_SINGLE', '1')
elif self.precision == '_quad': # source code currently does not support this
self.pushLanguage('C')
if not config.setCompilers.Configure.isIntel(self.compilers.getCompiler()): raise RuntimeError('Only Intel compiler supports _quad')
self.popLanguage()
self.addDefine('USE_REAL__QUAD', '1')
elif self.precision == 'double':
self.addDefine('USE_REAL_DOUBLE', '1')
elif self.precision == '__float128': # supported by gcc 4.6
if self.libraries.add('quadmath','logq',prototype='#include <quadmath.h>',call='__float128 f; logq(f);'):
self.addDefine('USE_REAL___FLOAT128', '1')
else:
raise RuntimeError('quadmath support not found. --with-precision=__float128 works with gcc-4.6 and newer compilers.')
else:
raise RuntimeError('--with-precision must be single, double,__float128')
self.framework.logPrint('Precision is '+str(self.precision))
if self.framework.argDB['with-mixed-precision']:
self.addDefine('USE_MIXED_PRECISION', '1')
return
def configure(self):
self.executeTest(self.configureScalarType)
self.executeTest(self.configurePrecision)
return
|
