mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-03-14 18:37:23 +00:00
101 lines
2.3 KiB
TOML
101 lines
2.3 KiB
TOML
[build-system]
|
|
# Minimum versions: setuptools for setup.py declarative usage, wheel for bdist_wheel
|
|
requires = ["setuptools>=61", "wheel", "cmake>=3.16", "pybind11"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "kt-kernel"
|
|
# Version is dynamically read from ../version.py via setup.py
|
|
dynamic = ["version"]
|
|
description = "KT-Kernel: High-performance kernel operations for KTransformers (AMX/AVX/KML optimizations)"
|
|
readme = "README.md"
|
|
authors = [{ name = "kvcache-ai" }]
|
|
# Use SPDX string form (table form deprecated in newer setuptools)
|
|
license = "Apache-2.0"
|
|
classifiers = [
|
|
"Programming Language :: Python :: 3",
|
|
"Programming Language :: C++",
|
|
"Operating System :: POSIX :: Linux",
|
|
"Operating System :: MacOS",
|
|
]
|
|
requires-python = ">=3.8"
|
|
dependencies = [
|
|
# Core dependencies
|
|
"torch>=2.0.0",
|
|
"safetensors>=0.4.0",
|
|
"compressed-tensors>=0.7.0",
|
|
"numpy>=1.24.0",
|
|
"triton>=2.0.0",
|
|
"gguf>=0.17.0",
|
|
# CLI dependencies
|
|
"typer[all]>=0.9.0",
|
|
"rich>=13.0.0",
|
|
"pyyaml>=6.0",
|
|
"httpx>=0.25.0",
|
|
"packaging>=23.0",
|
|
# Development dependencies
|
|
"black>=25.9.0",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
test = [
|
|
"pytest>=7.0.0",
|
|
"psutil>=5.9.0",
|
|
]
|
|
|
|
[project.scripts]
|
|
kt = "kt_kernel.cli.main:main"
|
|
|
|
[project.urls]
|
|
Homepage = "https://github.com/kvcache-ai"
|
|
|
|
[tool.setuptools]
|
|
packages = [
|
|
"kt_kernel",
|
|
"kt_kernel.utils",
|
|
"kt_kernel.cli",
|
|
"kt_kernel.cli.commands",
|
|
"kt_kernel.cli.config",
|
|
"kt_kernel.cli.utils",
|
|
"kt_kernel.cli.completions",
|
|
]
|
|
include-package-data = true
|
|
|
|
[tool.setuptools.package-dir]
|
|
kt_kernel = "python"
|
|
"kt_kernel.utils" = "python/utils"
|
|
"kt_kernel.cli" = "python/cli"
|
|
"kt_kernel.cli.commands" = "python/cli/commands"
|
|
"kt_kernel.cli.config" = "python/cli/config"
|
|
"kt_kernel.cli.utils" = "python/cli/utils"
|
|
"kt_kernel.cli.completions" = "python/cli/completions"
|
|
|
|
[tool.setuptools.package-data]
|
|
"kt_kernel.cli.completions" = ["*.bash", "*.fish", "_kt"]
|
|
|
|
[tool.setuptools.exclude-package-data]
|
|
# (empty)
|
|
|
|
[tool.cpuinfer]
|
|
# Custom section (example). You can place build options documentation here.
|
|
# CPUINFER_CPU_INSTRUCT: NATIVE|FANCY|AVX512|AVX2
|
|
# CPUINFER_ENABLE_AMX: ON/OFF
|
|
# CPUINFER_VERBOSE: 1/0
|
|
|
|
[tool.black]
|
|
# Code style for Black formatter
|
|
line-length = 120
|
|
target-version = ["py311"]
|
|
exclude = '''
|
|
(
|
|
/(\.
|
|
| build
|
|
| dist
|
|
| temp
|
|
| __pycache__
|
|
| kt_kernel\.egg-info
|
|
| third_party
|
|
)/
|
|
)
|
|
'''
|