hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

WarpX: Add sensei insitu variant (#31542)

Browse source
This commit is contained in:
kwryankrattiger 2022-07-27 14:17:44 -05:00 committed by GitHub
parent afc044cbad
commit a3f8dc1874
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
var/spack/repos/builtin/packages/warpx/package.py
View file

@ -47,7 +47,9 @@ class Warpx(CMakePackage):
variant('app', default=True, variant('app', default=True,
description='Build the WarpX executable application') description='Build the WarpX executable application')
variant('ascent', default=False, variant('ascent', default=False,
description='Enable Ascent in situ vis') description='Enable Ascent in situ visualization')
variant('sensei', default=False,
description='Enable SENSEI in situ visualization')
variant('compute', variant('compute',
default='omp', default='omp',
values=('omp', 'cuda', 'hip', 'sycl', 'noacc'), values=('omp', 'cuda', 'hip', 'sycl', 'noacc'),
@ -84,11 +86,12 @@ class Warpx(CMakePackage):
variant('tprof', default=True, variant('tprof', default=True,
description='Enable tiny profiling features') description='Enable tiny profiling features')
depends_on('sensei@4.0.0:', when='@22.07: +sensei')
conflicts('+sensei', when='@:22.06',
msg='WarpX supports SENSEI 4.0+ with 22.07 and newer')
depends_on('ascent', when='+ascent') depends_on('ascent', when='+ascent')
# note: ~shared is only needed until the new concretizer is in and depends_on('ascent +cuda', when='+ascent compute=cuda')
# honors the conflict inside the Ascent package to find this
# automatically
depends_on('ascent +cuda ~shared', when='+ascent compute=cuda')
depends_on('ascent +mpi', when='+ascent +mpi') depends_on('ascent +mpi', when='+ascent +mpi')
depends_on('boost@1.66.0: +math', when='+qedtablegen') depends_on('boost@1.66.0: +math', when='+qedtablegen')
depends_on('cmake@3.15.0:', type='build') depends_on('cmake@3.15.0:', type='build')
@ -166,6 +169,7 @@ def cmake_args(self):
# variants # variants
self.define_from_variant('WarpX_APP', 'app'), self.define_from_variant('WarpX_APP', 'app'),
self.define_from_variant('WarpX_ASCENT', 'ascent'), self.define_from_variant('WarpX_ASCENT', 'ascent'),
self.define_from_variant('WarpX_SENSEI', 'sensei'),
'-DWarpX_COMPUTE={0}'.format( '-DWarpX_COMPUTE={0}'.format(
spec.variants['compute'].value.upper()), spec.variants['compute'].value.upper()),
'-DWarpX_DIMS={0}'.format( '-DWarpX_DIMS={0}'.format(
@ -185,6 +189,10 @@ def cmake_args(self):
with when('+openpmd'): with when('+openpmd'):
args.append('-DWarpX_openpmd_internal=OFF') args.append('-DWarpX_openpmd_internal=OFF')
if '+sensei' in spec:
args.append(self.define('SENSEI_DIR',
join_path(spec['sensei'].prefix.lib, 'cmake')))
return args return args
@property @property