Add a package for the NAS Parallel Benchmarks (NPB) (#3762)
This commit is contained in:
Adam J. Stewart
GitHub
parent
7c4ed0a70a
commit
ed4f276b68
202
var/spack/repos/builtin/packages/npb/package.py
Normal file
202
var/spack/repos/builtin/packages/npb/package.py
Normal file
@ -0,0 +1,202 @@
|
|||||||
|
##############################################################################
|
||||||
|
# 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 Npb(MakefilePackage):
|
||||||
|
"""The NAS Parallel Benchmarks (NPB) are a small set of programs
|
||||||
|
designed to help evaluate the performance of parallel supercomputers.
|
||||||
|
The benchmarks are derived from computational fluid dynamics (CFD)
|
||||||
|
applications and consist of five kernels and three pseudo-applications
|
||||||
|
in the original "pencil-and-paper" specification (NPB 1). The benchmark
|
||||||
|
suite has been extended to include new benchmarks for unstructured
|
||||||
|
adaptive mesh, parallel I/O, multi-zone applications, and computational
|
||||||
|
grids. Problem sizes in NPB are predefined and indicated as different
|
||||||
|
classes. Reference implementations of NPB are available in commonly-used
|
||||||
|
programming models like MPI and OpenMP (NPB 2 and NPB 3)."""
|
||||||
|
|
||||||
|
homepage = "https://www.nas.nasa.gov/publications/npb.html"
|
||||||
|
url = "https://www.nas.nasa.gov/assets/npb/NPB3.3.1.tar.gz"
|
||||||
|
|
||||||
|
version('3.3.1', '8e5ec2c819480759725df67833619911')
|
||||||
|
|
||||||
|
# Valid Benchmark Names
|
||||||
|
valid_names = (
|
||||||
|
'is', # Integer Sort, random memory access
|
||||||
|
'ep', # Embarrassingly Parallel
|
||||||
|
'cg', # Conjugate Gradient, irregular memory access and communication
|
||||||
|
'mg', # Multi-Grid on a sequence of meshes, long- and short-distance
|
||||||
|
# communication, memory intensive
|
||||||
|
'ft', # discrete 3D fast Fourier Transform, all-to-all communication
|
||||||
|
'bt', # Block Tri-diagonal solver
|
||||||
|
'sp', # Scalar Penta-diagonal solver
|
||||||
|
'lu', # Lower-Upper Gauss-Seidel solver
|
||||||
|
)
|
||||||
|
|
||||||
|
# Valid Benchmark Classes
|
||||||
|
valid_classes = (
|
||||||
|
'S', # Small for quick test purposes
|
||||||
|
'W', # Workstation size
|
||||||
|
'A', 'B', 'C', # standard test problems
|
||||||
|
# ~4X size increase going from one class to the next
|
||||||
|
'D', 'E', # large test problems
|
||||||
|
# ~16X size increase from each of the previous classes
|
||||||
|
)
|
||||||
|
|
||||||
|
# TODO: Combine these into a single mutually exclusive variant
|
||||||
|
variant('mpi', default=True, description='Build the MPI implementation')
|
||||||
|
variant('openmp', default=False, description='Build the OpenMP implementation')
|
||||||
|
variant('serial', default=False, description='Build the serial version')
|
||||||
|
|
||||||
|
# TODO: Convert these to non-exclusive multi-valued variants
|
||||||
|
variant('names', default=','.join(valid_names),
|
||||||
|
description='Benchmark names (comma separated list)')
|
||||||
|
variant('classes', default=','.join(valid_classes),
|
||||||
|
description='Benchmark classes (comma separated list)')
|
||||||
|
# This variant only applies to the MPI implementation
|
||||||
|
variant('nprocs', default='1,2,4,8,16,32,64,128',
|
||||||
|
description='Number of processes (comma separated list)')
|
||||||
|
|
||||||
|
depends_on('mpi@2:', when='+mpi')
|
||||||
|
|
||||||
|
phases = ['edit', 'install']
|
||||||
|
|
||||||
|
# Cannot be built in parallel
|
||||||
|
parallel = False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def build_directory(self):
|
||||||
|
if '+mpi' in self.spec:
|
||||||
|
implementation = 'MPI'
|
||||||
|
elif '+openmp' in self.spec:
|
||||||
|
implementation = 'OMP'
|
||||||
|
elif '+serial' in self.spec:
|
||||||
|
implementation = 'SER'
|
||||||
|
else:
|
||||||
|
raise RuntimeError('You must choose an implementation to build')
|
||||||
|
|
||||||
|
return 'NPB{0}-{1}'.format(self.version.up_to(2), implementation)
|
||||||
|
|
||||||
|
def edit(self, spec, prefix):
|
||||||
|
names = spec.variants['names'].value.split(',')
|
||||||
|
classes = spec.variants['classes'].value.split(',')
|
||||||
|
nprocs = spec.variants['nprocs'].value.split(',')
|
||||||
|
|
||||||
|
if '+mpi' in spec:
|
||||||
|
definitions = {
|
||||||
|
# Parallel Fortran
|
||||||
|
'MPIF77': spec['mpi'].mpif77,
|
||||||
|
'FLINK': spec['mpi'].mpif77,
|
||||||
|
'FMPI_LIB': spec['mpi'].libs.ld_flags,
|
||||||
|
'FMPI_INC': '-I' + spec['mpi'].prefix.include,
|
||||||
|
'FFLAGS': '-O3',
|
||||||
|
'FLINKFLAGS': '-O3',
|
||||||
|
# Parallel C
|
||||||
|
'MPICC': spec['mpi'].mpicc,
|
||||||
|
'CLINK': spec['mpi'].mpicc,
|
||||||
|
'CMPI_LIB': spec['mpi'].libs.ld_flags,
|
||||||
|
'CMPI_INC': '-I' + spec['mpi'].prefix.include,
|
||||||
|
'CFLAGS': '-O3',
|
||||||
|
'CLINKFLAGS': '-O3',
|
||||||
|
# Utilities C
|
||||||
|
'CC': spack_cc + ' -g',
|
||||||
|
'BINDIR': prefix.bin,
|
||||||
|
'RAND': 'randi8',
|
||||||
|
}
|
||||||
|
elif '+openmp' in spec:
|
||||||
|
definitions = {
|
||||||
|
# Parallel Fortran
|
||||||
|
'F77': spack_f77,
|
||||||
|
'FLINK': spack_f77,
|
||||||
|
'F_LIB': '',
|
||||||
|
'F_INC': '',
|
||||||
|
'FFLAGS': '-O3 ' + self.compiler.openmp_flag,
|
||||||
|
'FLINKFLAGS': '-O3 ' + self.compiler.openmp_flag,
|
||||||
|
# Parallel C
|
||||||
|
'CC': spack_cc,
|
||||||
|
'CLINK': spack_cc,
|
||||||
|
'C_LIB': '-lm',
|
||||||
|
'C_INC': '',
|
||||||
|
'CFLAGS': '-O3 ' + self.compiler.openmp_flag,
|
||||||
|
'CLINKFLAGS': '-O3 ' + self.compiler.openmp_flag,
|
||||||
|
# Utilities C
|
||||||
|
'UCC': spack_cc,
|
||||||
|
'BINDIR': prefix.bin,
|
||||||
|
'RAND': 'randi8',
|
||||||
|
'WTIME': 'wtime.c',
|
||||||
|
}
|
||||||
|
elif '+serial' in spec:
|
||||||
|
definitions = {
|
||||||
|
# Parallel Fortran
|
||||||
|
'F77': spack_f77,
|
||||||
|
'FLINK': spack_f77,
|
||||||
|
'F_LIB': '',
|
||||||
|
'F_INC': '',
|
||||||
|
'FFLAGS': '-O3',
|
||||||
|
'FLINKFLAGS': '-O3',
|
||||||
|
# Parallel C
|
||||||
|
'CC': spack_cc,
|
||||||
|
'CLINK': spack_cc,
|
||||||
|
'C_LIB': '-lm',
|
||||||
|
'C_INC': '',
|
||||||
|
'CFLAGS': '-O3',
|
||||||
|
'CLINKFLAGS': '-O3',
|
||||||
|
# Utilities C
|
||||||
|
'UCC': spack_cc,
|
||||||
|
'BINDIR': prefix.bin,
|
||||||
|
'RAND': 'randi8',
|
||||||
|
'WTIME': 'wtime.c',
|
||||||
|
}
|
||||||
|
|
||||||
|
with working_dir(self.build_directory):
|
||||||
|
with open('config/make.def', 'w') as make_def:
|
||||||
|
for key in definitions:
|
||||||
|
make_def.write('{0} = {1}\n'.format(
|
||||||
|
key, definitions[key]))
|
||||||
|
|
||||||
|
with open('config/suite.def', 'w') as suite_def:
|
||||||
|
for name in names:
|
||||||
|
for classname in classes:
|
||||||
|
# Classes C, D and E are not available for DT
|
||||||
|
if name == 'dt' and classname in ('C', 'D', 'E'):
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Class E is not available for IS
|
||||||
|
if name == 'is' and classname == 'E':
|
||||||
|
continue
|
||||||
|
|
||||||
|
if '+mpi' in spec:
|
||||||
|
for nproc in nprocs:
|
||||||
|
suite_def.write('{0}\t{1}\t{2}\n'.format(
|
||||||
|
name, classname, nproc))
|
||||||
|
else:
|
||||||
|
suite_def.write('{0}\t{1}\n'.format(
|
||||||
|
name, classname))
|
||||||
|
|
||||||
|
def install(self, spec, prefix):
|
||||||
|
mkdir(prefix.bin)
|
||||||
|
|
||||||
|
with working_dir(self.build_directory):
|
||||||
|
make('suite')
|
||||||
Loading…
Reference in New Issue
Block a user