upd

sgl-project · Jan 15, 2025 · 7e9031a · 7e9031a
1 parent 2490694
commit 7e9031a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py
@@ -40,8 +40,8 @@ def update_wheel_platform_tag():
     "-U__CUDA_NO_HALF2_OPERATORS__",
 ]
 cxx_flags = ["-O3"]
-libraries = ["c10", "torch", "torch_python"]
-extra_link_args = ["-Wl,-rpath,$ORIGIN/../../torch/lib"]
+libraries = ["c10", "torch", "torch_python", "cuda"]
+extra_link_args = ["-Wl,-rpath,$ORIGIN/../../torch/lib", "-L/usr/lib/x86_64-linux-gnu"]
 ext_modules = [
     CUDAExtension(
         name="sgl_kernel.ops._kernels",