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
|
# 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 PyPennylaneLightningKokkos(CMakePackage, PythonExtension, CudaPackage, ROCmPackage):
"""The PennyLane-Lightning-Kokkos plugin provides
a fast state-vector simulator with Kokkos kernels."""
homepage = "https://docs.pennylane.ai/projects/lightning-kokkos"
git = "https://github.com/PennyLaneAI/pennylane-lightning-kokkos.git"
url = "https://github.com/PennyLaneAI/pennylane-lightning-kokkos/archive/refs/tags/v0.32.0.tar.gz"
maintainers("AmintorDusko", "vincentmr")
license("Apache-2.0")
version("main", branch="main")
version("0.32.0", sha256="06f19dfb1073387ef9ee30c38ea44884844a771373256b694a0e1ceb87195bb2")
version("0.31.0", sha256="fe10322fee0fa7df45cd3a81d6c229a79c7dfa7f20ff7d67c65c9a28f494dc89")
version("0.30.0", sha256="7c8f0e0431f8052993cd8033a316f53590c7bf5419445d0725e214b93cbc661b")
version("0.29.1", sha256="f51ba7718defc7bb5064f690f381e04b2ec58cb09f22a171ae5f410860716e30")
depends_on("kokkos@:3.7.2", when="@:0.30", type=("run", "build"))
depends_on("kokkos@4:", when="@0.31:", type=("run", "build"))
# kokkos backends
backends = {
"cuda": [False, "Whether to build CUDA backend"],
"openmp": [False, "Whether to build OpenMP backend"],
"openmptarget": [False, "Whether to build the OpenMPTarget backend"],
"threads": [False, "Whether to build the C++ threads backend"],
"rocm": [False, "Whether to build HIP backend"],
"serial": [True, "Whether to build serial backend"],
"sycl": [False, "Whether to build the SYCL backend"],
}
for backend in backends:
deflt_bool, descr = backends[backend]
variant(backend.lower(), default=deflt_bool, description=descr)
depends_on(f"kokkos+{backend.lower()}", when=f"+{backend.lower()}", type=("run", "build"))
# CUDA
for val in CudaPackage.cuda_arch_values:
depends_on("kokkos cuda_arch={0}".format(val), when="cuda_arch={0}".format(val))
# Use +wrapper when not %clang %cce
depends_on("kokkos+wrapper", when="%gcc+cuda")
# ROCm
for val in ROCmPackage.amdgpu_targets:
depends_on("kokkos amdgpu_target={0}".format(val), when="amdgpu_target={0}".format(val))
conflicts(
"+cuda",
when="+rocm",
msg="CUDA and ROCm are not compatible in PennyLane-Lightning-Kokkos.",
)
# build options
extends("python")
variant("cpptests", default=False, description="Build CPP tests")
variant("native", default=False, description="Build natively for given hardware")
variant("sanitize", default=False, description="Build with address sanitization")
# hard dependencies
depends_on("cmake@3.20:", type="build")
depends_on("ninja", type="build")
depends_on("python@3.8:", type=("build", "run"))
depends_on("py-setuptools", type="build")
depends_on("py-pybind11", type="link")
depends_on("py-pip", type="build")
depends_on("py-wheel", type="build")
depends_on("py-pennylane@0.28:", type=("build", "run"), when="@:0.30")
depends_on("py-pennylane@0.30:", type=("build", "run"), when="@0.31:")
# This requirement differs from `pennylane>=0.30` in `setup.py`,
# but the introduction of `StatePrep` demands `pennylane>=0.32`
depends_on("py-pennylane@0.32:", type=("build", "run"), when="@0.32:")
depends_on("py-pennylane-lightning~kokkos", type=("build", "run"), when="@:0.31")
# variant defined dependencies
depends_on("llvm-openmp", when="+openmp %apple-clang")
# Test deps
depends_on("py-pytest", type="test")
depends_on("py-pytest-mock", type="test")
depends_on("py-flaky", type="test")
class CMakeBuilder(spack.build_systems.cmake.CMakeBuilder):
build_directory = "build"
def cmake_args(self):
"""
Here we specify all variant options that can be dynamically specified at build time
"""
args = [
self.define_from_variant("PLKOKKOS_BUILD_TESTS", "cpptests"),
self.define_from_variant("PLKOKKOS_ENABLE_NATIVE", "native"),
self.define_from_variant("PLKOKKOS_ENABLE_SANITIZER", "sanitize"),
]
args.append("-DCMAKE_PREFIX_PATH=" + self.spec["kokkos"].prefix)
if "+rocm" in self.spec:
args.append(self.define("CMAKE_CXX_COMPILER", self.spec["hip"].hipcc))
args.append(
"-DPLKOKKOS_ENABLE_WARNINGS=OFF"
) # otherwise build might fail due to Kokkos::InitArguments deprecated
return args
def build(self, pkg, spec, prefix):
super().build(pkg, spec, prefix)
cm_args = ";".join([s[2:] for s in self.cmake_args()])
args = ["-i", f"--define={cm_args}"]
python("setup.py", "build_ext", *args)
def install(self, pkg, spec, prefix):
pip_args = std_pip_args + [f"--prefix={prefix}", "."]
pip(*pip_args)
super().install(pkg, spec, prefix)
@run_after("install")
@on_package_attributes(run_tests=True)
def install_test(self):
with working_dir(self.stage.source_path):
pl_device_test = Executable(join_path(self.prefix, "bin", "pl-device-test"))
pl_device_test("--device", "lightning.kokkos", "--shots", "None", "--skip-ops")
|
