blob: 44c6f1ad57391ec181fa737bd189f8045ce20862 (
plain) (
blame)
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
|
# 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 Whip(CMakePackage, CudaPackage, ROCmPackage):
"""whip is a small C++ abstraction layer for CUDA and HIP."""
homepage = "https://github.com/eth-cscs/whip/"
url = "https://github.com/eth-cscs/whip/archive/0.0.0.tar.gz"
git = "https://github.com/eth-cscs/whip.git"
maintainers("msimberg", "rasolca")
license("BSD-3-Clause")
version("main", branch="main")
version("0.2.0", sha256="d8fec662526accbd1624922fdf01a077d6f312cf253382660e4a2f65e28e8686")
version("0.1.0", sha256="5d557794f4afc8332fc660948a342f69e22bc9e5d575ffb3e3944cf526db5ec9")
depends_on("cmake@3.22:", type="build")
# Exactly one of +cuda and +rocm need to be set
conflicts("~cuda ~rocm")
conflicts("+cuda +rocm")
def cmake_args(self):
if self.spec.satisfies("+cuda"):
return [self.define("WHIP_BACKEND", "CUDA")]
else:
return [self.define("WHIP_BACKEND", "HIP")]
|