hip-attention/pyproject.toml at deepauto/dev · DeepAuto-AI/hip-attention · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
[build-system]
requires = [
    "setuptools>=80.9.0",
    "setuptools-scm",

    # Build dependencies. (Especially for flash-attn)
    "cmake>=3.21",
    "ninja",
    "packaging",
    "wheel>=0.45.1",
]
build-backend = "setuptools.build_meta"

[project]
name = "hip-attn"  # Name of package when installed using pip
version = "1.2.9"
description = "HiP Attention"
authors = [
    { name="DeepAuto.ai", email="contact@deepauto.ai" },
    { name="Heejun Lee", email="gmlwns5176@gmail.com" },
    { name="Geon Park", email="mujjingun@gmail.com" },
    { name="Bumsik Kim", email="k.bumsik@gmail.com" },
]
license = "LicenseRef-FSL-1.1-MIT"
license-files = ["LICENSE.md"]
readme = "README.md"
requires-python = ">=3.10, <3.13"
classifiers = [
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Environment :: GPU",
    "Environment :: GPU :: NVIDIA CUDA :: 12",
]
dependencies = [
    # Required
    "torch",
    "triton",
    "setuptools>=80.9.0",  # Required by triton amd backend
    "numba",
    "performer_pytorch",
    "peft",
    "seaborn",
    "sympy",
    "bitsandbytes",
    "tilelang",
    # CUDA 12
    "cupy-cuda12x",
    "nvtx",
    # Debugging, multimodal
    "opencv-python-headless>=4.11.0.86",
]

[project.urls]
Homepage = "https://github.com/DeepAuto-AI/hip-attention"
Repository = "https://github.com/DeepAuto-AI/hip-attention"
Issues = "https://github.com/DeepAuto-AI/hip-attention/issues"

[project.optional-dependencies]
sglang = [
    "sgl-kernel",
    # Needs adding `--find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer-python`
    # For uv, see tool.uv.sources and tool.uv.index sections
    "flashinfer-python",
    # See tool.uv.sources section
    "flash-attn>=2.8.3",
    # See tool.uv.sources section
    "sglang[all]",
]

[tool.uv.sources]
hip-attn = { workspace = true }
sglang = [
    # pip format: "sglang[all] @ git+https://github.com/DeepAuto-AI/sglang.git@deepauto/dev#subdirectory=python"
    # To update sglang, run `uv lock --upgrade-package sglang`
    { git = "https://github.com/DeepAuto-AI/sglang.git", subdirectory = "python", rev = "deepauto/dev" },
    # For local devs, you can use this instead:
    # { path = "../sglang/python", editable = true },
]
hip-research = { workspace = true }

[tool.uv]
no-build-isolation-package = ["flash-attn"]

[tool.uv.workspace]
members = ["hip-research"]

[dependency-groups]
dev = [
    "hip-attn[sglang]",
    "hip-research",
    "pre-commit>=4.2.0",
]