# Copyright 2013-2018 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 * import llnl.util.lang class Fftw(AutotoolsPackage): """FFTW is a C subroutine library for computing the discrete Fourier transform (DFT) in one or more dimensions, of arbitrary input size, and of both real and complex data (as well as of even/odd data, i.e. the discrete cosine/sine transforms or DCT/DST). We believe that FFTW, which is free software, should become the FFT library of choice for most applications.""" homepage = "http://www.fftw.org" url = "http://www.fftw.org/fftw-3.3.4.tar.gz" list_url = "http://www.fftw.org/download.html" version('3.3.8', '8aac833c943d8e90d51b697b27d4384d') version('3.3.7', '0d5915d7d39b3253c1cc05030d79ac47') version('3.3.6-pl2', '927e481edbb32575397eb3d62535a856') version('3.3.5', '6cc08a3b9c7ee06fdd5b9eb02e06f569') version('3.3.4', '2edab8c06b24feeb3b82bbb3ebf3e7b3') version('2.1.5', '8d16a84f3ca02a785ef9eb36249ba433') patch('pfft-3.3.5.patch', when="@3.3.5:+pfft_patches", level=0) patch('pfft-3.3.4.patch', when="@3.3.4+pfft_patches", level=0) patch('pgi-3.3.6-pl2.patch', when="@3.3.6-pl2%pgi", level=0) variant( 'float', default=True, description='Produces a single precision version of the library') variant( 'double', default=True, description='Produces a double precision version of the library') variant( 'long_double', default=True, description='Produces a long double precision version of the library') variant( 'quad', default=False, description='Produces a quad precision version of the library ' '(works only with GCC and libquadmath)') variant('openmp', default=False, description="Enable OpenMP support.") variant('mpi', default=True, description='Activate MPI support') variant( 'pfft_patches', default=False, description='Add extra transpose functions for PFFT compatibility') variant( 'simd', default='sse2,avx,avx2', values=( 'sse', 'sse2', 'avx', 'avx2', 'avx512', # Intel 'avx-128-fma', 'kcvi', # Intel 'altivec', 'vsx', # IBM 'neon', # ARM 'generic-simd128', 'generic-simd256' # Generic ), description='Optimizations that are enabled in this build', multi=True ) variant('fma', default=False, description='Activate support for fma') depends_on('mpi', when='+mpi') depends_on('automake', type='build', when='+pfft_patches') depends_on('autoconf', type='build', when='+pfft_patches') depends_on('libtool', type='build', when='+pfft_patches') # https://github.com/FFTW/fftw3/commit/902d0982522cdf6f0acd60f01f59203824e8e6f3 conflicts('%gcc@8:8.9999', when="@3.3.7") provides('fftw-api@2', when='@2.1.5') provides('fftw-api@3', when='@3:') @property def libs(self): # Reduce repetitions of entries query_parameters = list(llnl.util.lang.dedupe( self.spec.last_query.extra_parameters )) # List of all the suffixes associated with float precisions precisions = [ ('float', 'f'), ('double', ''), ('long_double', 'l'), ('quad', 'q') ] # Retrieve the correct suffixes, or use double as a default suffixes = [v for k, v in precisions if k in query_parameters] or [''] # Construct the list of libraries that needs to be found libraries = [] for sfx in suffixes: if 'mpi' in query_parameters and '+mpi' in self.spec: libraries.append('libfftw3' + sfx + '_mpi') if 'openmp' in query_parameters and '+openmp' in self.spec: libraries.append('libfftw3' + sfx + '_omp') libraries.append('libfftw3' + sfx) return find_libraries(libraries, root=self.prefix, recursive=True) def autoreconf(self, spec, prefix): if '+pfft_patches' in spec: autoreconf = which('autoreconf') autoreconf('-ifv') def configure(self, spec, prefix): # Base options options = [ '--prefix={0}'.format(prefix), '--enable-shared', '--enable-threads' ] if not self.compiler.f77 or not self.compiler.fc: options.append("--disable-fortran") if spec.satisfies('@:2'): options.append('--enable-type-prefix') # Variants that affect every precision if '+openmp' in spec: # Note: Apple's Clang does not support OpenMP. if spec.satisfies('%clang'): ver = str(self.compiler.version) if ver.endswith('-apple'): raise InstallError("Apple's clang does not support OpenMP") options.append('--enable-openmp') if spec.satisfies('@:2'): # TODO: libtool strips CFLAGS, so 2.x libxfftw_threads # isn't linked to the openmp library. Patch Makefile? options.insert(0, 'CFLAGS=' + self.compiler.openmp_flag) if '+mpi' in spec: options.append('--enable-mpi') # SIMD support float_options, double_options = [], [] if spec.satisfies('@3:', strict=True): for opts in (float_options, double_options): opts += self.enable_or_disable('simd') opts += self.enable_or_disable('fma') configure = Executable('../configure') # Build double/float/long double/quad variants if '+double' in spec: with working_dir('double', create=True): configure(*(options + double_options)) if '+float' in spec: with working_dir('float', create=True): configure('--enable-float', *(options + float_options)) if spec.satisfies('@3:+long_double'): with working_dir('long-double', create=True): configure('--enable-long-double', *options) if spec.satisfies('@3:+quad'): with working_dir('quad', create=True): configure('--enable-quad-precision', *options) def build(self, spec, prefix): if '+double' in spec: with working_dir('double'): make() if '+float' in spec: with working_dir('float'): make() if '+long_double' in spec: with working_dir('long-double'): make() if '+quad' in spec: with working_dir('quad'): make() def check(self): spec = self.spec if '+double' in spec: with working_dir('double'): make("check") if '+float' in spec: with working_dir('float'): make("check") if '+long_double' in spec: with working_dir('long-double'): make("check") if '+quad' in spec: with working_dir('quad'): make("check") def install(self, spec, prefix): if '+double' in spec: with working_dir('double'): make("install") if '+float' in spec: with working_dir('float'): make("install") if '+long_double' in spec: with working_dir('long-double'): make("install") if '+quad' in spec: with working_dir('quad'): make("install")