|
1 | 1 | [build-system] |
2 | | -requires = ["setuptools>=64.0", "scikit-build-core>=0.11", "pybind11"] |
| 2 | +# cmake locks to '>=3.20,<4.0' to sync with FlagTree |
| 3 | +requires = ["setuptools>=64.0", "scikit-build-core>=0.11", "pybind11>=3.0.2", "cmake>=3.2.0,<4.0", "ninja>=1.13.0"] |
| 4 | + |
3 | 5 | build-backend = "scikit_build_core.build" |
4 | 6 |
|
5 | 7 |
|
6 | 8 | [project] |
| 9 | + |
7 | 10 | name = "flag_gems" |
| 11 | + |
8 | 12 | version = "5.0.1.rc.0" |
9 | | -description = "FlagGems is a function library written in Triton." |
| 13 | + |
| 14 | +description = "FlagGems is a function library written in Triton language." |
| 15 | + |
10 | 16 | readme = { file = "README.md", content-type = "text/markdown" } |
| 17 | + |
11 | 18 | requires-python = ">=3.8.0" |
| 19 | + |
12 | 20 | license = { text = "Apache Software License" } |
| 21 | + |
13 | 22 | authors = [ |
14 | 23 | { name = "Zhixin Li", email = "strongspoon@outlook.com" }, |
15 | 24 | { name = "Tongxin Bai", email = "waffle.bai@gmail.com" }, |
16 | 25 | { name = "Yuming Huang", email = "jokmingwong@gmail.com" }, |
17 | 26 | { name = "Feiyu Chen", email = "iclementine@outlook.com" }, |
18 | 27 | ] |
| 28 | + |
19 | 29 | classifiers = [ |
20 | 30 | "Development Status :: 3 - Alpha", |
21 | 31 | "Programming Language :: Python :: 3", |
22 | 32 | "License :: OSI Approved :: Apache Software License", |
23 | 33 | ] |
| 34 | + |
24 | 35 | dependencies = [ |
| 36 | + # TODO: This is a very old version, should be bumped |
25 | 37 | "torch>=2.2.0", |
26 | | - "packaging", |
27 | | - "pybind11", |
28 | | - "PyYAML", |
29 | | - "packaging", |
30 | | - "sqlalchemy", |
| 38 | + "ninja>=1.13.0", |
| 39 | + "packaging>=26.0", |
| 40 | + "pybind11>=2.13.1", |
| 41 | + "PyYAML>=6.0.3", |
| 42 | + "sqlalchemy>=2.0.48", |
31 | 43 | ] |
32 | 44 |
|
33 | 45 | [project.optional-dependencies] |
| 46 | + |
34 | 47 | test = [ |
35 | | - "pytest>=7.1.0", |
| 48 | + "pytest>=9.0.2", |
36 | 49 | "numpy>=1.26", |
37 | | - "scipy>=1.14", |
| 50 | + "scipy>=1.17.0", |
38 | 51 | ] |
| 52 | + |
39 | 53 | example = [ |
40 | 54 | "transformers>=4.40.2", |
41 | 55 | ] |
42 | 56 |
|
43 | 57 | [project.urls] |
44 | | -Homepage = "https://github.com/FlagOpen/FlagGems" |
45 | 58 |
|
46 | | -[tool.setuptools] |
47 | | -package-dir = { "" = "src" } |
| 59 | +Repository = "https://github.com/flagos-ai/FlagGems" |
| 60 | +Documentation = "https://flagos-ai.github.io/FlagGems" |
| 61 | +"Bug Tracker" = "https://github.com/flagos-ai/FlagGems/issues" |
48 | 62 |
|
49 | 63 | [tool.setuptools.packages.find] |
| 64 | + |
50 | 65 | where = ["src"] |
51 | 66 |
|
52 | 67 | [tool.setuptools.package-data] |
| 68 | + |
53 | 69 | "flag_gems.runtime" = ["*/**/*.yaml"] |
54 | 70 |
|
55 | 71 | [tool.scikit-build] |
| 72 | + |
56 | 73 | build.verbose = true |
57 | 74 | logging.level = "INFO" |
58 | 75 | wheel.install-dir = "flag_gems" |
|
0 commit comments