diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/ffte/package.py | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/ffte/package.py b/var/spack/repos/builtin/packages/ffte/package.py new file mode 100644 index 0000000000..eeee5c14bc --- /dev/null +++ b/var/spack/repos/builtin/packages/ffte/package.py @@ -0,0 +1,113 @@ +# 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) + +import glob + +from spack import * + + +class Ffte(Package): + """Fastest Fourier Transform in the East. Provides FFT for powers of 2, 3, + and 5 lengths in one, two, and three dimensions. Support for vector + hardware, MPI, and CUDA Fortran is also included.""" + + homepage = 'http://www.ffte.jp/' + url = 'http://www.ffte.jp/ffte-7.0.tgz' + + maintainers = ['luszczek'] + + version('7.0', sha256='078d5f84a5f2479ca5c4a3bd777ad761fe98addf1642a045bac6602a0cae3da0') + version('6.0', sha256='fc82595a8f8323b2796cc5eeb1cc9f7e50ca9e511a14365cc3984da6b7a9b8b4') + version('5.0', sha256='1f46ca16badc3aca0ad13ca91a6a67829a57b403501cdc821b80cfa62b2a89c2') + version('4.0', sha256='61680f73c48659ac45aec60ef5a725547f763bb9017edbd3f44a6a9ad0fda62f') + version('3.0', sha256='dbaab8204a16072c8d572efa3733e9946a9be0d1a051fc19e2d9253be23247ff') + version('2.0', sha256='f5cf1d1f880288e359f4d517191980ffca4420f817ecaa2d754ca5c5421271e3') + version('1.0', sha256='35171e3324019018c25575b2807a6513fa85badad040f30f238fff03d4b4d1ab') + + variant('mpi', default=False, description='Build MPI library') + variant('cuda', default=False, description='Use CUDA Fortran') + variant('vector', default=False, description='Use vectorized FFT') + + depends_on('mpi', when='+mpi') + + conflicts('%cce', when='+cuda', msg='Must use NVHPC compiler') + conflicts('%clang', when='+cuda', msg='Must use NVHPC compiler') + conflicts('%gcc', when='+cuda', msg='Must use NVHPC compiler') + conflicts('%llvm', when='+cuda', msg='Must use NVHPC compiler') + conflicts('%nag', when='+cuda', msg='Must use NVHPC compiler') + conflicts('%intel', when='+cuda', msg='Must use NVHPC compiler') + + def edit(self, spec, prefix): + 'No make-file, must create one from scratch.' + + open('mpi/param.h', 'w').write(open('param.h').read()) + + for fpfx in ('.', 'mpi'): + vrs = '%sfftever' % ['', '', '', 'p'][len(fpfx)] + s6 = 6 * ' ' + fd = open('%s/%s.f' % (fpfx, vrs), 'w') + fd.write('%ssubroutine %s(i,j,k)\n' % (s6, vrs)) + fd.write('%sinteger i,j,k\n' % s6) + fd.write('%si=%s\n' % (s6, spec.version[0])) + fd.write('%sj=%s\n' % (s6, spec.version[1])) + fd.write('%sk=0\n%sreturn\n%send\n' % (3 * (s6,))) + fd.close() + + ff, cuf, vf = [], [], [] + for f in glob.glob('%s/*.f' % fpfx): + nf = f.replace('.f', '.o').replace('%s/' % fpfx, '') + if '/cu' in f or '/pcu' in f: + cuf.append(nf) + elif '/v' in f or '/pv' in f: + vf.append(nf) + else: + ff.append(nf) + + if '.' == fpfx: + lbnm = 'libffte.a' + else: + lbnm = 'libfftempi.a' + + for spc, vrf, rs, ip in ( + ('+vector', vf, 'v', 2), + ('+cuda', cuf, 'cu', 3), + ): + if spc in spec: + for f in vrf: # replace with variant versions + orgf = f[:ip].replace(rs, '') + f[ip:] + if orgf in ff: # already reference implementation? + del ff[ff.index(orgf)] + ff.extend(vrf) + + aff = ' '.join(ff) + fd = open('%s/Makefile' % fpfx, 'w') + fd.write('all: %s\n' % lbnm) + if '+mpi' in spec and '.' == fpfx: + fd.write('\tcd mpi ; $(MAKE)\n') + fd.write('\n%s: %s\n' % (lbnm, aff)) + fd.write('\tar rc %s %s\n' % (lbnm, aff)) + fd.write('\tranlib %s\n' % lbnm) + fd.close() + + def install(self, spec, prefix): + self.edit(spec, prefix) + if '+mpi' in spec: + env['CC'] = spec['mpi'].mpicc + env['F77'] = spec['mpi'].mpif77 + env['FC'] = spec['mpi'].mpifc + + # allow real/complex aliasing in GNU Fortran 10 and up + if spec.satisfies('%gcc@10:'): + env['FFLAGS'] = '-fallow-argument-mismatch' + + # enable CUDA Fortran in NVHPC + if spec.satisfies('%nvhpc'): + env['FFLAGS'] = '-Mcuda' + + make() + mkdirp(prefix.lib) + install('libffte.a', prefix.lib) + if '+mpi' in spec: + install('mpi/libfftempi.a', prefix.lib) |