From 1e24a32ff51a66c6238bcab74f68e21938133883 Mon Sep 17 00:00:00 2001
From: zhyncs <me@zhyncs.com>
Date: Thu, 12 Dec 2024 17:10:01 +0800
Subject: [PATCH] fix: set runtime path

---
 .github/workflows/release-pypi-kernel.yml | 6 +++++-
 sgl-kernel/pyproject.toml                 | 2 +-
 sgl-kernel/setup.py                       | 2 ++
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/release-pypi-kernel.yml b/.github/workflows/release-pypi-kernel.yml
index 1349aa4ad5c..f046538a6fa 100644
--- a/.github/workflows/release-pypi-kernel.yml
+++ b/.github/workflows/release-pypi-kernel.yml
@@ -8,12 +8,16 @@ on:
       - sgl-kernel/pyproject.toml
   workflow_dispatch:
 
+concurrency:
+  group: release-pypi-kernel-${{ github.ref }}
+  cancel-in-progress: true
+
 jobs:
   build-wheels:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.9', '3.10', '3.11', '3.12']
+        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
         cuda-version: ['12.1']
 
     steps:
diff --git a/sgl-kernel/pyproject.toml b/sgl-kernel/pyproject.toml
index ce434bc34ab..4fbdd9dae23 100644
--- a/sgl-kernel/pyproject.toml
+++ b/sgl-kernel/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sgl-kernel"
-version = "0.0.2"
+version = "0.0.2.post4"
 description = "Kernel Library for SGLang"
 readme = "README.md"
 requires-python = ">=3.8"
diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py
index cdf8bb2a066..724c634bc25 100644
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -82,6 +82,8 @@ def update_wheel_platform_tag():
                 ],
                 "cxx": ["-O3"],
             },
+            libraries=["c10", "torch", "torch_python"],
+            extra_link_args=["-Wl,-rpath,$ORIGIN/../../torch/lib"],
         )
     ],
     cmdclass={"build_ext": BuildExtension},