diff --git a/.github/workflows/release-pypi-kernel.yml b/.github/workflows/release-pypi-kernel.yml index 1b925b772..c07069c5d 100644 --- a/.github/workflows/release-pypi-kernel.yml +++ b/.github/workflows/release-pypi-kernel.yml @@ -5,7 +5,7 @@ on: branches: - main paths: - - sgl-kernel/pyproject.toml + - sgl-kernel/version.py workflow_dispatch: concurrency: diff --git a/sgl-kernel/developer_guide.md b/sgl-kernel/developer_guide.md index 8afb6b0e4..f41ce071e 100644 --- a/sgl-kernel/developer_guide.md +++ b/sgl-kernel/developer_guide.md @@ -26,8 +26,8 @@ Steps to add a new kernel: 1. Implement in [src/sgl-kernel/csrc/](https://github.com/sgl-project/sglang/tree/main/sgl-kernel/src/sgl-kernel/csrc) 2. Expose interface in [csrc/sgl_kernel_ops.cu](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/src/sgl-kernel/csrc/sgl_kernel_ops.cu) with pybind11 -3. Create Python wrapper in [src/sgl-kernel/ops/__init__.py](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/src/sgl-kernel/ops/__init__.py) -4. Expose Python interface in [src/sgl-kernel/__init__.py](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/src/sgl-kernel/__init__.py) +3. Create Python wrapper in [src/sgl-kernel/ops/\_\_init\_\_.py](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/src/sgl-kernel/ops/__init__.py) +4. Expose Python interface in [src/sgl-kernel/\_\_init\_\_.py](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/src/sgl-kernel/__init__.py) 5. Update [setup.py](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/setup.py) to include new CUDA source ### Build & Install @@ -48,4 +48,4 @@ pip3 install dist/*whl --force-reinstall --no-deps ### Release new version -Update version in [pyproject.toml](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/pyproject.toml) +Update version in [pyproject.toml](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/pyproject.toml) and [version.py](https://github.com/sgl-project/sglang/blob/main/sgl-kernel/version.py) diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py index c51fd7045..71952655c 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -3,17 +3,11 @@ from pathlib import Path import torch from setuptools import find_packages, setup from torch.utils.cpp_extension import BuildExtension, CUDAExtension +from version import __version__ root = Path(__file__).parent.resolve() -def get_version(): - with open(root / "pyproject.toml") as f: - for line in f: - if line.startswith("version"): - return line.split("=")[1].strip().strip('"') - - def update_wheel_platform_tag(): wheel_dir = Path("dist") if wheel_dir.exists() and wheel_dir.is_dir(): @@ -122,7 +116,7 @@ ext_modules = [ setup( name="sgl-kernel", - version=get_version(), + version=__version__, packages=find_packages(), package_dir={"": "src"}, ext_modules=ext_modules, diff --git a/sgl-kernel/version.py b/sgl-kernel/version.py new file mode 100644 index 000000000..4bb48c132 --- /dev/null +++ b/sgl-kernel/version.py @@ -0,0 +1 @@ +__version__ = "0.0.2.post15"