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 298a49a7205..61fe469b629 100644 --- a/var/spack/repos/builtin/packages/py-flash-attn/package.py +++ b/var/spack/repos/builtin/packages/py-flash-attn/package.py @@ -14,7 +14,7 @@ class PyFlashAttn(PythonPackage): pypi = "flash-attn/flash_attn-0.0.0.tar.gz" git = "https://github.com/Dao-AILab/flash-attention.git" - maintainers("aurianer") + maintainers("aurianer", "thomas-bouvier") license("BSD") @@ -41,6 +41,7 @@ class PyFlashAttn(PythonPackage): with default_args(type=("build", "run")): depends_on("py-torch+cuda") depends_on("py-einops") + depends_on("py-triton") with default_args(type=("build", "link", "run")): depends_on("py-pybind11") diff --git a/var/spack/repos/builtin/packages/py-triton/package.py b/var/spack/repos/builtin/packages/py-triton/package.py index 43bee5800a1..78068e590fc 100644 --- a/var/spack/repos/builtin/packages/py-triton/package.py +++ b/var/spack/repos/builtin/packages/py-triton/package.py @@ -8,9 +8,9 @@ class PyTriton(PythonPackage): """A language and compiler for custom Deep Learning operations.""" - homepage = "https://github.com/openai/triton" - url = "https://github.com/openai/triton/archive/refs/tags/v2.1.0.tar.gz" - git = "https://github.com/openai/triton.git" + homepage = "https://github.com/triton-lang/triton" + url = "https://github.com/triton-lang/triton/archive/refs/tags/v2.1.0.tar.gz" + git = "https://github.com/triton-lang/triton.git" license("MIT")