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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
# Copyright 2013-2023 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)
import sys
from spack.package import *
class HypreCmake(CMakePackage, CudaPackage):
"""Hypre is a library of high performance preconditioners that
features parallel multigrid methods for both structured and
unstructured grid problems."""
homepage = "http://computing.llnl.gov/project/linear_solvers/software.php"
url = "https://github.com/hypre-space/hypre/archive/v2.14.0.tar.gz"
git = "https://github.com/hypre-space/hypre.git"
maintainers("ulrikeyang", "osborn9", "balay")
test_requires_compiler = True
license("Apache-2.0")
version("develop", branch="master")
version("2.22.0", sha256="2c786eb5d3e722d8d7b40254f138bef4565b2d4724041e56a8fa073bda5cfbb5")
variant(
"shared",
default=(sys.platform != "darwin"),
description="Build shared library (disables static library)",
)
variant(
"superlu_dist", default=False, description="Activates support for SuperLU_Dist library"
)
variant("int64", default=False, description="Use 64bit integers")
variant("mixedint", default=False, description="Use 64bit integers while reducing memory use")
variant("complex", default=False, description="Use complex values")
variant("mpi", default=True, description="Enable MPI support")
variant("openmp", default=False, description="Enable OpenMP support")
variant("debug", default=False, description="Build debug instead of optimized version")
variant("unified_memory", default=False, description="Use unified memory")
depends_on("mpi", when="+mpi")
depends_on("blas")
depends_on("lapack")
depends_on("superlu-dist", when="+superlu_dist+mpi")
conflicts("+cuda", when="+int64")
conflicts("+unified_memory", when="~cuda")
def url_for_version(self, version):
if version >= Version("2.12.0"):
url = "https://github.com/hypre-space/hypre/archive/v{0}.tar.gz"
else:
url = "http://computing.llnl.gov/project/linear_solvers/download/hypre-{0}.tar.gz"
return url.format(version)
root_cmakelists_dir = "src"
def cmake_args(self):
from_variant = self.define_from_variant
args = [
from_variant("HYPRE_WITH_MPI", "mpi"),
from_variant("HYPRE_WITH_OPENMP", "openmp"),
from_variant("HYPRE_WITH_BIGINT", "int64"),
from_variant("HYPRE_WITH_MIXEDINT", "mixedint"),
from_variant("HYPRE_WITH_COMPLEX", "complex"),
from_variant("BUILD_SHARED_LIBS", "shared"),
from_variant("HYPRE_ENABLE_SHARED", "shared"),
from_variant("HYPRE_WITH_DSUPERLU", "superlu_dist"),
from_variant("HYPRE_WITH_CUDA", "cuda"),
from_variant("HYPRE_ENABLE_UNIFIED_MEMORY", "unified_memory"),
]
return args
def setup_build_environment(self, env):
if "+cuda" in self.spec:
env.set("CUDA_HOME", self.spec["cuda"].prefix)
env.set("CUDA_PATH", self.spec["cuda"].prefix)
cuda_arch = self.spec.variants["cuda_arch"].value
if cuda_arch:
arch_sorted = list(sorted(cuda_arch, reverse=True))
env.set("HYPRE_CUDA_SM", arch_sorted[0])
# In CUDA builds hypre currently doesn't handle flags correctly
env.append_flags("CXXFLAGS", "-O2" if "~debug" in self.spec else "-g")
extra_install_tests = join_path("src", "examples")
@run_after("install")
def cache_test_sources(self):
self.cache_extra_test_sources(self.extra_install_tests)
@property
def _cached_tests_work_dir(self):
"""The working directory for cached test sources."""
return join_path(self.test_suite.current_test_cache_dir, self.extra_install_tests)
def test(self):
"""Perform smoke test on installed HYPRE package."""
if "+mpi" not in self.spec:
print("Skipping: HYPRE must be installed with +mpi to run tests")
return
# Build copied and cached test examples
self.run_test(
"make",
["HYPRE_DIR={0}".format(self.prefix), "bigint"],
purpose="test: building selected examples",
work_dir=self._cached_tests_work_dir,
)
# Run the examples built above
for exe in ["./ex5big", "./ex15big"]:
self.run_test(
exe,
[],
[],
installed=False,
purpose="test: ensuring {0} runs".format(exe),
skip_missing=True,
work_dir=self._cached_tests_work_dir,
)
@property
def headers(self):
"""Export the main hypre header, HYPRE.h; all other headers can be found
in the same directory.
Sample usage: spec['hypre'].headers.cpp_flags
"""
hdrs = find_headers("HYPRE", self.prefix.include, recursive=False)
return hdrs or None
@property
def libs(self):
"""Export the hypre library.
Sample usage: spec['hypre'].libs.ld_flags
"""
is_shared = "+shared" in self.spec
libs = find_libraries("libHYPRE", root=self.prefix, shared=is_shared, recursive=True)
return libs or None
|
