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
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
# Copyright 2013-2024 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.package import *
from spack.pkg.builtin.boost import Boost
class Gunrock(CMakePackage, CudaPackage):
"""High-Performance Graph Primitives on GPUs"""
homepage = "https://gunrock.github.io/docs/"
git = "https://github.com/gunrock/gunrock.git"
license("Apache-2.0")
version("master", submodules=True)
version("1.2", submodules=True, tag="v1.2", commit="5ee3df50c45f702eb247ef1abcea7a490b60b2ea")
# v1.1 build is broken. See:
# https://github.com/gunrock/gunrock/issues/777
version("1.1", submodules=True, tag="v1.1", commit="7c197d6a498806fcfffd1f9304c663379a77f5e4")
version("1.0", submodules=True, tag="v1.0", commit="04279c89f394b97c81c63ad286048893e02f769e")
version(
"0.5.1", submodules=True, tag="v0.5.1", commit="0c9a96dd64ddeecae9208644631983fc889b32b4"
)
version("0.5", submodules=True, tag="v0.5", commit="91a6218868ecdab7986ef42b4b76ff17eec61ca3")
version("0.4", submodules=True, tag="v0.4", commit="4c33a0d5a7ad7e468362c9c081a567450a78ab97")
version(
"0.3.1", submodules=True, tag="v0.3.1", commit="897f170e3006e58c9a602201e5f9fc56162a3cb9"
)
version("0.3", submodules=True, tag="v0.3", commit="0b146a70f52f699a2a50d1b1aa26b92c45e834d7")
version("0.2", submodules=True, tag="v0.2", commit="f9d85343ee68c65567184d74021b9483cd142ea0")
version("0.1", submodules=True, tag="v0.1", commit="4c00284f6b7d490a83fa7afe5cdff60923316448")
variant("cuda", default=True, description="Build with Cuda support")
variant("lib", default=True, description="Build main gunrock library")
variant("shared_libs", default=True, description="Turn off to build for static libraries")
variant("tests", default=True, description="Build functional tests / examples")
variant(
"mgpu_tests",
default=False,
description=(
"Builds Gunrock applications and enables the ctest framework "
"for single GPU implementations"
),
)
variant(
"cuda_verbose_ptxas",
default=False,
description="Enable verbose output from the PTXAS assembler",
)
variant("google_tests", default=False, description="Build unit tests using googletest")
variant(
"code_coverage", default=False, description="run code coverage on Gunrock's source code"
)
# apps
msg = "select either all or individual applications"
variant(
"applications",
values=disjoint_sets(
("all",),
("bc", "bfs", "cc", "pr", "sssp", "dobfs", "hits", "salsa", "mst", "wtf", "topk"),
)
.allow_empty_set()
.with_default("all")
.with_error(msg),
description="Application to be built",
)
variant("boost", default=True, description="Build with Boost")
variant("metis", default=False, description="Build with Metis support")
depends_on("googletest", when="+google_tests")
depends_on("lcov", when="+code_coverage")
depends_on("boost@1.58.0:", when="+boost")
# TODO: replace this with an explicit list of components of Boost,
# for instance depends_on('boost +filesystem')
# See https://github.com/spack/spack/pull/22303 for reference
depends_on(Boost.with_default_variants, when="+boost")
depends_on("metis", when="+metis")
conflicts(
"cuda_arch=none",
when="+cuda",
msg='Must specify CUDA compute capabilities of your GPU. \
See "spack info gunrock"',
)
def cmake_args(self):
spec = self.spec
from_variant = self.define_from_variant
args = [
from_variant("GUNROCK_BUILD_LIB", "lib"),
from_variant("GUNROCK_BUILD_SHARED_LIBS", "shared_libs"),
from_variant("GUNROCK_BUILD_TESTS", "tests"),
from_variant("GUNROCK_MGPU_TESTS", "mgpu_tests"),
from_variant("CUDA_VERBOSE_PTXAS", "cuda_verbose_ptxas"),
from_variant("GUNROCK_GOOGLE_TESTS", "google_tests"),
from_variant("GUNROCK_CODE_COVERAGE", "code_coverage"),
]
# turn off auto detect, which undoes custom cuda arch options
args.append("-DCUDA_AUTODETECT_GENCODE=OFF")
cuda_arch_list = self.spec.variants["cuda_arch"].value
if cuda_arch_list[0] != "none":
for carch in cuda_arch_list:
args.append("-DGUNROCK_GENCODE_SM" + carch + "=ON")
app_list = self.spec.variants["applications"].value
if app_list[0] != "none":
args.extend(
[
"-DGUNROCK_BUILD_APPLICATIONS={0}".format(
"ON" if spec.satisfies("applications=all") else "OFF"
),
"-DGUNROCK_APP_BC={0}".format(
"ON" if spec.satisfies("applications=bc") else "OFF"
),
"-DGUNROCK_APP_BFS={0}".format(
"ON" if spec.satisfies("applications=bfs") else "OFF"
),
"-DGUNROCK_APP_CC={0}".format(
"ON" if spec.satisfies("applications=cc") else "OFF"
),
"-DGUNROCK_APP_PR={0}".format(
"ON" if spec.satisfies("applications=pr") else "OFF"
),
"-DGUNROCK_APP_SSSP={0}".format(
"ON" if spec.satisfies("applications=sssp") else "OFF"
),
"-DGUNROCK_APP_DOBFS={0}".format(
"ON" if spec.satisfies("applications=dobfs") else "OFF"
),
"-DGUNROCK_APP_HITS={0}".format(
"ON" if spec.satisfies("applications=hits") else "OFF"
),
"-DGUNROCK_APP_SALSA={0}".format(
"ON" if spec.satisfies("applications=salsa") else "OFF"
),
"-DGUNROCK_APP_MST={0}".format(
"ON" if spec.satisfies("applications=mst") else "OFF"
),
"-DGUNROCK_APP_WTF={0}".format(
"ON" if spec.satisfies("applications=wtf") else "OFF"
),
"-DGUNROCK_APP_TOPK={0}".format(
"ON" if spec.satisfies("applications=topk") else "OFF"
),
]
)
return args
def install(self, spec, prefix):
with working_dir(self.build_directory):
install_tree("lib", prefix.lib)
# bin dir is created only if tests/examples are built
if "+tests" in spec:
install_tree("bin", prefix.bin)
|
