TileOPs/pyproject.toml at main · MooreThreads/TileOPs · 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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
[project]
name = "tileops"
version = "0.0.1.dev2"
description = "TileOPs kernels for efficient LLM inference on MUSA."
readme = "README.md"
authors = [{ name = "Moore Threads Technology Co., Ltd" }]
dependencies = [
    "torch>=2.1.0,<2.11.0",
    # TileOPs requires a MUSA-enabled TileLang runtime. Do not install
    # upstream PyPI tilelang here; use the environment-provided MUSA build.
    # FIXME(apache-tvm-ffi-pin): Remove once tilelang carries its own
    # apache-tvm-ffi<0.1.10 pin (already on tilelang main, not yet released).
    # Tracking source issue: https://github.com/tile-ai/TileOPs/issues/969
    "apache-tvm-ffi>=0.1.2,<0.1.10",
    "einops",
    "pyyaml>=6.0",
]
requires-python = ">=3.10"
license = {text = "MIT"}
keywords = ["llm", "inference", "gpu", "kernel", "tiling"]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

[project.urls]
Homepage = "https://github.com/MooreThreads/TileOPs"
Repository = "https://github.com/MooreThreads/TileOPs.git"
Issues = "https://github.com/MooreThreads/TileOPs/issues"

[project.optional-dependencies]
dev = [
    "ruff==0.14.13",
    "codespell==2.4.1",
    "pytest>=8.0",
    "pytest-xdist>=3.0",
    "pyyaml>=6.0",
]
bench = [
    "flash-attn==2.8.3",
    "flash-attn-interface>=2.7.2",
    "flash-linear-attention==0.4.2",
    "flashinfer>=0.6.6",
    "vllm==0.18.0",
    "sgl-kernel==0.3.21",
]

[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
where = ["."]
exclude = ["tests", "tests.*", "benchmarks", "benchmarks.*"]

[tool.setuptools.package-data]
tileops = ["kernels/moe/_atomic_helper.h", "manifest/*.yaml"]

[tool.codespell]
ignore-words-list = "nd, te, ist, LOD, offen, NotIn, HSA"
skip = [
    "build",
    "dist",
    ".venv"
]

[tool.ruff]
line-length = 100

# NOTE: Please gradually tighten static code checks and other validations in the future.
# The current exemptions are in place to ensure normal project development.
[tool.ruff.lint]
select = [
    # pycodestyle
    "E",
    # Pyflakes
    "F",
    # pyupgrade
    # "UP",
    # flake8-bugbear
    "B",
    # flake8-simplify
    "SIM",
    # isort
    "I",
    # pydocstyle (Google Docstrings)
    # "D",
    # pep8-naming (naming conventions)
    "N",
    # ruff-specific: sort __all__
    "RUF022",
    # flake8-annotations (type hints) — enable when codebase has sufficient coverage
    # "ANN",
]
ignore = [
    # ambiguous name
    "E741",
    # line too long (allow some flexibility)
    "E501",
    # key in dict.keys()
    "SIM118",
    # memory leaks
    "B019",
    # No such file or directory
    "E902",
    # pep8-naming: allow 'import as T'
    "N812",
    # pep8-naming: ignore naming convention violations (common in ML/kernel code)
    "N801",  # Class name should use CapWords convention
    "N802",  # Function name should be lowercase
    "N803",  # Argument name should be lowercase
    "N806",  # Variable in function should be lowercase
]

[tool.ruff.lint.isort]
known-first-party = ["tileops"]

[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["F401"]
"bench_*.py" = ["F401"]
# __all__ intentionally grouped by category, not alphabetical
"tileops/kernels/elementwise.py" = ["RUF022"]
"tileops/ops/__init__.py" = ["RUF022"]
"tileops/ops/elementwise.py" = ["RUF022"]
"tileops/ops/reduction/__init__.py" = ["RUF022"]

[tool.pytest.ini_options]
testpaths = ["tests", "benchmarks/tests"]
python_files = ["test_*.py", "bench_*.py"]
markers = [
    "smoke: fast PR critical-path tests",
    "full: standard correctness coverage",
    "nightly: exhaustive or long-running coverage",
    "packaging: minimal wheel-install sanity check (one case per op family)",
]