diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/sw4lite/package.py | 46 |
1 files changed, 42 insertions, 4 deletions
diff --git a/var/spack/repos/builtin/packages/sw4lite/package.py b/var/spack/repos/builtin/packages/sw4lite/package.py index 3378e723e2..7606deaa2a 100644 --- a/var/spack/repos/builtin/packages/sw4lite/package.py +++ b/var/spack/repos/builtin/packages/sw4lite/package.py @@ -23,32 +23,70 @@ # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA ############################################################################## from spack import * +import glob class Sw4lite(MakefilePackage): """Sw4lite is a bare bone version of SW4 intended for testing performance optimizations in a few important numerical kernels of SW4.""" - tags = ['proxy-app'] + tags = ['proxy-app', 'ecp-proxy-app'] - homepage = "https://github.com/geodynamics/sw4lite" + homepage = "https://geodynamics.org/cig/software/sw4" url = "https://github.com/geodynamics/sw4lite/archive/v1.0.zip" version('develop', git='https://github.com/geodynamics/sw4lite', branch='master') version('1.0', '3d911165f4f2ff6d5f9c1bd56ab6723f') + variant('openmp', default=True, description='Build with OpenMP support') + variant('precision', default='double', values=('float', 'double'), + multi=False, description='Floating point precision') + variant('ckernel', default=False, description='C or Fortran kernel') + + depends_on('blas') + depends_on('lapack') depends_on('mpi') @property def build_targets(self): targets = [] spec = self.spec - targets.append('ckernel=yes') + + if spec.variants['precision'].value == 'double': + cxxflags = ['-I../src', '-I../src/double'] + else: + cxxflags = ['-I../src', '-I../src/float'] + cflags = [] + fflags = [] + + if '+openmp' in self.spec: + cflags.append('-DSW4_OPENMP') + cflags.append(self.compiler.openmp_flag) + cxxflags.append('-DSW4_OPENMP') + cxxflags.append(self.compiler.openmp_flag) + fflags.append(self.compiler.openmp_flag) + + if spec.variants['ckernel'].value is True: + cxxflags.append('-DSW4_CROUTINES') + targets.append('ckernel=yes') + targets.append('FC=' + spec['mpi'].mpifc) targets.append('CXX=' + spec['mpi'].mpicxx) + + targets.append('CFLAGS={0}'.format(' '.join(cflags))) + targets.append('CXXFLAGS={0}'.format(' '.join(cxxflags))) + targets.append('FFLAGS={0}'.format(' '.join(fflags))) + + targets.append('EXTRA_CXX_FLAGS=') + targets.append('EXTRA_FORT_FLAGS=') + lapack_blas = spec['lapack'].libs + spec['blas'].libs + targets.append('EXTRA_LINK_FLAGS={0}'.format(lapack_blas.ld_flags)) + return targets def install(self, spec, prefix): mkdir(prefix.bin) - install('optimize_c/sw4lite', prefix.bin) + exe_name = glob.glob('*/sw4lite')[0] + install(exe_name, prefix.bin) + install_tree('tests', prefix.tests) |