Add py-flash-attn@2.4.2 (#43960)

2024-05-04 01:40:46 +02:00
parent 4adbfa3835
commit 4d7cd4c0bf
1 changed files with 4 additions and 0 deletions
--- 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"))