feat: use warp reduce as a simple example (#2304)

This commit is contained in:
Yineng Zhang
2024-12-01 22:43:50 +08:00
committed by GitHub
parent 5c18a03733
commit 47eb139f81
7 changed files with 193 additions and 17 deletions

View File

@@ -1,37 +1,34 @@
[build-system]
requires = ["setuptools>=61.0", "wheel"]
requires = ["setuptools>=61.0", "wheel", "torch"]
build-backend = "setuptools.build_meta"
[project]
name = "sgl-kernel"
version = "0.0.1"
version = "0.0.2"
description = "Kernel Library for SGLang"
readme = "README.md"
requires-python = ">=3.8"
license = { file = "LICENSE" }
classifiers = [
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Programming Language :: C++",
"Programming Language :: CUDA",
]
dependencies = [
"torch",
]
dependencies = ["numpy"]
[project.optional-dependencies]
srt = ["torch"]
all = ["sgl-kernel[srt]"]
[project.urls]
"Homepage" = "https://github.com/sgl-project/sglang"
"Bug Tracker" = "https://github.com/sgl-project/sglang/issues"
[tool.setuptools.packages.find]
exclude = [
"dist*",
"tests*",
]
[tool.setuptools]
package-dir = {"sgl_kernel" = "src/sgl-kernel"}
packages = ["sgl_kernel", "sgl_kernel.ops", "sgl_kernel.csrc"]
[tool.wheel]
exclude = [
"dist*",
"tests*",
"dist*",
"tests*",
]