89 lines
3.5 KiB
Python
89 lines
3.5 KiB
Python
##############################################################################
|
|
# Copyright (c) 2013, Lawrence Livermore National Security, LLC.
|
|
# Produced at the Lawrence Livermore National Laboratory.
|
|
#
|
|
# This file is part of Spack.
|
|
# Written by Todd Gamblin, tgamblin@llnl.gov, All rights reserved.
|
|
# LLNL-CODE-647188
|
|
#
|
|
# For details, see https://scalability-llnl.github.io/spack
|
|
# Please also see the LICENSE file for our notice and the LGPL.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License (as published by
|
|
# the Free Software Foundation) version 2.1 dated February 1999.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the IMPLIED WARRANTY OF
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the terms and
|
|
# conditions of the GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
##############################################################################
|
|
from spack import *
|
|
import os
|
|
|
|
class Mpich(Package):
|
|
"""MPICH is a high performance and widely portable implementation of
|
|
the Message Passing Interface (MPI) standard."""
|
|
homepage = "http://www.mpich.org"
|
|
url = "http://www.mpich.org/static/downloads/3.0.4/mpich-3.0.4.tar.gz"
|
|
list_url = "http://www.mpich.org/static/downloads/"
|
|
list_depth = 2
|
|
|
|
version('3.0.4', '9c5d5d4fe1e17dd12153f40bc5b6dbc0')
|
|
|
|
provides('mpi@:3', when='@3:')
|
|
provides('mpi@:1', when='@1:')
|
|
|
|
|
|
def setup_dependent_environment(self, module, spec, dep_spec):
|
|
"""For dependencies, make mpicc's use spack wrapper."""
|
|
os.environ['MPICH_CC'] = 'cc'
|
|
os.environ['MPICH_CXX'] = 'c++'
|
|
os.environ['MPICH_F77'] = 'f77'
|
|
os.environ['MPICH_F90'] = 'f90'
|
|
|
|
|
|
def install(self, spec, prefix):
|
|
config_args = ["--prefix=" + prefix,
|
|
"--enable-shared"]
|
|
|
|
# TODO: Spack should make it so that you can't actually find
|
|
# these compilers if they're "disabled" for the current
|
|
# compiler configuration.
|
|
if not self.compiler.f77:
|
|
config_args.append("--disable-f77")
|
|
|
|
if not self.compiler.fc:
|
|
config_args.append("--disable-fc")
|
|
|
|
configure(*config_args)
|
|
make()
|
|
make("install")
|
|
|
|
self.filter_compilers()
|
|
|
|
|
|
def filter_compilers(self):
|
|
"""Run after install to make the MPI compilers use the
|
|
compilers that Spack built the package with.
|
|
|
|
If this isn't done, they'll have CC, CXX, F77, and FC set
|
|
to Spack's generic cc, c++, f77, and f90. We want them to
|
|
be bound to whatever compiler they were built with.
|
|
"""
|
|
bin = self.prefix.bin
|
|
mpicc = os.path.join(bin, 'mpicc')
|
|
mpicxx = os.path.join(bin, 'mpicxx')
|
|
mpif77 = os.path.join(bin, 'mpif77')
|
|
mpif90 = os.path.join(bin, 'mpif90')
|
|
|
|
kwargs = { 'ignore_absent' : True, 'backup' : False, 'string' : True }
|
|
filter_file('CC="cc"', 'CC="%s"' % self.compiler.cc, mpicc, **kwargs)
|
|
filter_file('CXX="c++"', 'CXX="%s"' % self.compiler.cxx, mpicxx, **kwargs)
|
|
filter_file('F77="f77"', 'F77="%s"' % self.compiler.f77, mpif77, **kwargs)
|
|
filter_file('FC="f90"', 'FC="%s"' % self.compiler.fc, mpif90, **kwargs)
|