diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/fftw/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/fftw/package.py | 224 |
1 files changed, 103 insertions, 121 deletions
diff --git a/var/spack/repos/builtin/packages/fftw/package.py b/var/spack/repos/builtin/packages/fftw/package.py index 9d8fb69936..7ce50f5df9 100644 --- a/var/spack/repos/builtin/packages/fftw/package.py +++ b/var/spack/repos/builtin/packages/fftw/package.py @@ -17,55 +17,50 @@ class FftwBase(AutotoolsPackage): """ variant( - 'precision', values=any_combination_of( - 'float', 'double', 'long_double', 'quad' - ).prohibit_empty_set().with_default('float,double'), - description='Build the selected floating-point precision libraries' + "precision", + values=any_combination_of("float", "double", "long_double", "quad") + .prohibit_empty_set() + .with_default("float,double"), + description="Build the selected floating-point precision libraries", ) - variant('openmp', default=False, description="Enable OpenMP support.") - variant('mpi', default=True, description='Activate MPI support') + variant("openmp", default=False, description="Enable OpenMP support.") + variant("mpi", default=True, description="Activate MPI support") - depends_on('mpi', when='+mpi') - depends_on('llvm-openmp', when='%apple-clang +openmp') + depends_on("mpi", when="+mpi") + depends_on("llvm-openmp", when="%apple-clang +openmp") # https://github.com/FFTW/fftw3/commit/902d0982522cdf6f0acd60f01f59203824e8e6f3 - conflicts('%gcc@8.0:8', when="@3.3.7") - conflicts('precision=long_double', when='@2.1.5', - msg='Long double precision is not supported in FFTW 2') - conflicts('precision=quad', when='@2.1.5', - msg='Quad precision is not supported in FFTW 2') - conflicts('precision=quad', when='+mpi', - msg='Quad precision is not supported in MPI') + conflicts("%gcc@8.0:8", when="@3.3.7") + conflicts( + "precision=long_double", + when="@2.1.5", + msg="Long double precision is not supported in FFTW 2", + ) + conflicts("precision=quad", when="@2.1.5", msg="Quad precision is not supported in FFTW 2") + conflicts("precision=quad", when="+mpi", msg="Quad precision is not supported in MPI") @property def libs(self): # Reduce repetitions of entries - query_parameters = list(llnl.util.lang.dedupe( - self.spec.last_query.extra_parameters - )) + 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') - ] + 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 [''] + 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 "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') + if "openmp" in query_parameters and "+openmp" in self.spec: + libraries.append("libfftw3" + sfx + "_omp") - libraries.append('libfftw3' + sfx) + libraries.append("libfftw3" + sfx) return find_libraries(libraries, root=self.prefix, recursive=True) @@ -73,131 +68,116 @@ class FftwBase(AutotoolsPackage): # If fftw/config.h exists in the source tree, it will take precedence # over the copy in build dir. As only the latter has proper config # for our build, this is a problem. See e.g. issue #7372 on github - if os.path.isfile('fftw/config.h'): - os.rename('fftw/config.h', 'fftw/config.h.SPACK_RENAMED') + if os.path.isfile("fftw/config.h"): + os.rename("fftw/config.h", "fftw/config.h.SPACK_RENAMED") def autoreconf(self, spec, prefix): - if '+pfft_patches' in spec: - autoreconf = which('autoreconf') - autoreconf('-ifv') + if "+pfft_patches" in spec: + autoreconf = which("autoreconf") + autoreconf("-ifv") @property def selected_precisions(self): """Precisions that have been selected in this build""" - return self.spec.variants['precision'].value + return self.spec.variants["precision"].value def setup_build_environment(self, env): - if self.spec.satisfies('%apple-clang +openmp'): - env.append_flags( - 'CPPFLAGS', self.compiler.openmp_flag) - env.append_flags( - 'CFLAGS', self.spec['llvm-openmp'].headers.include_flags) - env.append_flags( - 'CXXFLAGS', self.spec['llvm-openmp'].headers.include_flags) - env.append_flags( - 'LDFLAGS', self.spec['llvm-openmp'].libs.ld_flags) + if self.spec.satisfies("%apple-clang +openmp"): + env.append_flags("CPPFLAGS", self.compiler.openmp_flag) + env.append_flags("CFLAGS", self.spec["llvm-openmp"].headers.include_flags) + env.append_flags("CXXFLAGS", self.spec["llvm-openmp"].headers.include_flags) + env.append_flags("LDFLAGS", self.spec["llvm-openmp"].libs.ld_flags) # FFTW first configures libtool without MPI, and later uses it with # MPI. libtool then calls wrong linker to create shared libraries # (it calls `$CC` instead of `$MPICC`), and MPI symbols # remain undefined because `-lmpi` is not passed to the linker. # https://github.com/FFTW/fftw3/issues/274 # https://github.com/spack/spack/issues/29224 - if self.spec.satisfies('+mpi') and self.spec.satisfies('platform=darwin'): - env.append_flags('LIBS', self.spec['mpi'].libs.ld_flags) + if self.spec.satisfies("+mpi") and self.spec.satisfies("platform=darwin"): + env.append_flags("LIBS", self.spec["mpi"].libs.ld_flags) def configure(self, spec, prefix): # Base options - options = [ - '--prefix={0}'.format(prefix), - '--enable-shared', - '--enable-threads' - ] + 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') + if spec.satisfies("@:2"): + options.append("--enable-type-prefix") # Variants that affect every precision - if '+openmp' in spec: - options.append('--enable-openmp') - if spec.satisfies('@:2'): + if "+openmp" in spec: + 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') + options.insert(0, "CFLAGS=" + self.compiler.openmp_flag) + if "+mpi" in spec: + options.append("--enable-mpi") # Specific SIMD support. # all precisions - simd_features = ['sse2', 'avx', 'avx2', 'avx512', 'avx-128-fma', - 'kcvi', 'vsx'] + simd_features = ["sse2", "avx", "avx2", "avx512", "avx-128-fma", "kcvi", "vsx"] # float only - float_simd_features = ['altivec', 'sse', 'neon'] + float_simd_features = ["altivec", "sse", "neon"] # Workaround PGI compiler bug when avx2 is enabled - if spec.satisfies('%pgi') and 'avx2' in simd_features: - simd_features.remove('avx2') + if spec.satisfies("%pgi") and "avx2" in simd_features: + simd_features.remove("avx2") # Workaround NVIDIA/PGI compiler bug when avx512 is enabled - if spec.satisfies('%nvhpc') or spec.satisfies('%pgi'): - if 'avx512' in simd_features: - simd_features.remove('avx512') + if spec.satisfies("%nvhpc") or spec.satisfies("%pgi"): + if "avx512" in simd_features: + simd_features.remove("avx512") # NVIDIA compiler does not support Altivec intrinsics - if spec.satisfies('%nvhpc') and 'vsx' in simd_features: - simd_features.remove('vsx') - if spec.satisfies('%nvhpc') and 'altivec' in float_simd_features: - float_simd_features.remove('altivec') + if spec.satisfies("%nvhpc") and "vsx" in simd_features: + simd_features.remove("vsx") + if spec.satisfies("%nvhpc") and "altivec" in float_simd_features: + float_simd_features.remove("altivec") # NVIDIA compiler does not support Neon intrinsics - if spec.satisfies('%nvhpc') and 'neon' in simd_features: - simd_features.remove('neon') + if spec.satisfies("%nvhpc") and "neon" in simd_features: + simd_features.remove("neon") simd_options = [] for feature in simd_features: - msg = '--enable-{0}' if feature in spec.target else '--disable-{0}' + msg = "--enable-{0}" if feature in spec.target else "--disable-{0}" simd_options.append(msg.format(feature)) # If no features are found, enable the generic ones - if not any(f in spec.target for f in - simd_features + float_simd_features): + if not any(f in spec.target for f in simd_features + float_simd_features): # Workaround NVIDIA compiler bug - if not spec.satisfies('%nvhpc'): - simd_options += [ - '--enable-generic-simd128', - '--enable-generic-simd256' - ] + if not spec.satisfies("%nvhpc"): + simd_options += ["--enable-generic-simd128", "--enable-generic-simd256"] - simd_options += [ - '--enable-fma' if 'fma' in spec.target else '--disable-fma' - ] + simd_options += ["--enable-fma" if "fma" in spec.target else "--disable-fma"] # Double is the default precision, for all the others we need # to enable the corresponding option. enable_precision = { - 'float': ['--enable-float'], - 'double': None, - 'long_double': ['--enable-long-double'], - 'quad': ['--enable-quad-precision'] + "float": ["--enable-float"], + "double": None, + "long_double": ["--enable-long-double"], + "quad": ["--enable-quad-precision"], } # Different precisions must be configured and compiled one at a time - configure = Executable('../configure') + configure = Executable("../configure") for precision in self.selected_precisions: opts = (enable_precision[precision] or []) + options[:] # SIMD optimizations are available only for float and double # starting from FFTW 3 - if precision in ('float', 'double') and spec.satisfies('@3:'): + if precision in ("float", "double") and spec.satisfies("@3:"): opts += simd_options # float-only acceleration - if precision == 'float': + if precision == "float": for feature in float_simd_features: if feature in spec.target: - msg = '--enable-{0}' + msg = "--enable-{0}" else: - msg = '--disable-{0}' + msg = "--disable-{0}" opts.append(msg.format(feature)) with working_dir(precision, create=True): @@ -212,46 +192,48 @@ class FftwBase(AutotoolsPackage): self.for_each_precision_make() def check(self): - self.for_each_precision_make('check') + self.for_each_precision_make("check") def install(self, spec, prefix): - self.for_each_precision_make('install') + self.for_each_precision_make("install") class Fftw(FftwBase): """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.""" + 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 = "https://www.fftw.org" url = "https://www.fftw.org/fftw-3.3.4.tar.gz" list_url = "https://www.fftw.org/download.html" - version('3.3.10', sha256='56c932549852cddcfafdab3820b0200c7742675be92179e59e6215b340e26467') - version('3.3.9', sha256='bf2c7ce40b04ae811af714deb512510cc2c17b9ab9d6ddcf49fe4487eea7af3d') - version('3.3.8', sha256='6113262f6e92c5bd474f2875fa1b01054c4ad5040f6b0da7c03c98821d9ae303') - version('3.3.7', sha256='3b609b7feba5230e8f6dd8d245ddbefac324c5a6ae4186947670d9ac2cd25573') - version('3.3.6-pl2', sha256='a5de35c5c824a78a058ca54278c706cdf3d4abba1c56b63531c2cb05f5d57da2') - version('3.3.5', sha256='8ecfe1b04732ec3f5b7d279fdb8efcad536d555f9d1e8fabd027037d45ea8bcf') - version('3.3.4', sha256='8f0cde90929bc05587c3368d2f15cd0530a60b8a9912a8e2979a72dbe5af0982') - version('2.1.5', sha256='f8057fae1c7df8b99116783ef3e94a6a44518d49c72e2e630c24b689c6022630') + version("3.3.10", sha256="56c932549852cddcfafdab3820b0200c7742675be92179e59e6215b340e26467") + version("3.3.9", sha256="bf2c7ce40b04ae811af714deb512510cc2c17b9ab9d6ddcf49fe4487eea7af3d") + version("3.3.8", sha256="6113262f6e92c5bd474f2875fa1b01054c4ad5040f6b0da7c03c98821d9ae303") + version("3.3.7", sha256="3b609b7feba5230e8f6dd8d245ddbefac324c5a6ae4186947670d9ac2cd25573") + version("3.3.6-pl2", sha256="a5de35c5c824a78a058ca54278c706cdf3d4abba1c56b63531c2cb05f5d57da2") + version("3.3.5", sha256="8ecfe1b04732ec3f5b7d279fdb8efcad536d555f9d1e8fabd027037d45ea8bcf") + version("3.3.4", sha256="8f0cde90929bc05587c3368d2f15cd0530a60b8a9912a8e2979a72dbe5af0982") + version("2.1.5", sha256="f8057fae1c7df8b99116783ef3e94a6a44518d49c72e2e630c24b689c6022630") variant( - 'pfft_patches', default=False, - description='Add extra transpose functions for PFFT compatibility') + "pfft_patches", + default=False, + description="Add extra transpose functions for PFFT compatibility", + ) - depends_on('automake', type='build', when='+pfft_patches') - depends_on('autoconf', type='build', when='+pfft_patches') - depends_on('libtool', type='build', when='+pfft_patches') + depends_on("automake", type="build", when="+pfft_patches") + depends_on("autoconf", type="build", when="+pfft_patches") + depends_on("libtool", type="build", when="+pfft_patches") - provides('fftw-api@2', when='@2.1.5') - provides('fftw-api@3', when='@3:') + provides("fftw-api@2", when="@2.1.5") + provides("fftw-api@3", when="@3:") - patch('pfft-3.3.9.patch', when="@3.3.9:+pfft_patches", level=0) - patch('pfft-3.3.5.patch', when="@3.3.5:3.3.8+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) - patch('intel-configure.patch', when="@3:3.3.8%intel", level=0) + patch("pfft-3.3.9.patch", when="@3.3.9:+pfft_patches", level=0) + patch("pfft-3.3.5.patch", when="@3.3.5:3.3.8+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) + patch("intel-configure.patch", when="@3:3.3.8%intel", level=0) |