summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/gearshifft/package.py
blob: 3d2bc53f0829ba84d23ae8fba8d724927796439a (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
# Copyright 2013-2022 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 Gearshifft(CMakePackage):
    """Benchmark Suite for Heterogenuous FFT Implementations"""

    homepage = "https://github.com/mpicbg-scicomp/gearshifft"
    url      = "https://github.com/mpicbg-scicomp/gearshifft/archive/v0.2.1-lw.tar.gz"

    maintainers = ['zyzzyxdonta']

    version('0.4.0', sha256='15b9e4bfa1d9b4fe4ae316f289c67b7be0774cdada5bd7310df4d0e026d9d227')

    # gearshifft used the variable name `CMAKE_DEFAULT_BUILD_TYPE` which was
    # later introduced by CMake leading to an error in newer CMake versions.
    # This patch renames the variable to `GEARSHIFFT_DEFAULT_BUILD_TYPE`.
    patch('gearshifft-v0.4.0-cmake-variable-name.patch', when='@0.4.0')

    variant('cufft', default=True,
            description='Compile gearshifft_cufft')
    variant('clfft', default=True,
            description='Compile gearshifft_clfft')
    variant('fftw', default=True,
            description='Compile gearshifft_fftw')
    variant('openmp', default=True,
            description='use OpenMP parallel fftw libraries')
    # variant('hcfft', default=True,
    #         description='Not implemented yet')
    variant('mkl', default=True,
            description='Compile gearshifft_fftwwrappers')
    variant('rocfft', default=True,
            description='Compile gearshifft_rocfft')

    # depends_on C++14 compiler, e.g. GCC 5.0+
    depends_on('cmake@2.8.0:', type='build')
    depends_on('boost@1.59.0:+system+test+program_options+thread')
    depends_on('cuda@8.0:', when='+cufft')
    depends_on('opencl@1.2:', when='+clfft')
    depends_on('clfft@2.12.0:', when='+clfft')
    depends_on('fftw@3.3.4:~mpi~openmp', when='+fftw~openmp')
    depends_on('fftw@3.3.4:~mpi+openmp', when='+fftw+openmp')
    depends_on('intel-mkl threads=openmp', when='+mkl')
    depends_on('rocfft', when='+rocfft')

    def cmake_args(self):
        spec = self.spec

        args = [
            self.define('GEARSHIFFT_FLOAT16_SUPPORT', False),
            self.define('GEARSHIFFT_BACKEND_HCFFT', False),
            self.define_from_variant('GEARSHIFFT_BACKEND_FFTW', 'fftw'),
            self.define('GEARSHIFFT_BACKEND_FFTW_PTHREADS', '~openmp' in spec),
            self.define_from_variant('GEARSHIFFT_BACKEND_FFTW_OPENMP', 'openmp'),
            self.define_from_variant('GEARSHIFFT_BACKEND_CUFFT', 'cufft'),
            self.define_from_variant('GEARSHIFFT_BACKEND_CLFFT', 'clfft'),
            self.define_from_variant('GEARSHIFFT_BACKEND_FFTWWRAPPERS', 'mkl'),
            self.define_from_variant('GEARSHIFFT_BACKEND_ROCFFT', 'rocfft')
        ]
        return args