From 4cfd3add6d53115e9351c7985d1e4ba8f82ae48e Mon Sep 17 00:00:00 2001
From: Yineng Zhang <me@zhyncs.com>
Date: Mon, 10 Feb 2025 03:49:52 +0800
Subject: [PATCH] support version in sgl-kernel (#3439)

---
 .github/workflows/release-pypi-kernel.yml  | 2 +-
 .github/workflows/release-whl-kernel.yml   | 2 +-
 python/sglang/check_env.py                 | 1 +
 sgl-kernel/developer_guide.md              | 2 +-
 sgl-kernel/src/sgl-kernel/__init__.py      | 2 ++
 sgl-kernel/{ => src/sgl-kernel}/version.py | 0
 6 files changed, 6 insertions(+), 3 deletions(-)
 rename sgl-kernel/{ => src/sgl-kernel}/version.py (100%)

diff --git a/.github/workflows/release-pypi-kernel.yml b/.github/workflows/release-pypi-kernel.yml
index af34c8423..fa8299807 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/version.py
+      - sgl-kernel/src/sgl-kernel/version.py
   workflow_dispatch:
 
 concurrency:
diff --git a/.github/workflows/release-whl-kernel.yml b/.github/workflows/release-whl-kernel.yml
index 70c451778..7508561cf 100644
--- a/.github/workflows/release-whl-kernel.yml
+++ b/.github/workflows/release-whl-kernel.yml
@@ -9,7 +9,7 @@ on:
     branches:
       - main
     paths:
-      - sgl-kernel/version.py
+      - sgl-kernel/src/sgl-kernel/version.py
 
 jobs:
   build-wheels:
diff --git a/python/sglang/check_env.py b/python/sglang/check_env.py
index 19b8a8f9b..524ad0d04 100644
--- a/python/sglang/check_env.py
+++ b/python/sglang/check_env.py
@@ -19,6 +19,7 @@ def is_cuda_v2():
 # List of packages to check versions
 PACKAGE_LIST = [
     "sglang",
+    "sgl_kernel",
     "flashinfer",
     "triton",
     "transformers",
diff --git a/sgl-kernel/developer_guide.md b/sgl-kernel/developer_guide.md
index 2b9859d94..8212babc5 100644
--- a/sgl-kernel/developer_guide.md
+++ b/sgl-kernel/developer_guide.md
@@ -52,4 +52,4 @@ The `sgl-kernel` is rapidly evolving. If you experience a compilation failure, t
 
 ### Release new version
 
-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)
+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/src/sgl-kernel/version.py)
diff --git a/sgl-kernel/src/sgl-kernel/__init__.py b/sgl-kernel/src/sgl-kernel/__init__.py
index 13bae677a..314416a4f 100644
--- a/sgl-kernel/src/sgl-kernel/__init__.py
+++ b/sgl-kernel/src/sgl-kernel/__init__.py
@@ -36,6 +36,8 @@ from sgl_kernel.ops import (
     tree_speculative_sampling_target_only,
 )
 
+from .version import __version__
+
 __all__ = [
     "apply_rope_with_cos_sin_cache_inplace",
     "bmm_fp8",
diff --git a/sgl-kernel/version.py b/sgl-kernel/src/sgl-kernel/version.py
similarity index 100%
rename from sgl-kernel/version.py
rename to sgl-kernel/src/sgl-kernel/version.py