fix: set runtime path (#2466)

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"

sgl-kernel/setup.py

@@ -82,6 +82,8 @@ setup(
                 ],
                 "cxx": ["-O3"],
             },
+            libraries=["c10", "torch", "torch_python"],
+            extra_link_args=["-Wl,-rpath,$ORIGIN/../../torch/lib"],
         )
     ],
     cmdclass={"build_ext": BuildExtension},