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
|
# 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)
from spack.package import *
class Cmdstan(MakefilePackage):
"""CmdStan is the command line interface to Stan."""
homepage = "https://mc-stan.org/users/interfaces/cmdstan"
url = "https://github.com/stan-dev/cmdstan/releases/download/v2.30.1/cmdstan-2.30.1.tar.gz"
license("BSD-3-Clause")
version("2.30.1", sha256="bab76dcefa7f4c955595c0bf0496770507fc6ab0df5896e8cf8c2db0a17eedb9")
variant("threads", default=True, description="enable thread support")
variant("opencl", default=False, description="enable OpenCl support")
variant("mpi", default=False, description="enable MPI support")
depends_on("opencl", when="+opencl")
depends_on("mpi", when="+mpi")
build_targets = ["build"]
filter_compiler_wrappers("local", relative_root="make")
def edit(self, spec, prefix):
if spec.compiler.name == "intel":
cxx_type = "icc"
else:
cxx_type = spec.compiler.name
if "+mpi" in spec:
cxx = spec["mpi"].mpicxx
else:
cxx = spack_cxx
make_options = [
"CXX={0}\n".format(cxx),
"CXXFLAGS+= -O2 -funroll-loops\n",
"LDFLAGS+={0}{1}\n".format(
self.compiler.cc_rpath_arg,
join_path(prefix, "stan", "lib", "stan_math", "lib", "tbb"),
),
"STANCFLAGS+= --warn-pedantic\n",
"TBB_CXX_TYPE={0}\n".format(cxx_type),
]
if "+threads" in spec:
make_options.append("STAN_THREADS=true\n")
if "+opencl" in spec:
make_options.append("STAN_OPENCL=true\n")
if "+mpi" in spec:
make_options.append("STAN_MPI=true\n")
filepath = join_path(self.stage.source_path, "make", "local")
with open(filepath, "w") as make_file:
make_file.writelines(make_options)
def install(self, spec, prefix):
make(join_path("examples", "bernoulli", "bernoulli"))
mkdir(prefix.bin)
with working_dir(self.build_directory):
copy("makefile", prefix)
copy_tree("make", prefix.make)
copy_tree("examples", prefix.examples)
copy_tree("lib", prefix.lib)
copy_tree("src", prefix.src)
copy_tree("stan", prefix.stan)
with working_dir(join_path(self.build_directory, "bin")):
install("diagnose", prefix.bin)
install("print", prefix.bin)
install("stanc", prefix.bin)
install("stansummary", prefix.bin)
def setup_run_environment(self, env):
env.set("CMDSTAN", self.prefix)
|
