From 729055e496b162c7db26ca1d8833cee80b44660b Mon Sep 17 00:00:00 2001 From: "david.cheng" Date: Mon, 24 Mar 2025 12:59:09 +0000 Subject: [PATCH] get the python version from env --- 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 a7b61222029..0e3461cfe20 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -229,6 +229,12 @@ def copy_deepgemm_to_build_lib(self): ), ] + +def get_py_ver(): + major, minor = sys.version_info[:2] + return f"cp{major}{minor}" + + setup( name="sgl-kernel", version=_get_version(), @@ -239,5 +245,5 @@ def copy_deepgemm_to_build_lib(self): "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()}}, )