From 4d7cd4c0bfdfeaec9a790ac2ce6d35a186483aa6 Mon Sep 17 00:00:00 2001 From: Auriane R <48684432+aurianer@users.noreply.github.com> Date: Sat, 4 May 2024 01:40:46 +0200 Subject: [PATCH] Add py-flash-attn@2.4.2 (#43960) --- var/spack/repos/builtin/packages/py-flash-attn/package.py | 4 ++++ 1 file changed, 4 insertions(+) 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"))