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
|
# Copyright 2013-2019 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 = ['ax3l']
version('0.2.1-lw', sha256='04ba7401615ab29a37089c0dce8580270c0c4aa1ba328c9d438d6e4f163899c5')
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')
# depends_on C++14 compiler, e.g. GCC 5.0+
depends_on('cmake@2.8.0:', type='build')
depends_on('boost@1.56.0:')
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')
def cmake_args(self):
spec = self.spec
args = [
'-DGEARSHIFFT_HCFFT:BOOL=OFF',
'-DGEARSHIFFT_FFTW_PTHREADS:BOOL=ON',
'-DGEARSHIFFT_CLFFT:BOOL=OFF'
]
args.extend([
'-DGEARSHIFFT_FFTW:BOOL={0}'.format(
'ON' if '+fftw' in spec else 'OFF'),
'-DGEARSHIFFT_FFTW_OPENMP:BOOL={0}'.format(
'ON' if '+openmp' in spec else 'OFF'),
'-DGEARSHIFFT_CUFFT:BOOL={0}'.format(
'ON' if '+cufft' in spec else 'OFF'),
'-DGEARSHIFFT_CLFFT:BOOL={0}'.format(
'ON' if '+clfft' in spec else 'OFF')
])
return args
|
