98162aa2e1
* py-earth2mip: add new package * Fix audit tests * Make things easier to concretize * Make things easier to concretize * Fix magics build * Fix onnx build * blacken * cmake > py-cmake * Fix onnxruntime build * onnxruntime: remove unknown cmake vars * Trick bazelisk into using Spack bazel * Different deps for main * protobuf: add v3.25.3 * Enforce upper bound on jaxlib version * py-chex: add v0.1.85 * py-dm-haiku: add v0.0.12 * py-jax: add v0.4.25 * Older jaxlib doesn't build on ppc64le either
25 lines
802 B
Python
25 lines
802 B
Python
# 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 PyTorchHarmonics(PythonPackage):
|
|
"""A differentiable spherical harmonic transform for PyTorch."""
|
|
|
|
homepage = "https://github.com/NVIDIA/torch-harmonics"
|
|
pypi = "torch_harmonics/torch_harmonics-0.6.5.tar.gz"
|
|
|
|
maintainers("adamjstewart")
|
|
|
|
license("BSD")
|
|
|
|
version("0.6.5", sha256="e467d04bc58eb2dc800eb21870025407d38ebcbf8df4de479bd5b4915daf987e")
|
|
|
|
depends_on("py-setuptools", type="build")
|
|
depends_on("py-torch", type=("build", "run"))
|
|
depends_on("py-numpy", type=("build", "run"))
|
|
depends_on("py-triton", type=("build", "run"))
|