summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-kornia-rs/package.py43
-rw-r--r--var/spack/repos/builtin/packages/py-kornia/package.py16
2 files changed, 56 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/py-kornia-rs/package.py b/var/spack/repos/builtin/packages/py-kornia-rs/package.py
new file mode 100644
index 0000000000..21f54eb79b
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-kornia-rs/package.py
@@ -0,0 +1,43 @@
+# Copyright 2013-2024 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 PyKorniaRs(PythonPackage):
+ """Low level implementations for computer vision in Rust."""
+
+ homepage = "http://www.kornia.org/"
+ url = "https://github.com/kornia/kornia-rs/archive/refs/tags/v0.1.1.tar.gz"
+
+ license("Apache-2.0")
+ maintainers(
+ "edgarriba",
+ "ducha-aiki",
+ "lferraz",
+ "shijianjian",
+ "cjpurackal",
+ "johnnv1",
+ "adamjstewart",
+ )
+
+ version("0.1.1", sha256="b9ac327fae6e982e6d7df9faeadd1d4f6453e65521819ae9ae5b90e9da0ed1a5")
+ version("0.1.0", sha256="0fca64f901dddff49b72e51fc92a25f0a7606e9a1a72ef283606245ea6b4f90d")
+
+ build_directory = "py-kornia"
+
+ depends_on("py-maturin@1.3.2:", type="build")
+
+ # rav1e needs rustdoc
+ depends_on("rust+dev", type="build")
+
+ # pyo3 needs cmake
+ depends_on("cmake", type="build")
+
+ # turbojpeg-sys needs an assembly compiler
+ depends_on("nasm", type="build")
+
+ # dlpack-rs needs libclang
+ depends_on("llvm+clang")
diff --git a/var/spack/repos/builtin/packages/py-kornia/package.py b/var/spack/repos/builtin/packages/py-kornia/package.py
index bb9c788f57..3fc1c0de7e 100644
--- a/var/spack/repos/builtin/packages/py-kornia/package.py
+++ b/var/spack/repos/builtin/packages/py-kornia/package.py
@@ -9,11 +9,21 @@ from spack.package import *
class PyKornia(PythonPackage):
"""Open Source Differentiable Computer Vision Library for PyTorch."""
- homepage = "https://www.kornia.org/"
+ homepage = "http://www.kornia.org/"
pypi = "kornia/kornia-0.5.10.tar.gz"
license("Apache-2.0")
-
+ maintainers(
+ "edgarriba",
+ "ducha-aiki",
+ "lferraz",
+ "shijianjian",
+ "cjpurackal",
+ "johnnv1",
+ "adamjstewart",
+ )
+
+ version("0.7.2", sha256="f834ccd51188d071ed286a6727471c94344ea2a718903cc6f0e56a92f9c66ac5")
version("0.7.1", sha256="65b54a50f70c1f88240b557fda3fdcc1ab866982a5d062e52213130f5a48465c")
version("0.7.0", sha256="72cba6a0965a15caf10a664647654412effb7c0b9afcf40e458bc005f976ffac")
version("0.6.12", sha256="e30bd3d830226f7a159dff1f7757c6200e8f27d1333f06e9d2f98bdb33ce18d3")
@@ -31,11 +41,11 @@ class PyKornia(PythonPackage):
version("0.5.10", sha256="428b4b934a2ba7360cc6cba051ed8fd96c2d0f66611fdca0834e82845f14f65d")
# pyproject.toml
- depends_on("python@3.8:", when="@0.7:", type=("build", "run"))
depends_on("py-setuptools@61.2:", when="@0.6.11:", type="build")
depends_on("py-setuptools", type="build")
# requirements/requirements.txt
+ depends_on("py-kornia-rs@0.1:", when="@0.7.2:", type=("build", "run"))
depends_on("py-packaging", when="@0.6:", type=("build", "run"))
depends_on("py-torch@1.9.1:", when="@0.6.9:", type=("build", "run"))
depends_on("py-torch@1.8.1:", when="@0.6:", type=("build", "run"))