fix: set runtime path (#2466)

2024-12-12 18:05:48 +08:00
parent dedaf8cd48
commit 2673fa29d4
3 changed files with 8 additions and 2 deletions
--- a/sgl-kernel/setup.py
+++ b/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},