1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
# Copyright 2013-2022 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)
from spack.error import SpackError
from spack.package import *
def class_validator(values):
"""1, 2, 3, 4, 5, 6"""
values = int(values[0])
if values < 1 or values > 6:
error_msg = ("class: Choose one of the following:\n"
"1 - 8x8x8x32 (default MPI config: 1x1x1)\n"
"2 - 32x32x32x32 (default MPI config: 4x4x4)\n"
"3 - 64x64x64x32 (default MPI config: 8x8x8)\n"
"4 - 160x160x160x160 (default MPI config: 20x20x20)\n"
"5 - 256x256x256x256 (default MPI config: 32x32x32)\n"
"6 - 192x192x192x192 (default MPI config: 24x24x24)")
raise SpackError(error_msg)
class CcsQcd(MakefilePackage):
"""This program benchmarks the performance of a linear equation solver
with a large sparse coefficient matrix appering in a lattice QCD
probrem. Lattice QCD describes the property of protons, neutrons and
neucleons in terms of more fundamental interacting elementary particles
gluon and quarks."""
homepage = "https://github.com/fiber-miniapp/ccs-qcd"
git = "https://github.com/fiber-miniapp/ccs-qcd.git"
tags = ['hep']
version('master', branch='master')
version('1.2.1', commit='d7c6b6923f35a824e997ba8db5bd12dc20dda45c')
variant('class', default=1, values=class_validator,
description='This miniapp has five problem classes, for which the'
' first three are relatively small problems just for testing'
' this miniapp itself. The remaining two are the target problem'
' sizes for the HPCI FS evaluation.',
multi=False)
depends_on('mpi')
parallel = False
def edit(self, spec, prefix):
if spec.satisfies('%gcc') and spec.satisfies('arch=aarch64:'):
chgopt = 'FFLAGS =-O3 -ffixed-line-length-132 -g -fopenmp' \
' -mcmodel=large -funderscoring'
filter_file('FFLAGS =.*', chgopt, join_path(
self.stage.source_path, 'src', 'make.gfortran.inc'))
if '%fj' in spec:
filter_file('mpifrtpx', spec['mpi'].mpifc, join_path(
self.stage.source_path, 'src', 'make.fx10.inc'))
filter_file('mpifccpx', spec['mpi'].mpicc, join_path(
self.stage.source_path, 'src', 'make.fx10.inc'))
else:
filter_file('mpif90', spec['mpi'].mpifc, join_path(
self.stage.source_path, 'src', 'make.gfortran.inc'))
filter_file('mpicc', spec['mpi'].mpicc, join_path(
self.stage.source_path, 'src', 'make.gfortran.inc'))
def build(self, spec, prefix):
ccs_class = 'CLASS=' + spec.variants['class'].value
with working_dir('src'):
if '%fj' in spec:
make('MAKE_INC=make.fx10.inc', ccs_class)
else:
make('MAKE_INC=make.gfortran.inc', ccs_class)
def install(self, spec, prefix):
mkdirp(prefix.bin)
install('./src/ccs_qcd_solver_bench_class' +
spec.variants['class'].value, prefix.bin)
|