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
|
# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
import itertools
from spack import *
def process_amrex_constraints():
"""Map constraints when building with external AMReX"""
a1 = ['+', '~']
a2 = ['mpi', 'hypre', 'cuda']
a3 = [[x + y for x in a1] for y in a2]
for k in itertools.product(*a3):
if '+cuda' in k:
for arch in CudaPackage.cuda_arch_values:
yield ''.join(k) + " cuda_arch=%s" % arch
else:
yield ''.join(k)
class AmrWind(CMakePackage, CudaPackage):
"""AMR-Wind is a massively parallel, block-structured adaptive-mesh,
incompressible flow sover for wind turbine and wind farm simulations. """
homepage = "https://github.com/Exawind/amr-wind"
git = "https://github.com/exawind/amr-wind.git"
maintainers = ['sayerhs', 'jrood-nrel', 'michaeljbrazell']
tags = ['ecp', 'ecp-apps']
version('main', branch='main', submodules=True)
variant('shared', default=True,
description='Build shared libraries')
variant('unit', default=True,
description='Build unit tests')
variant('tests', default=True,
description='Activate regression tests')
variant('mpi', default=True,
description='Enable MPI support')
variant('openmp', default=False,
description='Enable OpenMP for CPU builds')
variant('netcdf', default=True,
description='Enable NetCDF support')
variant('hypre', default=True,
description='Enable Hypre integration')
variant('masa', default=False,
description='Enable MASA integration')
variant('openfast', default=False,
description='Enable OpenFAST integration')
variant('internal-amrex', default=True,
description='Use AMRex submodule to build')
variant('fortran', default=False,
description='Build fortran interfaces')
conflicts('+openmp', when='+cuda')
depends_on('mpi', when='+mpi')
for opt in process_amrex_constraints():
dopt = '+particles' + opt
if '+hypre' in dopt:
dopt = "+fortran" + dopt
depends_on('amrex@develop' + dopt, when='~internal-amrex' + opt)
depends_on('hypre+mpi+int64~cuda@2.20.0:', when='+mpi~cuda+hypre')
depends_on('hypre~mpi+int64~cuda@2.20.0:', when='~mpi~cuda+hypre')
for arch in CudaPackage.cuda_arch_values:
depends_on('hypre+mpi~int64+cuda cuda_arch=%s @2.20.0:' % arch,
when='+mpi+cuda+hypre cuda_arch=%s' % arch)
depends_on('hypre~mpi~int64+cuda cuda_arch=%s @2.20.0:' % arch,
when='~mpi+cuda+hypre cuda_arch=%s' % arch)
depends_on('netcdf-c', when='+netcdf')
depends_on('masa', when='+masa')
depends_on('openfast+cxx', when='+openfast')
def cmake_args(self):
define = CMakePackage.define
vs = ["mpi", "cuda", "openmp", "netcdf", "hypre", "masa",
"openfast", "tests", "fortran"]
args = [
self.define_from_variant("AMR_WIND_ENABLE_%s" % v.upper(), v)
for v in vs
]
args += [
define('CMAKE_EXPORT_COMPILE_COMMANDS', True),
define('AMR_WIND_ENABLE_ALL_WARNINGS', True),
self.define_from_variant('BUILD_SHARED_LIBS', 'shared'),
self.define_from_variant('AMR_WIND_TEST_WITH_FCOMPARE', 'tests'),
]
if '+cuda' in self.spec:
amrex_arch = ['{0:.1f}'.format(float(i) / 10.0)
for i in self.spec.variants['cuda_arch'].value]
if amrex_arch:
args.append(define('AMReX_CUDA_ARCH', amrex_arch))
if '+internal-amrex' in self.spec:
args.append(self.define('AMR_WIND_USE_INTERNAL_AMREX', True))
else:
args += [
self.define('AMR_WIND_USE_INTERNAL_AMREX', False),
self.define('AMReX_ROOT', self.spec['amrex'].prefix)
]
return args
|
