Merge branch 'develop' of https://github.com/llnl/spack into develop
This commit is contained in:
@@ -27,6 +27,26 @@
|
||||
from spack import *
|
||||
|
||||
|
||||
def _verbs_dir():
|
||||
"""
|
||||
Try to find the directory where the OpenFabrics verbs package is
|
||||
installed. Return None if not found.
|
||||
"""
|
||||
try:
|
||||
# Try to locate Verbs by looking for a utility in the path
|
||||
ibv_devices = which("ibv_devices")
|
||||
# Run it (silently) to ensure it works
|
||||
ibv_devices(output=str, error=str)
|
||||
# Get path to executable
|
||||
path = ibv_devices.exe[0]
|
||||
# Remove executable name and "bin" directory
|
||||
path = os.path.dirname(path)
|
||||
path = os.path.dirname(path)
|
||||
return path
|
||||
except:
|
||||
return None
|
||||
|
||||
|
||||
class Openmpi(Package):
|
||||
"""Open MPI is a project combining technologies and resources from
|
||||
several other projects (FT-MPI, LA-MPI, LAM/MPI, and PACX-MPI)
|
||||
@@ -54,7 +74,7 @@ class Openmpi(Package):
|
||||
variant('psm', default=False, description='Build support for the PSM library.')
|
||||
variant('psm2', default=False, description='Build support for the Intel PSM2 library.')
|
||||
variant('pmi', default=False, description='Build support for PMI-based launchers')
|
||||
variant('verbs', default=False, description='Build support for OpenFabrics verbs.')
|
||||
variant('verbs', default=_verbs_dir() is not None, description='Build support for OpenFabrics verbs.')
|
||||
variant('mxm', default=False, description='Build Mellanox Messaging support')
|
||||
|
||||
variant('thread_multiple', default=False, description='Enable MPI_THREAD_MULTIPLE support')
|
||||
@@ -113,7 +133,6 @@ def install(self, spec, prefix):
|
||||
# Fabrics
|
||||
'--with-psm' if '+psm' in spec else '--without-psm',
|
||||
'--with-psm2' if '+psm2' in spec else '--without-psm2',
|
||||
('--with-%s' % self.verbs) if '+verbs' in spec else ('--without-%s' % self.verbs),
|
||||
'--with-mxm' if '+mxm' in spec else '--without-mxm',
|
||||
# Other options
|
||||
'--enable-mpi-thread-multiple' if '+thread_multiple' in spec else '--disable-mpi-thread-multiple',
|
||||
@@ -121,6 +140,14 @@ def install(self, spec, prefix):
|
||||
'--with-sqlite3' if '+sqlite3' in spec else '--without-sqlite3',
|
||||
'--enable-vt' if '+vt' in spec else '--disable-vt'
|
||||
])
|
||||
if '+verbs' in spec:
|
||||
path = _verbs_dir()
|
||||
if path is not None:
|
||||
config_args.append('--with-%s=%s' % (self.verbs, path))
|
||||
else:
|
||||
config_args.append('--with-%s' % self.verbs)
|
||||
else:
|
||||
config_args.append('--without-%s' % self.verbs)
|
||||
|
||||
# TODO: use variants for this, e.g. +lanl, +llnl, etc.
|
||||
# use this for LANL builds, but for LLNL builds, we need:
|
||||
|
62
var/spack/repos/builtin/packages/stream/package.py
Normal file
62
var/spack/repos/builtin/packages/stream/package.py
Normal file
@@ -0,0 +1,62 @@
|
||||
##############################################################################
|
||||
# Copyright (c) 2013-2016, Lawrence Livermore National Security, LLC.
|
||||
# Produced at the Lawrence Livermore National Laboratory.
|
||||
#
|
||||
# This file is part of Spack.
|
||||
# Created by Todd Gamblin, tgamblin@llnl.gov, All rights reserved.
|
||||
# LLNL-CODE-647188
|
||||
#
|
||||
# For details, see https://github.com/llnl/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 Lesser General Public License (as
|
||||
# published by the Free Software Foundation) version 2.1, 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 Lesser 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 *
|
||||
|
||||
|
||||
class Stream(Package):
|
||||
"""The STREAM benchmark is a simple synthetic benchmark program that
|
||||
measures sustainable memory bandwidth (in MB/s) and the corresponding
|
||||
computation rate for simple vector kernels."""
|
||||
|
||||
homepage = "https://www.cs.virginia.edu/stream/ref.html"
|
||||
|
||||
version('5.10', git='https://github.com/jeffhammond/STREAM.git')
|
||||
|
||||
variant('openmp', default=False, description='Build with OpenMP support')
|
||||
|
||||
def patch(self):
|
||||
makefile = FileFilter('Makefile')
|
||||
|
||||
# Use the Spack compiler wrappers
|
||||
makefile.filter('CC = .*', 'CC = cc')
|
||||
makefile.filter('FC = .*', 'FC = f77')
|
||||
|
||||
cflags = '-O2'
|
||||
fflags = '-O2'
|
||||
if '+openmp' in self.spec:
|
||||
cflags += ' ' + self.compiler.openmp_flag
|
||||
fflags += ' ' + self.compiler.openmp_flag
|
||||
|
||||
# Set the appropriate flags for this compiler
|
||||
makefile.filter('CFLAGS = .*', 'CFLAGS = {0}'.format(cflags))
|
||||
makefile.filter('FFLAGS = .*', 'FFLAGS = {0}'.format(fflags))
|
||||
|
||||
def install(self, spec, prefix):
|
||||
make()
|
||||
|
||||
# Manual installation
|
||||
mkdir(prefix.bin)
|
||||
install('stream_c.exe', prefix.bin)
|
||||
install('stream_f.exe', prefix.bin)
|
Reference in New Issue
Block a user