From a50c45f00cc32d7da2396cc02cec8158dd64e708 Mon Sep 17 00:00:00 2001 From: Thomas Bouvier Date: Tue, 28 Jan 2025 10:43:27 +0000 Subject: [PATCH] py-flash-attn: add missing triton dependency (#48645) --- var/spack/repos/builtin/packages/py-flash-attn/package.py | 3 ++- var/spack/repos/builtin/packages/py-triton/package.py | 6 +++--- 2 files changed, 5 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 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")