hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

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

Browse source
This commit is contained in:
Auriane R 2024-05-04 01:40:46 +02:00 committed by GitHub
parent 4adbfa3835
commit 4d7cd4c0bf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
var/spack/repos/builtin/packages/py-flash-attn/package.py
View file

@ -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"))