fix: sgl-kernel link cuda (#2906)

2025-01-16 04:53:23 +08:00
parent 6cb3974e77
commit a53454c55e
5 changed files with 41 additions and 45 deletions
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -41,7 +41,7 @@ nvcc_flags = [
 ]
 cxx_flags = ["-O3"]
 libraries = ["c10", "torch", "torch_python", "cuda"]
-extra_link_args = ["-Wl,-rpath,$ORIGIN/../../torch/lib"]
+extra_link_args = ["-Wl,-rpath,$ORIGIN/../../torch/lib", "-L/usr/lib/x86_64-linux-gnu"]
 ext_modules = [
    CUDAExtension(
        name="sgl_kernel.ops._kernels",