From 5eae67cb1ffc63734c9e1ca3b714a7ee4fc1127c Mon Sep 17 00:00:00 2001
From: DavidChan <chengwei0519@163.com>
Date: Fri, 28 Mar 2025 13:26:42 +0800
Subject: [PATCH] get the python version from env (#4729)

---
 sgl-kernel/setup.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py
index 7b21ba593..807f9404c 100644
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -235,6 +235,12 @@ ext_modules = [
     ),
 ]
 
+
+def get_py_ver():
+    major, minor = sys.version_info[:2]
+    return f"cp{major}{minor}"
+
+
 setup(
     name="sgl-kernel",
     version=_get_version(),
@@ -245,5 +251,5 @@ setup(
         "build_ext": BuildExtension.with_options(use_ninja=True),
         "build_py": CustomBuildPy,
     },
-    options={"bdist_wheel": {"py_limited_api": "cp39"}},
+    options={"bdist_wheel": {"py_limited_api": get_py_ver()}},
 )