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
|
# coding=utf-8
import os
import os.path as osp
from functools import partial
from waflib import Options, Configure, Errors, Logs, Utils
def options(self):
self.load('compiler_c')
self.load('compiler_cxx')
self.load('compiler_fc')
group = self.get_option_group("Code_Aster options")
group.add_option('--enable-mpi', dest='parallel', action='store_true',
help='Build a parallel version with mpi')
def configure(self):
from Options import options as opts
if opts.parallel:
default = ['mpicc', 'mpicxx', 'mpif90']
else:
default = [''] * 3
for var in ('CC', 'CXX', 'FC'):
if not self.env[var]:
self.add_os_flags(var)
val = Utils.to_list(self.env[var])
os.environ[var] = (val and val[0]) or default.pop(0)
self.load_compilers()
self.check_openmp()
self.check_fortran_clib()
###############################################################################
@Configure.conf
def load_compilers(self):
self.env.stash() # Store a snapshot of the environment
self.load_compilers_mpi() # |
if not self.env.HAVE_MPI: # |
self.env.revert() # <-'
self.load('compiler_cc')
self.load('compiler_cxx')
self.load('compiler_fc')
@Configure.conf
def load_compilers_mpi(self):
from Options import options as opts
check = partial(self.check_cfg, args='--showme:compile --showme:link -show',
package='', uselib_store='MPI', mandatory=False)
cc = os.environ.get('CC')
cxx = os.environ.get('CXX')
fc = os.environ.get('FC')
if (cc and check(path=cc)) and (fc and check(path=fc)):
self.env.append_unique('CCNAME', osp.basename(cc))
self.env.append_unique('CXXNAME', osp.basename(cxx))
self.env.append_unique('FCNAME', osp.basename(fc))
self.check_mpi()
elif opts.parallel:
self.fatal("Unable to configure the parallel environment")
@Configure.conf
def check_mpi(self):
self.load('compiler_cc')
self.load('compiler_cxx')
self.load('compiler_fc')
self.check_cc(header_name='mpi.h', use='MPI', define_name='_USE_MPI')
@Configure.conf
def check_openmp(self):
try:
self.check_fortran_verbose_flag()
self.detect_openmp()
except (Errors.ConfigurationError, Errors.BuildError):
self.env.append_value('FCFLAGS_OPENMP', ['-fopenmp'])
self.env.append_value('FCLINKFLAGS_OPENMP', ['-fopenmp'])
if self.env.FCFLAGS_OPENMP:
self.define('_USE_OPENMP', 1)
|
