diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py b/var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py new file mode 100644 index 0000000000..8197e8291e --- /dev/null +++ b/var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py @@ -0,0 +1,29 @@ +# 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 PyCurvlinopsForPytorch(PythonPackage): + """scipy Linear operators for curvature matrices in PyTorch.""" + + homepage = "https://github.com/f-dangel/curvlinops" + pypi = "curvlinops_for_pytorch/curvlinops_for_pytorch-2.0.0.tar.gz" + + license("MIT") + + version("2.0.0", sha256="01f9925db9454fc9b0a31c7b83fc8ec2534c2eb12b7de7825a5298fc14e460e7") + + with default_args(type="build"): + depends_on("py-setuptools@61:") + depends_on("py-setuptools-scm") + + with default_args(type=("build", "run")): + depends_on("py-backpack-for-pytorch@1.6:1") + depends_on("py-torch@2:") + depends_on("py-scipy@1.7.1:1") + depends_on("py-tqdm@4.61:4") + depends_on("py-einops") + depends_on("py-einconv") |