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
|
# 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 PyCuml(PythonPackage):
"""cuML is a suite of libraries that implement machine
learning algorithms and mathematical primitives functions
that share compatible APIs with other RAPIDS projects."""
homepage = "https://github.com/rapidsai/cuml"
url = "https://github.com/rapidsai/cuml/archive/v0.15.0.tar.gz"
version("0.15.0", sha256="5c9c656ae4eaa94a426e07d7385fd5ea0e5dc7abff806af2941aee10d4ca99c7")
depends_on("python@3.7:", type=("build", "run"))
depends_on("py-setuptools", type="build")
depends_on("py-cython", type="build")
depends_on("py-numpy", type=("build", "run"))
depends_on("py-numba", type=("build", "run"))
depends_on("py-cudf", type=("build", "run"))
depends_on("cuda")
depends_on("py-cupy", type=("build", "run"))
depends_on("py-scipy", type=("build", "run"))
depends_on("treelite+python", type=("build", "run"))
depends_on("py-joblib", type=("build", "run"))
depends_on("py-scikit-learn", type=("build", "run"))
for v in ("11.0", "10.2", "10.1"):
depends_on("libcumlprims@0.15.0-cuda{0}_gdbd0d39_0".format(v), when="^cuda@{0}".format(v))
for v in ("@0.15.0",):
depends_on("libcuml{0}".format(v), when=v)
build_directory = "python"
|