zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cray-mpich: add variant 'wrappers' (#30249)

Browse Source
This commit is contained in:
Sergey Kosukhin 2022-05-10 19:22:04 +02:00 committed by GitHub
parent 6c309bbb32
commit 56d76766b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
var/spack/repos/builtin/packages/cray-mpich/package.py
View File

@ -29,6 +29,10 @@ class CrayMpich(Package):
version('7.7.14') version('7.7.14')
version('7.7.13') version('7.7.13')
# cray-mpich 8.1.7: features MPI compiler wrappers
variant('wrappers', default=True, when='@8.1.7:',
description='enable MPI wrappers')
provides('mpi@3') provides('mpi@3')
canonical_names = { canonical_names = {
@ -59,17 +63,16 @@ def external_prefix(self):
return os.path.dirname(os.path.normpath(libdir)) return os.path.dirname(os.path.normpath(libdir))
def setup_run_environment(self, env): def setup_run_environment(self, env):
if(self.spec.satisfies('@:8.1.6')): if '+wrappers' in self.spec:
env.set('MPICC', spack_cc)
env.set('MPICXX', spack_cxx)
env.set('MPIF77', spack_fc)
env.set('MPIF90', spack_fc)
else:
# cray-mpich 8.1.7: features MPI compiler wrappers
env.set('MPICC', join_path(self.prefix.bin, 'mpicc')) env.set('MPICC', join_path(self.prefix.bin, 'mpicc'))
env.set('MPICXX', join_path(self.prefix.bin, 'mpicxx')) env.set('MPICXX', join_path(self.prefix.bin, 'mpicxx'))
env.set('MPIF77', join_path(self.prefix.bin, 'mpif77')) env.set('MPIF77', join_path(self.prefix.bin, 'mpif77'))
env.set('MPIF90', join_path(self.prefix.bin, 'mpif90')) env.set('MPIF90', join_path(self.prefix.bin, 'mpif90'))
else:
env.set('MPICC', spack_cc)
env.set('MPICXX', spack_cxx)
env.set('MPIF77', spack_fc)
env.set('MPIF90', spack_fc)
def setup_dependent_build_environment(self, env, dependent_spec): def setup_dependent_build_environment(self, env, dependent_spec):
self.setup_run_environment(env) self.setup_run_environment(env)
@ -81,17 +84,16 @@ def setup_dependent_build_environment(self, env, dependent_spec):
def setup_dependent_package(self, module, dependent_spec): def setup_dependent_package(self, module, dependent_spec):
spec = self.spec spec = self.spec
if(spec.satisfies('@:8.1.6')): if '+wrappers' in spec:
spec.mpicc = join_path(self.prefix.bin, 'mpicc')
spec.mpicxx = join_path(self.prefix.bin, 'mpicxx')
spec.mpifc = join_path(self.prefix.bin, 'mpif90')
spec.mpif77 = join_path(self.prefix.bin, 'mpif77')
else:
spec.mpicc = spack_cc spec.mpicc = spack_cc
spec.mpicxx = spack_cxx spec.mpicxx = spack_cxx
spec.mpifc = spack_fc spec.mpifc = spack_fc
spec.mpif77 = spack_f77 spec.mpif77 = spack_f77
else:
# cray-mpich 8.1.7: features MPI compiler wrappers
spec.mpicc = join_path(self.prefix.bin, 'mpicc')
spec.mpicxx = join_path(self.prefix.bin, 'mpicxx')
spec.mpifc = join_path(self.prefix.bin, 'mpif90')
spec.mpif77 = join_path(self.prefix.bin, 'mpif77')
def install(self, spec, prefix): def install(self, spec, prefix):
raise InstallError( raise InstallError(