Skip to content

GitLab

Unverified Commit 29daf498 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix cu118 link issue (#3421)

parent 6702592d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
sgl-kernel/setup.py
View file @ 29daf498
...@@ -63,7 +63,6 @@ include_dirs = [ ...@@ -63,7 +63,6 @@ include_dirs = [
flashinfer.resolve() / "include" / "gemm", flashinfer.resolve() / "include" / "gemm",
flashinfer.resolve() / "csrc", flashinfer.resolve() / "csrc",
"cublas", "cublas",
"cublasLt",
turbomind.resolve(), turbomind.resolve(),
turbomind.resolve() / "src", turbomind.resolve() / "src",
] ]
...@@ -143,7 +142,7 @@ for flag in [ ...@@ -143,7 +142,7 @@ for flag in [
pass pass
cxx_flags = ["-O3"] cxx_flags = ["-O3"]
libraries = ["c10", "torch", "torch_python", "cuda", "cublas", "cublasLt"] libraries = ["c10", "torch", "torch_python", "cuda", "cublas"]
extra_link_args = ["-Wl,-rpath,$ORIGIN/../../torch/lib", "-L/usr/lib/x86_64-linux-gnu"] extra_link_args = ["-Wl,-rpath,$ORIGIN/../../torch/lib", "-L/usr/lib/x86_64-linux-gnu"]
ext_modules = [ ext_modules = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment