MagiAttention/pyproject.toml at main · SandAI-org/MagiAttention · 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
[project]
name = "magi_attention"
dynamic = ["version"]
description = "A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Training"
authors = [
    {name = "Zewei Tao", email = "zeweitao@sand.ai"},
    {name = "Yunpeng Huang", email = "yunpenghuang@sand.ai"},
]
license = "Apache-2.0"
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
    "Programming Language :: Python :: 3",
    "Operating System :: OS Independent",
]

[project.urls]
"Homepage" = "https://github.com/SandAI-org/MagiAttention/"
"Bug Tracker" = "https://github.com/SandAI-org/MagiAttention/issues"

[tool.isort]
profile = "black"

[build-system]
requires = [
    "setuptools>=61.0",
    "wheel",
    "torch",
    "ninja",
    "packaging"
]
build-backend = "setuptools.build_meta"

[tool.versioningit]
vcs = "git"
default-version = "1.1.0"

[tool.versioningit.write]
file = "magi_attention/_version.py"
encoding = "utf-8"
template = "__version__ = '{version}'"

[tool.setuptools]
include-package-data = false

[tool.setuptools.packages.find]
include = ["magi_attention*"]
exclude = [
    "build*",
    "tests*",
    "dist*",
    "docs*",
    "tools*",
    "assets*",
    "scripts*",
    "extensions*",
    "examples*",
]

[tool.setuptools.package-data]
"magi_attention" = [
    # include pyi files
    "*.pyi",
    "**/*.pyi",
    # common headers and utils for runtime/JIT
    "csrc/common/*.h",
    "csrc/extensions/*.hpp",
    "csrc/extensions/*.cpp",
    "csrc/extensions/*.cu",
    "csrc/extensions/*.cuh",
    "csrc/extensions/*.h",
    "csrc/flexible_flash_attention/**/*.h",
    "csrc/flexible_flash_attention/**/*.hpp",
    "csrc/flexible_flash_attention/**/*.cuh",
    "csrc/flexible_flash_attention/**/*.cu",
    "csrc/flexible_flash_attention/**/*.cpp",
    "csrc/flexible_flash_attention/**/*.jinja",
    "csrc/flexible_flash_attention/**/*.py",
    "csrc/comm/**/*.hpp",
    "csrc/comm/**/*.cpp",
    "csrc/comm/**/*.cuh",
    "csrc/comm/**/*.cu",
    # only ship CUTLASS public headers
    "csrc/cutlass/include/**",
    # include instantiations of flexible_flash_attention
    "lib/*/*.so",
    # include ext modules besides flexible_flash_attention
    "*.so",
]

[tool.setuptools_scm]