LightKernel/pyproject.toml at main · ModelTC/LightKernel · 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
[build-system]
requires = [
  "scikit-build-core>=0.10",
  "cmake>=3.22",
  "ninja",
  "wheel",
]
build-backend = "scikit_build_core.build"

[project]
name = "lightllm-kernel"
version = "0.1.0"
description = "CUDA kernel library for LightLLM"
readme = "README.md"
requires-python = ">=3.9"
license = { text = "Apache-2.0" }
keywords = ["cuda", "lightllm"]
classifiers = [
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3 :: Only",
  "License :: OSI Approved :: Apache Software License",
  "Environment :: GPU :: NVIDIA CUDA",
  "Operating System :: POSIX :: Linux",
]
dependencies = [
  "pybind11>=2.11",
]

[project.optional-dependencies]
dev = [
  "black",
  "ruff",
  "pre-commit",
]
test = [
  "pytest",
  "pytest-cov",
]

[project.urls]
Homepage = "https://github.com/ModelTC/lightllm/tree/main/lightllm-kernel"
Source = "https://github.com/ModelTC/lightllm/tree/main/lightllm-kernel"
Issues = "https://github.com/ModelTC/lightllm/issues"

[tool.wheel]

exclude = ["dist*", "tests*"]

[tool.scikit-build]
cmake.minimum-version = "3.22"
cmake.build-type = "Release"
cmake.verbose = true

cmake.args = ["-DCMAKE_CUDA_ARCHITECTURES=80;86;89;90;100;100a;120;120a"]

wheel.packages = ["lightllm_kernel"]
wheel.license-files = ["LICENSE"]

sdist.include = [
  "CMakeLists.txt",
  "csrc/**/*",
  "third-party/cutlass/**/*",
  "lightllm_kernel/**/*",
  "LICENSE",
  "README.md",
]

install.components = ["Python"]