diff --git a/var/spack/repos/builtin/packages/spectrum-mpi/package.py b/var/spack/repos/builtin/packages/spectrum-mpi/package.py index 2ada9c593ec..e094f1c3ad9 100644 --- a/var/spack/repos/builtin/packages/spectrum-mpi/package.py +++ b/var/spack/repos/builtin/packages/spectrum-mpi/package.py @@ -46,6 +46,11 @@ def setup_dependent_package(self, module, dependent_spec): self.spec.mpicxx = join_path(self.prefix.bin, 'mpixlC') self.spec.mpif77 = join_path(self.prefix.bin, 'mpixlf') self.spec.mpifc = join_path(self.prefix.bin, 'mpixlf') + elif '%pgi' in dependent_spec: + self.spec.mpicc = join_path(self.prefix.bin, 'mpipgicc') + self.spec.mpicxx = join_path(self.prefix.bin, 'mpipgic++') + self.spec.mpif77 = join_path(self.prefix.bin, 'mpipgifort') + self.spec.mpifc = join_path(self.prefix.bin, 'mpipgifort') else: self.spec.mpicc = join_path(self.prefix.bin, 'mpicc') self.spec.mpicxx = join_path(self.prefix.bin, 'mpicxx') @@ -58,6 +63,11 @@ def setup_dependent_environment(self, spack_env, run_env, dependent_spec): spack_env.set('MPICXX', join_path(self.prefix.bin, 'mpixlC')) spack_env.set('MPIF77', join_path(self.prefix.bin, 'mpixlf')) spack_env.set('MPIF90', join_path(self.prefix.bin, 'mpixlf')) + elif '%pgi' in dependent_spec: + spack_env.set('MPICC', join_path(self.prefix.bin, 'mpipgicc')) + spack_env.set('MPICXX', join_path(self.prefix.bin, 'mpipgic++')) + spack_env.set('MPIF77', join_path(self.prefix.bin, 'mpipgifort')) + spack_env.set('MPIF90', join_path(self.prefix.bin, 'mpipgifort')) else: spack_env.set('MPICC', join_path(self.prefix.bin, 'mpicc')) spack_env.set('MPICXX', join_path(self.prefix.bin, 'mpic++'))