diff options
| author | Auriane R. <48684432+aurianer@users.noreply.github.com> | 2024-10-02 16:11:50 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-02 16:11:50 +0200 |
| commit | f3faeb0f7708a8f6de2425876630b4af52559201 (patch) | |
| tree | fdcc43d0af4bbc1a2ded05d5a13ad695f50fa713 /var | |
| parent | b1707c2e3c07130d892951a3c92ec57327848dab (diff) | |
| download | spack-f3faeb0f7708a8f6de2425876630b4af52559201.tar.gz spack-f3faeb0f7708a8f6de2425876630b4af52559201.tar.bz2 spack-f3faeb0f7708a8f6de2425876630b4af52559201.tar.xz spack-f3faeb0f7708a8f6de2425876630b4af52559201.zip | |
py-flash-attn: add v2.6.3 (#46487)
* [py-flash-attn] Add version 2.6.3
* Update dependencies according to the latest version
* Add max_jobs environmental variable to avoid oom error
---------
Co-authored-by: aurianer <8trash-can8@protonmail.ch>
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-flash-attn/package.py | 12 |
1 files changed, 8 insertions, 4 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 692f5f71de..8d4c043cf9 100644 --- a/var/spack/repos/builtin/packages/py-flash-attn/package.py +++ b/var/spack/repos/builtin/packages/py-flash-attn/package.py @@ -20,6 +20,7 @@ class PyFlashAttn(PythonPackage): license("BSD") version("main", branch="main") + version("2.6.3", sha256="5bfae9500ad8e7d2937ebccb4906f3bc464d1bf66eedd0e4adabd520811c7b52") version( "2.5.9.post1", sha256="a92db1683a5b141a0f4371d251ae9f73e9aef629b3a58a50d0ef430266c68782" ) @@ -33,18 +34,21 @@ class PyFlashAttn(PythonPackage): depends_on("cxx", type="build") # generated with default_args(type="build"): - depends_on("py-ninja") depends_on("py-packaging") depends_on("py-psutil") depends_on("py-setuptools") + depends_on("ninja") with default_args(type=("build", "run")): depends_on("py-torch+cuda") - depends_on("ninja") depends_on("py-einops") - depends_on("py-packaging") with default_args(type=("build", "link", "run")): depends_on("py-pybind11") - depends_on("python@3.7:", type=("build", "run")) + depends_on("python@3.7:", type=("build", "run"), when="@:2.5") + depends_on("python@3.8:", type=("build", "run"), when="@2.6:") + + def setup_build_environment(self, env): + # If oom error, try lowering the number of jobs with `spack install -j` + env.set("MAX_JOBS", make_jobs) |