mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-03-14 18:37:23 +00:00
74 lines
1.8 KiB
TOML
74 lines
1.8 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 = "0.1.0"
|
|
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",
|
|
# Development dependencies
|
|
"black>=25.9.0",
|
|
]
|
|
|
|
# No optional dev group needed for formatting; using custom git hooks instead of pre-commit
|
|
|
|
[project.urls]
|
|
Homepage = "https://github.com/kvcache-ai"
|
|
|
|
[tool.setuptools]
|
|
packages = ["kt_kernel", "kt_kernel.utils"]
|
|
include-package-data = true
|
|
|
|
[tool.setuptools.package-dir]
|
|
kt_kernel = "python"
|
|
"kt_kernel.utils" = "python/utils"
|
|
|
|
[tool.setuptools.package-data]
|
|
# (empty) placeholder if you later add resources
|
|
|
|
[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
|
|
)/
|
|
)
|
|
'''
|