Skip to content
Snippets Groups Projects
Unverified Commit 07422f95 authored by Kurt Sansom's avatar Kurt Sansom Committed by GitHub
Browse files

sprng: added new package (#17570)

parent 18d2682f
No related branches found
No related tags found
No related merge requests found
# Copyright 2013-2019 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
class Sprng(AutotoolsPackage):
"""SPRNG: A Scalable Library For Pseudorandom Number Generation
Sprng is a distributed process-aware random number generator that
avoids correlations in random number sequences across processes.
"""
maintainers = ['kayarre']
homepage = "http://www.sprng.org"
url = "http://www.sprng.org/Version5.0/sprng5.tar.bz2"
version('5.0', sha256='9172a495472cc24893e7489ce9b5654300dc60cba4430e436ce50d28eb749a66')
variant('mpi', default=True, description='Enable MPI support')
variant('fortran', default=False, description='Enable Fortran support')
depends_on('mpi', when='+mpi')
def url_for_version(self, version):
url = "http://www.sprng.org/Version{0}/sprng{1}.tar.bz2"
return url.format(version, version.up_to(1))
def configure_args(self):
configure_args = []
configure_args += self.with_or_without('mpi')
configure_args += self.with_or_without('fortran')
if '+mpi' in self.spec:
mpi_link_flags = self.spec['mpi:cxx'].libs.link_flags
configure_args.append('LIBS={0}'.format(mpi_link_flags))
configure_args.append('CC={0}'.format(self.spec['mpi'].mpicc))
configure_args.append('CXX={0}'.format(self.spec['mpi'].mpicxx))
if '+fortran' in self.spec:
configure_args.append('FC={0}'.format(self.spec['mpi'].mpifc))
return configure_args
# TODO: update after solution for virtual depedencies
@run_before('configure')
def mpicxx_check(self):
# print(self.spec['mpi:fortran'].libs.names)
if '+mpi' in self.spec:
if 'mpi_cxx' not in self.spec['mpi:cxx'].libs.names:
msg = 'SPRNG requires a mpi Cxx bindings to build'
raise RuntimeError(msg)
if '+fortran' in self.spec:
if 'fmpi' not in self.spec['fortran'].libs.names:
msg = ('SPRNG requires fortran mpi '
'libraries with mpi enabled')
raise RuntimeError(msg)
# raise RuntimeError("test")
# FIXME: update after features in #15702 are enabled
@run_after('build')
@on_package_attributes(run_tests=True)
def check_build(self):
def listisclose(a, b, rel_tol=1e-09, abs_tol=1.0e-20):
for ai, bi in zip(a, b):
if (not abs(ai - bi) <=
max(rel_tol * max(abs(ai), abs(bi)), abs_tol)):
return False
return True
# Build and run a small program to test the installed sprng library
spec = self.spec
print("Checking sprng installation...")
checkdir = "spack-check"
with working_dir(checkdir, create=True):
source = r"""
#include <cstdio>
#define SIMPLE_SPRNG /* simple interface */
#include "sprng_cpp.h" /* SPRNG header file */
#define SEED 985456376
int main() {
int seed = SEED;
int i, j;
double rn;
for(j=0; j < 5; ++j) {
init_sprng(seed, SPRNG_DEFAULT, j);
for (i=0; i<3; ++i) {
rn = sprng();
printf("%f ", rn);
}
}
printf("\n");
return 0;
}
"""
expected = [0.504272, 0.558437, 0.000848,
0.707488, 0.664048, 0.005616,
0.060190, 0.415195, 0.933915,
0.085215, 0.456461, 0.244497,
0.626037, 0.917948, 0.135160
]
with open("check.c", 'w') as f:
f.write(source)
if '+mpi' in spec:
cc = Executable(spec['mpi'].mpicxx)
else:
cc = Executable(self.compiler.cxx)
cc(*(['-c', "check.c"] + spec['sprng'].headers.cpp_flags.split()))
cc(*(['-o', "check",
"check.o"] + spec['sprng'].libs.ld_flags.split()))
try:
check = Executable('./check')
output = check(output=str)
except ProcessError:
output = ""
out2float = [float(num) for num in output.split(' ')]
success = listisclose(expected, out2float)
if not success:
print("Produced output does not match expected output.")
print("Expected output:")
print('-' * 80)
print(expected)
print('-' * 80)
print("Produced output:")
print('-' * 80)
print(output)
print('-' * 80)
raise RuntimeError("sprng install check failed")
else:
print("test passed")
shutil.rmtree(checkdir)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment