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
|
# 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 Enzyme(CMakePackage):
"""
The Enzyme project is a tool for performing reverse-mode automatic
differentiation (AD) of statically-analyzable LLVM IR.
This allows developers to use Enzyme to automatically create gradients
of their source code without much additional work.
"""
homepage = "https://enzyme.mit.edu"
url = "https://github.com/wsmoses/Enzyme/archive/v0.0.15.tar.gz"
list_url = "https://github.com/wsmoses/Enzyme/releases"
git = "https://github.com/wsmoses/Enzyme"
maintainers("wsmoses", "vchuravy", "tgymnich")
root_cmakelists_dir = "enzyme"
version("main", branch="main")
version("0.0.32", sha256="9d42e42f7d0faf9beed61b2b1d27c82d1b369aeb9629539d5b7eafbe95379292")
version("0.0.15", sha256="1ec27db0d790c4507b2256d851b256bf7e074eec933040e9e375d6e352a3c159")
version("0.0.14", sha256="740641eeeeadaf47942ac88cc52e62ddc0e8c25767a501bed36ec241cf258b8d")
version("0.0.13", sha256="d4a53964ec1f763772db2c56e6734269b7656c8b2ecd41fa7a41315bcd896b5a")
variant(
"build_type",
default="Release",
description="CMake build type",
values=("Debug", "Release", "RelWithDebInfo", "MinSizeRel"),
)
depends_on("llvm@7:12", when="@0.0.13:0.0.15")
depends_on("llvm@7:14", when="@0.0.32:") # TODO actual lower bound
depends_on("cmake@3.9:", type="build")
def cmake_args(self):
spec = self.spec
args = ["-DLLVM_DIR=" + spec["llvm"].prefix.lib + "/cmake/llvm"]
return args
@property
def libs(self):
ver = self.spec["llvm"].version.up_to(1)
libs = ["LLVMEnzyme-{0}".format(ver), "ClangEnzyme-{0}".format(ver)]
if self.version >= Version("0.0.32"): # TODO actual lower bound
libs.append("LLDEnzyme-{0}".format(ver))
return find_libraries(libs, root=self.prefix, recursive=True)
def setup_dependent_build_environment(self, env, dependent_spec):
# Get the LLVMEnzyme and ClangEnzyme lib paths
llvm, clang = self.libs
if "LLVMEnzyme-" in clang:
llvm, clang = clang, llvm
env.set("LLVMENZYME", llvm)
env.set("CLANGENZYME", clang)
|