diff options
| author | Christopher Christofi <77968333+ChristopherChristofi@users.noreply.github.com> | 2024-08-06 08:17:41 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-08-06 09:17:41 +0200 |
| commit | 8de03e2bf59afa1c2457c4a8c0531181db330f77 (patch) | |
| tree | 77723ea6fe8c962c119e10d97807034966e514ff | |
| parent | 2fa314b6b63be8cd264c7787852886e9b6c827aa (diff) | |
| download | spack-8de03e2bf59afa1c2457c4a8c0531181db330f77.tar.gz spack-8de03e2bf59afa1c2457c4a8c0531181db330f77.tar.bz2 spack-8de03e2bf59afa1c2457c4a8c0531181db330f77.tar.xz spack-8de03e2bf59afa1c2457c4a8c0531181db330f77.zip | |
py-numba4jax: add new package (#45481)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-numba4jax/package.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-numba4jax/package.py b/var/spack/repos/builtin/packages/py-numba4jax/package.py new file mode 100644 index 0000000000..ab24f93237 --- /dev/null +++ b/var/spack/repos/builtin/packages/py-numba4jax/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 PyNumba4jax(PythonPackage): + """Use numba-compiled kernels from within Jax""" + + homepage = "https://github.com/PhilipVinc/numba4jax" + pypi = "numba4jax/numba4jax-0.0.12.tar.gz" + + license("MIT") + + version("0.0.12", sha256="e1faf6a0566f4fb941abf8821b9c854b7398eb08a0c8157927f8b4717a393446") + + with default_args(type="build"): + depends_on("py-hatchling@1.8.0:") + depends_on("py-hatch-vcs") + + with default_args(type=("build", "run")): + depends_on("python@3.9:") + depends_on("py-numpy@1.22:1.23") + depends_on("py-numba@0.53:0.61") + depends_on("py-cffi@1.14.4:") + depends_on("py-jax@0.4.16:0.5") + depends_on("py-jaxlib@0.4.16:0.5") |