diff options
| author | Auriane R <48684432+aurianer@users.noreply.github.com> | 2024-05-04 01:40:46 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-03 16:40:46 -0700 |
| commit | 4d7cd4c0bfdfeaec9a790ac2ce6d35a186483aa6 (patch) | |
| tree | 6bccd2335ff9d19094bc9c65c44f54c975d41b46 | |
| parent | 4adbfa38354ce80925114582290ce587e99f0022 (diff) | |
| download | spack-4d7cd4c0bfdfeaec9a790ac2ce6d35a186483aa6.tar.gz spack-4d7cd4c0bfdfeaec9a790ac2ce6d35a186483aa6.tar.bz2 spack-4d7cd4c0bfdfeaec9a790ac2ce6d35a186483aa6.tar.xz spack-4d7cd4c0bfdfeaec9a790ac2ce6d35a186483aa6.zip | |
Add py-flash-attn@2.4.2 (#43960)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-flash-attn/package.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-flash-attn/package.py b/var/spack/repos/builtin/packages/py-flash-attn/package.py index 308db11453..7aec6c6d02 100644 --- a/var/spack/repos/builtin/packages/py-flash-attn/package.py +++ b/var/spack/repos/builtin/packages/py-flash-attn/package.py @@ -19,6 +19,7 @@ class PyFlashAttn(PythonPackage): version("2.5.5", sha256="751cee17711d006fe7341cdd78584af86a6239afcfe43b9ed11c84db93126267") version("2.5.4", sha256="d83bb427b517b07e9db655f6e5166eb2607dccf4d6ca3229e3a3528c206b0175") + version("2.4.2", sha256="eb822a8c4219b610e9d734cbc8cd9ee4547f27433815a2b90dc1462766feefc1") depends_on("py-setuptools", type="build") @@ -28,6 +29,9 @@ class PyFlashAttn(PythonPackage): depends_on("py-einops") depends_on("py-packaging") + with default_args(type=("build", "link", "run")): + depends_on("py-pybind11") + depends_on("py-psutil", type="build") depends_on("python@3.7:", type=("build", "run")) |