tiramisu: new package (#35364)

* add initial package. * Update package.py * Update var/spack/repos/builtin/packages/tiramisu/package.py Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> * Update var/spack/repos/builtin/packages/tiramisu/package.py Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> * Update var/spack/repos/builtin/packages/tiramisu/package.py Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> * Update package.py * Hopefully this will be fine. * Update package.py * Update package.py * Update package.py * Update var/spack/repos/builtin/packages/tiramisu/package.py Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> --------- Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2023-02-09 12:09:30 -05:00 · 2023-02-09 12:09:30 -05:00 · 8c0b8c785f
commit 8c0b8c785f
parent 02d3bd782d
1 changed files with 71 additions and 0 deletions
--- a/var/spack/repos/builtin/packages/tiramisu/package.py
+++ b/var/spack/repos/builtin/packages/tiramisu/package.py
@ -0,0 +1,71 @@
+# Copyright 2013-2022 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 Tiramisu(CMakePackage, CudaPackage, PythonExtension):
+    """Tiramisu is a polyhedral compiler for dense and sparse deep learning
+    and data parallel algorithms.It provides a simple C++ API for expressing
+    algorithms and how these algorithms should be optimized by the compiler."""
+
+    homepage = "http://tiramisu-compiler.org"
+    url = "https://github.com/Tiramisu-Compiler/tiramisu"
+    git = "https://github.com/Tiramisu-Compiler/tiramisu.git"
+
+    maintainers("wraith1995")
+
+    generator = "Ninja"
+
+    version("master", branch="master")
+    version("2023-2-8", commit="2cd0c43cc1656bfa43cfb6e81d06f770cbf7251e")
+
+    variant("python", default=True, description="Install python bindings.")
+    extends("python", when="+python")
+    variant(
+        "debug",
+        default="0",
+        description="Set the debug level.",
+        multi=False,
+        values=("0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "10"),
+    )
+
+    depends_on("cmake@3.5:", type="build")
+    depends_on("ninja", type="build")
+
+    depends_on("halide@14.0.0:", type=("build", "link", "run"))
+    depends_on("isl", type=("build", "link", "run"))
+    depends_on("python@3.8:", type=("build", "link", "run"), when="+python")
+    depends_on("py-pybind11@2.6.2:", type="build", when="+python")
+    depends_on("py-numpy", type=("build", "run"), when="+python")
+    depends_on("py-cython", type="run", when="+python")
+
+    def cmake_args(self):
+        spec = self.spec
+        llvmdir = spec["llvm"].prefix.bin
+        args = [
+            self.define("HALIDE_LIB_DIRECTORY", spec["halide"].libs.directories[0]),
+            self.define("LLVM_CONFIG_BIN", llvmdir),
+            self.define("HALIDE_SOURCE_DIRECTORY", spec["halide"].prefix),
+            self.define("ISL_INCLUDE_DIRECTORY", spec["isl"].headers.directories[0]),
+            self.define("ISL_LIB_DIRECTORY", spec["isl"].libs.directories[0]),
+            self.define_from_variant("DEBUG_LEVEL", "debug"),
+            self.define_from_variant("WITH_PYTHON_BINDINGS", "python"),
+            self.define_from_variant("USE_GPU", "cuda"),
+            self.define("PYBIND11_USE_FETCHCONTENT", False),
+            self.define("USE_FLEXNLP", False),
+        ]
+        if "+python" in spec:
+            args += [
+                self.define("Tiramisu_INSTALL_PYTHONDIR", python_platlib),
+                self.define("Python3_EXECUTABLE", spec["python"].command.path),
+            ]
+        return args
+
+    @property
+    def build_targets(self):
+        if "+python" in self.spec:
+            return ["tiramisu", "Tiramisu_Python"]
+        else:
+            return ["tiramisu"]