summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/hipace/package.py
blob: 545ddd5d7edc021eaea79bdf04e1a2f4c5ad1e62 (plain) (blame)
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
# 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)

from spack import *


class Hipace(CMakePackage):
    """Highly efficient Plasma Accelerator Emulation, quasistatic
    particle-in-cell code
    """

    homepage = "https://hipace.readthedocs.io"
    url      = "https://github.com/Hi-PACE/hipace/archive/refs/tags/v21.09.tar.gz"
    git      = "https://github.com/Hi-PACE/hipace.git"

    maintainers = ['ax3l', 'MaxThevenet', 'SeverinDiederichs']

    version('develop', branch='development')
    version('21.09', sha256='5d27824fe6aac47ce26ca69759140ab4d7844f9042e436c343c03ea4852825f1')

    variant('compute',
            default='noacc',
            values=('omp', 'cuda', 'hip', 'sycl', 'noacc'),
            multi=False,
            description='On-node, accelerated computing backend')
    variant('mpi', default=True,
            description='Enable MPI support')
    variant('openpmd', default=True,
            description='Enable openPMD I/O')
    variant('precision',
            default='double',
            values=('single', 'double'),
            multi=False,
            description='Floating point precision (single/double)')

    depends_on('cmake@3.15.0:', type='build')
    depends_on('cuda@9.2.88:', when='compute=cuda')
    depends_on('mpi', when='+mpi')
    with when('+openpmd'):
        depends_on('openpmd-api@0.14.2:')
        depends_on('openpmd-api ~mpi', when='~mpi')
        depends_on('openpmd-api +mpi', when='+mpi')
    with when('compute=omp'):
        depends_on('fftw@3: +openmp')
        depends_on('fftw +mpi', when='+mpi')
        depends_on('pkgconfig', type='build')
        depends_on('llvm-openmp', when='%apple-clang')
    with when('compute=hip'):
        depends_on('rocfft')
        depends_on('rocprim')
        depends_on('rocrand')

    def cmake_args(self):
        spec = self.spec

        args = [
            # do not super-build dependencies
            '-HiPACE_openpmd_internal=OFF',
            # variants
            '-DHiPACE_COMPUTE={0}'.format(
                spec.variants['compute'].value.upper()),
            self.define_from_variant('HiPACE_MPI', 'mpi'),
            self.define_from_variant('HiPACE_OPENPMD', 'openpmd'),
            '-DHiPACE_PRECISION={0}'.format(
                spec.variants['precision'].value.upper()),
        ]

        return args