zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-flash-attn: add missing triton dependency (#48645)

Browse Source
This commit is contained in:
Thomas Bouvier 2025-01-28 10:43:27 +00:00 committed by GitHub
parent 87e65e5377
commit a50c45f00c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
var/spack/repos/builtin/packages/py-flash-attn/package.py
View File

@ -14,7 +14,7 @@ class PyFlashAttn(PythonPackage):
pypi = "flash-attn/flash_attn-0.0.0.tar.gz" pypi = "flash-attn/flash_attn-0.0.0.tar.gz"
git = "https://github.com/Dao-AILab/flash-attention.git" git = "https://github.com/Dao-AILab/flash-attention.git"
maintainers("aurianer") maintainers("aurianer", "thomas-bouvier")
license("BSD") license("BSD")
@ -41,6 +41,7 @@ class PyFlashAttn(PythonPackage):
with default_args(type=("build", "run")): with default_args(type=("build", "run")):
depends_on("py-torch+cuda") depends_on("py-torch+cuda")
depends_on("py-einops") depends_on("py-einops")
depends_on("py-triton")
with default_args(type=("build", "link", "run")): with default_args(type=("build", "link", "run")):
depends_on("py-pybind11") depends_on("py-pybind11")

6
var/spack/repos/builtin/packages/py-triton/package.py
View File

@ -8,9 +8,9 @@
class PyTriton(PythonPackage): class PyTriton(PythonPackage):
"""A language and compiler for custom Deep Learning operations.""" """A language and compiler for custom Deep Learning operations."""
homepage = "https://github.com/openai/triton" homepage = "https://github.com/triton-lang/triton"
url = "https://github.com/openai/triton/archive/refs/tags/v2.1.0.tar.gz" url = "https://github.com/triton-lang/triton/archive/refs/tags/v2.1.0.tar.gz"
git = "https://github.com/openai/triton.git" git = "https://github.com/triton-lang/triton.git"
license("MIT") license("MIT")