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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
from __future__ import generators
import config.base
import os
class Configure(config.base.Configure):
def __init__(self, framework):
config.base.Configure.__init__(self, framework)
self.headerPrefix = ''
self.substPrefix = ''
return
def __str__(self):
return ''
def setupHelp(self, help):
import nargs
help.addArgument('PETSc', '-with-log=<bool>', nargs.ArgBool(None, 1, 'Activate logging code in PETSc'))
help.addArgument('PETSc', '-with-threadsafety=<bool>', nargs.ArgBool(None, 0, 'Allow individual threads in PETSc to call PETSc routines'))
help.addArgument('PETSc', '-with-info=<bool>', nargs.ArgBool(None, 1, 'Activate PetscInfo() (i.e. -info) code in PETSc'))
help.addArgument('PETSc', '-with-ctable=<bool>', nargs.ArgBool(None, 1, 'Use hash maps in certain places in PETSc, instead of non-memory-scalable arrays'))
help.addArgument('PETSc', '-with-dmlandau-3d=<bool>', nargs.ArgBool(None, 0, 'Enable full 3D DM Landau, default is 2.5D'))
help.addArgument('PETSc', '-with-fortran-kernels=<bool>', nargs.ArgBool(None, 0, 'Use Fortran for linear algebra kernels'))
help.addArgument('PETSc', '-with-avx512-kernels=<bool>', nargs.ArgBool(None, 1, 'Use AVX-512 intrinsics for linear algebra kernels when available'))
help.addArgument('PETSc', '-with-is-color-value-type=<char,short>',nargs.ArgString(None, 'short', 'char, short can store 256, 65536 colors'))
return
def setupDependencies(self, framework):
config.base.Configure.setupDependencies(self, framework)
self.compilers = framework.require('config.compilers', self)
self.setCompilers = framework.require('config.setCompilers', self)
self.libraries = framework.require('config.libraries', self)
self.types = framework.require('config.types', self)
self.compilerFlags = framework.require('config.compilerFlags', self)
self.sharedLibraries = framework.require('PETSc.options.sharedLibraries', None)
self.petscConfigure = framework.require('PETSc.Configure', None)
return
def configureLibraryOptions(self):
'''Sets PETSC_USE_DEBUG, PETSC_USE_INFO, PETSC_USE_LOG, PETSC_USE_CTABLE, PETSC_USE_DMLANDAU_2D, PETSC_USE_FORTRAN_KERNELS, and PETSC_USE_AVX512_KERNELS'''
'''Also sets PETSC_AssertAlignx() in Fortran for IBM BG/P compiler '''
if self.framework.argDB['with-threadsafety']:
self.addDefine('HAVE_THREADSAFETY',1)
self.useThreadSafety = 1
else:
self.useThreadSafety = 0
if self.useThreadSafety and not ((self.sharedLibraries.useShared and self.setCompilers.dynamicLibraries) or self.framework.argDB['with-single-library']):
raise RuntimeError('Must use --with-shared-libraries or --with-single-library with --with-threadsafety')
self.useLog = self.framework.argDB['with-log']
self.addDefine('USE_LOG', self.useLog)
if self.compilerFlags.debugging:
self.addDefine('USE_DEBUG',1)
elif not config.setCompilers.Configure.isIBM(self.framework.getCompiler(), self.log):
# IBM XLC version 12.1 (BG/Q and POWER) miscompiles PetscMalloc3()
# by reordering "*(void**)&ptr = x" as though ptr was not modified
# by this statement.
self.addDefine('USE_MALLOC_COALESCED',1)
self.useInfo = self.framework.argDB['with-info']
self.addDefine('USE_INFO', self.useInfo)
self.useCtable = self.framework.argDB['with-ctable']
if self.useCtable:
self.addDefine('USE_CTABLE', '1')
self.useDmLandau3d = self.framework.argDB['with-dmlandau-3d']
if not self.useDmLandau3d:
self.addDefine('USE_DMLANDAU_2D',1)
# used in src/mat/impls/sbaij/seq/relax.h
self.libraries.saveLog()
if not self.libraries.isBGL():
self.addDefine('USE_BACKWARD_LOOP','1')
self.logWrite(self.libraries.restoreLog())
self.useFortranKernels = self.framework.argDB['with-fortran-kernels']
if not hasattr(self.compilers, 'FC') and self.useFortranKernels:
raise RuntimeError('Cannot use fortran kernels without a Fortran compiler')
if self.useFortranKernels:
self.addDefine('USE_FORTRAN_KERNELS', 1)
if self.libraries.isBGL():
self.addDefine('AssertAlignx(a,b)','call ALIGNX(a,b)')
else:
self.addDefine('AssertAlignx(a,b)',' ')
self.useAVX512Kernels = self.framework.argDB['with-avx512-kernels']
if self.useAVX512Kernels:
self.addDefine('USE_AVX512_KERNELS', 1)
return
def configureISColorValueType(self):
'''Sets PETSC_IS_COLORING_VALUE_TYPE, PETSC_MPIU_IS_COLORING_VALUE_TYPE, and PETSC_IS_COLORING_MAX as required by ISColoring'''
self.isColorValueType = self.framework.argDB['with-is-color-value-type']
if self.isColorValueType != 'char' and self.isColorValueType != 'short':
raise RuntimeError('Incorrect --with-is-color-value-type value specified. Can be either char or short. Specified value is :'+self.isColorValueType)
if self.isColorValueType == 'char':
max_value = 'UCHAR_MAX'
mpi_type = 'MPI_UNSIGNED_CHAR'
type_f = 'integer1'
else:
max_value = 'USHRT_MAX'
mpi_type = 'MPI_UNSIGNED_SHORT'
type_f = 'integer2'
self.addDefine('MPIU_IS_COLORING_VALUE_TYPE',mpi_type)
self.addDefine('IS_COLORING_MAX',max_value)
self.addDefine('IS_COLORING_VALUE_TYPE',self.isColorValueType)
self.addDefine('IS_COLORING_VALUE_TYPE_F',type_f)
return
def configure(self):
self.executeTest(self.configureLibraryOptions)
self.executeTest(self.configureISColorValueType)
return
|
